Update app.py

app.py

@@ -359,7 +359,7 @@ def extract_cleaned_gemini_output(gemini_output):
 
 def clean_output(output):
     """
-    Cleans the output from Qwen to ensure only required sections are displayed.
+    Cleans the final output to include only documentation sections.
     """
     lines = output.splitlines()
     filtered_lines = []
@@ -367,27 +367,26 @@ def clean_output(output):
 
     for line in lines:
         line = line.strip()
-        # Start processing when encountering desired sections
+        # Include only valid 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:"):
+        # Stop processing at any undesired section
+        elif line.startswith("User-specified functionality:") or line.startswith("Tasks identified by Gemini:"):
             in_valid_section = False
-        # Continue collecting lines within valid sections
         elif in_valid_section and line:
             filtered_lines.append(line)
 
-    # 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.