Update app.py

app.py

@@ -433,6 +433,8 @@ def validate_and_generate_documentation(api_url, headers, gemini_output, functio
     5. Return only the required information for the above tasks, and exclude everything else.
     """
 
+    qwen_prompt_length = len(prompt)
+    
     payload = {"inputs": prompt, "parameters": {"max_new_tokens": 1024}}
     response = requests.post(api_url, headers=headers, json=payload)
 
@@ -442,7 +444,7 @@ def validate_and_generate_documentation(api_url, headers, gemini_output, functio
         output = api_response.get("generated_text", "") if isinstance(api_response, dict) else api_response[0].get("generated_text", "")
 
         # Remove the Gemini content from the top of the Qwen output
-        trimmed_output = output[gemini_output_length:].strip()
+        trimmed_output = output[gemini_output_length + qwen_prompt_length:].strip()
 
         return clean_output(trimmed_output)  # Final cleanup if necessary
     else: