Update app.py

app.py

@@ -287,7 +287,7 @@ def identify_required_functions(project_path, functionality_description):
 
 def extract_cleaned_gemini_output(gemini_output):
     """
-    Removes the 'Functions' section from Gemini output and cleans irrelevant content.
+    Removes the 'Functions' section and any irrelevant content from the Gemini output.
     Args:
         gemini_output (str): The raw output returned by Gemini.
     Returns:
@@ -300,18 +300,21 @@ def extract_cleaned_gemini_output(gemini_output):
     for line in lines:
         line = line.strip()
 
-        # Skip the "Functions" section entirely
+        # Detect the start of the "Functions:" section and skip it
         if line.startswith("Functions:"):
             skip_section = True
-        elif skip_section and line.startswith("Project Summary:"):
+
+        # Detect the end of the "Functions:" section and resume processing
+        if skip_section and (line.startswith("Tasks:") or line.startswith("Project Summary:")):
             skip_section = False
 
-        # Append lines that are not part of the skipped section
+        # Append lines outside the skipped section
         if not skip_section:
             cleaned_output.append(line)
 
-    # Filter out any additional blank lines for cleaner formatting
-    return "\n".join([line for line in cleaned_output if line])
+    # Return the cleaned output with blank lines filtered out
+    return "\n".join(line for line in cleaned_output if line)
+