Update app.py

app.py

@@ -367,22 +367,27 @@ def clean_output(output):
 
     for line in lines:
         line = line.strip()
+        # Start processing when encountering desired sections
         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)
+        # Stop processing at unwanted sections or echo of the prompt
+        elif line.startswith("User-specified functionality:") or line.startswith("Functions identified by Gemini:"):
+            in_valid_section = False
+        # Continue collecting lines within valid sections
         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 only relevant lines
     return "\n".join(filtered_lines)
 
 
 
 
+
 def validate_and_generate_documentation(api_url, headers, gemini_output, functionality_description):
     """
     Uses the Hugging Face Inference API to generate clean and relevant documentation using Qwen.