Update app.py

app.py

@@ -641,7 +641,7 @@ 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.session_state.current_project else 'No Project Selected'}")
+    st.sidebar.title(f"Project: {st.session_state.current_project}")
     if st.sidebar.button("Back to Project Staging"):
         st.session_state.page = "workspace"
         st.rerun()
@@ -651,68 +651,44 @@ 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
+    # 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)
+    num_files = sum([len(files) for _, _, files in os.walk(project_folder)])
+    num_lines = sum([sum(1 for _ in open(os.path.join(root, file), 'r', encoding="utf-8"))
+                     for root, _, files in os.walk(project_folder) for file in files if file.endswith(".py")])
 
-    # 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
-
-    for root, dirs, files in os.walk(project_folder):
-        file_count += len(files)
-        for file in files:
-            file_path = os.path.join(root, file)
-            try:
-                with open(file_path, "r", encoding="utf-8") as f:
-                    total_lines += sum(1 for line in f)
-            except (UnicodeDecodeError, FileNotFoundError):
-                # Skip non-text files or files that can't be read
-                continue
-
-    # Display project metrics
-    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)
-    st.write("### Actions")
-    col1, col2, col3 = st.columns(3)
-
+    # Display metrics side by side
+    col1, col2 = st.columns(2)
     with col1:
+        st.metric(label="Number of Files", value=num_files)
+    with col2:
+        st.metric(label="Lines of Code", value=num_lines)
+
+    # Buttons for documentation functionality displayed side by side
+    col3, col4, col5 = st.columns(3)
+    with col3:
         if st.button("Generate Documentation"):
             st.session_state.page = "generate_documentation"
             st.rerun()
-
-    with col2:
+    with col4:
         if st.button("Saved Documentation"):
             st.session_state.page = "saved_documentation"
             st.rerun()
-
-    with col3:
+    with col5:
         if st.button("Show File Structure"):
             st.session_state.show_file_structure = not st.session_state.show_file_structure
 
-    # Toggle file structure display (if required)
+    # Display file structure if toggled
     if "show_file_structure" not in st.session_state:
         st.session_state.show_file_structure = False
 
     if st.session_state.show_file_structure:
-        st.write("### Project File Structure")
+        st.write("File structure:")
         for root, dirs, files in os.walk(project_folder):
             level = root.replace(project_folder, "").count(os.sep)
             indent = " " * 4 * level
@@ -727,6 +703,7 @@ def project_view_page():
 
 
 
+
 if __name__ == "__main__":
     main()