Update app.py

app.py

@@ -295,23 +295,27 @@ def extract_cleaned_gemini_output(gemini_output):
     """
     lines = gemini_output.splitlines()
     cleaned_output = []
-    functions_section = False
+    in_functions_section = False
 
     for line in lines:
         line = line.strip()
+
+        # Include only relevant sections
         if line.startswith("Project Summary:") or line.startswith("Functionality:"):
             cleaned_output.append(line)
         elif line.startswith("Functions:"):
             cleaned_output.append(line)
-            functions_section = True
-        elif functions_section and line:
+            in_functions_section = True
+        elif in_functions_section and line.startswith("-public"):
             cleaned_output.append(line)
-        elif line.startswith("File:") or "Qwen," in line:
-            break
+        elif in_functions_section and not line.startswith("-"):
+            # Exit functions section when encountering non-relevant lines
+            in_functions_section = 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)]