Update app.py

app.py

@@ -287,38 +287,32 @@ 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 or display to the user.
+    Removes the 'Functions' section from Gemini output and cleans irrelevant content.
     Args:
         gemini_output (str): The raw output returned by Gemini.
     Returns:
-        str: Cleaned and formatted output.
+        str: Cleaned output without the 'Functions' section.
     """
     lines = gemini_output.splitlines()
     cleaned_output = []
-    capture = False  # Flag to determine whether to capture the lines
+    skip_section = False
 
     for line in lines:
         line = line.strip()
 
-        # Start capturing from "Project Summary:" and include only relevant sections
-        if line.startswith("Project Summary:"):
-            capture = True
-            cleaned_output.append(line)
-        elif line.startswith("Functionality:") or line.startswith("Functionality Flow:"):
-            cleaned_output.append(line)
-        elif line.startswith("Function Documentation:"):
-            cleaned_output.append(line)
-        elif line.startswith("Functions:"):
-            capture = True  # Start capturing function definitions
+        # Skip the "Functions" section entirely
+        if line.startswith("Functions:"):
+            skip_section = True
+        elif skip_section and line.startswith("Project Summary:"):
+            skip_section = False
+
+        # Append lines that are not part of the skipped section
+        if not skip_section:
             cleaned_output.append(line)
-        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)
+    # Filter out any additional blank lines for cleaner formatting
+    return "\n".join([line for line in cleaned_output if line])
+