Update app.py

app.py

@@ -349,6 +349,28 @@ def extract_cleaned_gemini_output(gemini_output):
 
 
 
+def clean_output(output):
+    """
+    Cleans the output from Qwen to ensure only required sections are displayed.
+    """
+    lines = output.splitlines()
+    filtered_lines = []
+    in_valid_section = False
+
+    for line in lines:
+        line = line.strip()
+        if line.startswith("Project Summary:") or line.startswith("Functionality Summary:") or line.startswith("Functionality Flow:"):
+            in_valid_section = True
+            filtered_lines.append(line)
+        elif line.startswith("Function Documentation:"):
+            in_valid_section = True
+            filtered_lines.append(line)
+        elif in_valid_section and line:
+            filtered_lines.append(line)
+        elif line.startswith("File:") or line.startswith("User-specified functionality:"):
+            in_valid_section = False
+
+    return "\n".join(filtered_lines)