Update app.py

app.py

@@ -641,9 +641,9 @@ def saved_documentation_page():
 def project_view_page():
     # Sidebar with logout and return buttons
     st.sidebar.image("SimplifAI Logo Long.jpeg", use_container_width=True)
-    st.sidebar.title(f"Project: {st.session_state.current_project}")
-    if st.sidebar.button("Back to Project Staging"):  # Updated button label
-        st.session_state.page = "project_staging"  # Updated routing
+    st.sidebar.title(f"Project: {st.session_state.current_project if st.session_state.current_project else 'No Project Selected'}")
+    if st.sidebar.button("Back to Project Staging"):
+        st.session_state.page = "workspace"
         st.rerun()
     if st.sidebar.button("Log Out"):
         st.session_state.authenticated = False
@@ -651,10 +651,24 @@ def project_view_page():
         st.session_state.page = "login"
         st.rerun()
 
+    # Ensure a project is selected
+    if not st.session_state.current_project:
+        st.error("No project selected. Please select a project from the Project Staging page.")
+        return
+
     # Main content for project page
     st.subheader(f"Project: {st.session_state.current_project}")
     st.write("Manage your project and explore its files.")
 
+    # Get project folder path
+    user_folder = os.path.join("user_projects", st.session_state.username)
+    project_folder = os.path.join(user_folder, st.session_state.current_project)
+
+    # Check if the project folder exists
+    if not os.path.exists(project_folder):
+        st.error(f"Project folder not found: {project_folder}")
+        return
+
     # Calculate number of files and lines of code
     file_count = 0
     total_lines = 0
@@ -674,8 +688,8 @@ def project_view_page():
     st.write("### Project Metrics")
     st.metric(label="Total Files", value=file_count)
     st.metric(label="Lines of Code", value=total_lines)
-    
-  # Buttons for documentation functionality (side-by-side)
+
+    # Buttons for documentation functionality (side-by-side)
     st.write("### Actions")
     col1, col2, col3 = st.columns(3)
 
@@ -698,14 +712,10 @@ def project_view_page():
         st.session_state.show_file_structure = False
 
     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)
-
-        st.write("File structure:")
-
+        st.write("### Project File Structure")
         for root, dirs, files in os.walk(project_folder):
             level = root.replace(project_folder, "").count(os.sep)
-            indent = "->" * 4 * level
+            indent = " " * 4 * level
 
             if level == 0:
                 st.write(f"📂 {os.path.basename(root)}")