Update app.py

app.py

@@ -645,7 +645,7 @@ def project_view_page():
     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"):
-        st.session_state.page = "workspace"
+        st.session_state.page = "project_staging"  # Corrected navigation
         st.rerun()
     if st.sidebar.button("Log Out"):
         st.session_state.authenticated = False
@@ -660,19 +660,16 @@ def project_view_page():
     # Calculate project metrics
     user_folder = os.path.join("user_projects", st.session_state.username)
     project_folder = os.path.join(user_folder, st.session_state.current_project)
-    
+
     # Count number of files
-    num_files = sum([len(files) for _, _, files in os.walk(project_folder) if ".git" not in _])
+    num_files = sum([len(files) for _, _, files in os.walk(project_folder)])
 
     # Count total lines of code in all text files
     num_lines = 0
     for root, _, files in os.walk(project_folder):
-        if ".git" in root:
-            continue  # Skip .git directory
         for file in files:
             file_path = os.path.join(root, file)
             try:
-                # Only include text files (e.g., .py, .txt, .json, etc.)
                 with open(file_path, 'r', encoding='utf-8') as f:
                     num_lines += sum(1 for _ in f)
             except (UnicodeDecodeError, IsADirectoryError):
@@ -706,16 +703,17 @@ def project_view_page():
     if st.session_state.show_file_structure:
         st.write("File structure:")
         for root, dirs, files in os.walk(project_folder):
-            if ".git" in root:
-                continue  # Skip .git directory
             level = root.replace(project_folder, "").count(os.sep)
             indent = " " * 4 * level
 
+            # Display folders and files recursively
             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)
+                    for folder in dirs:
+                        st.write(f"{sub_indent}📂 {folder}")
                     for file in files:
                         st.write(f"{sub_indent}📄 {file}")