Update app.py

app.py

@@ -287,35 +287,42 @@ def identify_required_functions(project_path, functionality_description):
 
 def extract_cleaned_gemini_output(gemini_output):
     """
-    Extracts and formats the cleaned output from Gemini to send to Qwen.
+    Extracts and formats the cleaned output from Gemini to send to Qwen or display to the user.
     Args:
-        gemini_output (str): The output returned by Gemini.
+        gemini_output (str): The raw output returned by Gemini.
     Returns:
-        str: Cleaned and formatted output for Qwen.
+        str: Cleaned and formatted output.
     """
     lines = gemini_output.splitlines()
     cleaned_output = []
-    in_functions_section = False
+    capture = False  # Flag to determine whether to capture the lines
 
     for line in lines:
         line = line.strip()
 
-        # Include only relevant sections
-        if line.startswith("Project Summary:") or line.startswith("Functionality:"):
+        # Start capturing from "Project Summary:" and include only relevant sections
+        if line.startswith("Project Summary:"):
+            capture = True
             cleaned_output.append(line)
-        elif line.startswith("Functions:"):
+        elif line.startswith("Functionality:") or line.startswith("Functionality Flow:"):
             cleaned_output.append(line)
-            in_functions_section = True
-        elif in_functions_section and line.startswith("-public"):
+        elif line.startswith("Function Documentation:"):
+            cleaned_output.append(line)
+        elif line.startswith("Functions:"):
+            capture = True  # Start capturing function definitions
             cleaned_output.append(line)
-        elif in_functions_section and not line.startswith("-"):
-            # Exit functions section when encountering non-relevant lines
-            in_functions_section = False
+        elif capture:
+            # Capture relevant content under active sections
+            if line.startswith("-public") or line.startswith("-private"):
+                cleaned_output.append(line)
+            elif not line:  # Stop capturing on empty lines after a valid section
+                capture = False
 
     return "\n".join(cleaned_output)
 
 
 
+
 def split_into_chunks(content, chunk_size=1000):
     """Splits large content into smaller chunks."""
     return [content[i:i + chunk_size] for i in range(0, len(content), chunk_size)]