Update app.py

app.py

@@ -69,6 +69,10 @@ def main():
         workspace_page()
     elif st.session_state.page == "project_view":
         project_view_page()
+    elif st.session_state.page == "generate_documentation":
+        generate_documentation_page()
+    elif st.session_state.page == "view_documentation":
+        view_documentation_page()
 
 def login_page():
     st.subheader("Please Log In or Register to Continue")
@@ -194,29 +198,52 @@ def workspace_page():
                 except Exception as e:
                     st.error(f"Failed to clone repository: {e}")
 
+def generate_documentation_page():
+    st.subheader(f"Generate Documentation for {st.session_state.current_project}")
+    st.write("This page will allow the user to generate documentation for the selected project.")
+    if st.button("Back to Project"):
+        st.session_state.page = "project_view"
+        st.rerun()
+
+def view_documentation_page():
+    st.subheader(f"View Documentation for {st.session_state.current_project}")
+    st.write("This page will display the generated documentation for the selected project.")
+    if st.button("Back to Project"):
+        st.session_state.page = "project_view"
+        st.rerun()
+
 def project_view_page():
     # Sidebar with logout and return buttons
     st.sidebar.title(f"Project: {st.session_state.current_project}")
     if st.sidebar.button("Back to Workspace"):
         st.session_state.page = "workspace"
+        st.rerun()
     if st.sidebar.button("Log Out"):
         st.session_state.authenticated = False
         st.session_state.username = None
         st.session_state.page = "login"
-
-    # Initialize toggle state for file structure
-    if "show_file_structure" not in st.session_state:
-        st.session_state.show_file_structure = False
+        st.rerun()
 
     # Main content for project page
     st.subheader(f"Project: {st.session_state.current_project}")
     st.write("Manage your project and explore its files.")
 
-    # Toggle button for file structure
+    # Buttons for documentation functionality
+    if st.button("Generate Documentation"):
+        st.session_state.page = "generate_documentation"
+        st.rerun()
+
+    if st.button("View Documentation"):
+        st.session_state.page = "view_documentation"
+        st.rerun()
+
+    # Toggle file structure display (if required)
+    if "show_file_structure" not in st.session_state:
+        st.session_state.show_file_structure = False
+
     if st.button("Show File Structure"):
         st.session_state.show_file_structure = not st.session_state.show_file_structure
 
-    # Display file structure if toggled on
     if st.session_state.show_file_structure:
         user_folder = os.path.join("user_projects", st.session_state.username)
         project_folder = os.path.join(user_folder, st.session_state.current_project)
@@ -227,18 +254,16 @@ def project_view_page():
             level = root.replace(project_folder, "").count(os.sep)
             indent = " " * 4 * level
 
-            # Display the folder name
             if level == 0:
                 st.write(f"📂 {os.path.basename(root)}")
             else:
                 with st.expander(f"{indent}📂 {os.path.basename(root)}"):
                     sub_indent = " " * 4 * (level + 1)
-
-                    # List files in the folder
                     for file in files:
                         st.write(f"{sub_indent}📄 {file}")
 
 
+
 if __name__ == "__main__":
     main()