Update command/basic/documentation.py

command/basic/documentation.py

@@ -1,6 +1,13 @@
+import os 
+import logging
+
+import streamlit as st 
+import core.form_documentation as fd 
+
 from typing import Optional
 from ..command_handler import CommandHandler
 
+
 class DocumentationHandler(CommandHandler):
    def __init__(self, commands, successor: Optional["CommandHandler"] = None):
         super().__init__(successor)
@@ -8,10 +15,18 @@ class DocumentationHandler(CommandHandler):
 
     def handle_command(self, command):
         if command.lower() in self.commands:
-            print(self.execute_command())
+            self.execute_command()
         else:
             super().handle_command(command)
 
     def execute_command(self):
-        return "Download ..."
+        current_dir = os.path.dirname(os.path.realpath(__file__)) 
+        path_to_file = fd.generate_doc(path=current_dir)
+
+        if path_to_file is not None:
+             with st.sidebar:
+                 st.download_button('Download doc', path_to_file)
+                 st.success("Done", icon="✅")
+        else:
+            logging.error("path_to_file is None")