Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import sqlite3
 import hashlib
 import os
+import google.generativeai as genai
 import zipfile
 from git import Repo
 
@@ -198,6 +199,66 @@ def workspace_page():
                 except Exception as e:
                     st.error(f"Failed to clone repository: {e}")
 
+#------------------------------------------------------------------------------------------------------------------------------------------------------------------------
+
+
+# Configure Gemini API
+api_key = os.getenv("GEMINI")
+genai.configure(api_key)
+model = genai.GenerativeModel("gemini-1.5-flash")
+
+def read_project_files(project_path):
+    """Reads all files in the project directory and its subdirectories."""
+    file_paths = []
+    for root, _, files in os.walk(project_path):
+        for file in files:
+            file_paths.append(os.path.join(root, file))
+    return file_paths
+
+def read_files(file_paths):
+    """Reads content from a list of file paths."""
+    file_contents = {}
+    for file_path in file_paths:
+        if os.path.exists(file_path):
+            with open(file_path, 'r') as file:
+                file_contents[file_path] = file.read()
+        else:
+            print(f"File not found: {file_path}")
+    return file_contents
+
+def generate_prompt(file_contents, functionality_description):
+    """Generates a prompt for Gemini to analyze the files."""
+    prompt = "Analyze the following code files to identify all functions required to implement the functionality: "
+    prompt += f"'{functionality_description}'.\n\n"
+
+    for file_path, content in file_contents.items():
+        prompt += f"File: {os.path.basename(file_path)}\n{content}\n\n"
+    
+    prompt += "For each relevant function, provide:\n"
+    prompt += "1. The function name.\n"
+    prompt += "2. A brief description of its purpose.\n"
+    prompt += "3. Its inputs and outputs.\n"
+    prompt += "4. Dependencies on other functions or modules.\n"
+
+    return prompt
+
+def identify_required_functions(project_path, functionality_description):
+    """Identifies required functions for a specified functionality."""
+    # Gather all file paths in the project directory
+    file_paths = read_project_files(project_path)
+    
+    # Read file contents
+    file_contents = read_files(file_paths)
+    
+    # Generate a refined prompt for Gemini
+    prompt = generate_prompt(file_contents, functionality_description)
+    
+    # Call the Gemini model
+    response = model.generate_content(prompt)
+    
+    # Process and return the response
+    return response.text
+
 def generate_documentation_page():
     st.subheader(f"Generate Documentation for {st.session_state.current_project}")
     st.write("Enter the functionality or parts of the project for which you'd like to generate documentation.")
@@ -208,8 +269,19 @@ def generate_documentation_page():
     # Button to start generating documentation
     if st.button("Generate"):
         if functionality.strip():
-            # Placeholder for documentation generation logic
-            st.write(f"Generating documentation for: {functionality}")
+            st.write("Analyzing project files... Please wait.")
+            
+            # Get the path of the current project
+            user_folder = os.path.join("user_projects", st.session_state.username)
+            project_folder = os.path.join(user_folder, st.session_state.current_project)
+            
+            # Call the function to identify required functions
+            try:
+                result = identify_required_functions(project_folder, functionality)
+                st.success("Documentation generated successfully!")
+                st.text_area("Generated Documentation", result, height=400)
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
         else:
             st.error("Please enter the functionality to generate documentation.")
 
@@ -219,6 +291,9 @@ def generate_documentation_page():
         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.")