Update app.py

app.py

@@ -56,8 +56,7 @@ def analyze_requirement(requirement):
     defects = call_mistral_api(defects_prompt)
 
     # Rewrite requirement
-    rewrite_prompt = f"Rewrite the following requirement to address the defects listed below. Ensure the rewritten requirement is clear, complete, and addresses all issues:"
-    rewritten = call_mistral_api(rewrite_prompt)
+    rewritten = rewrite_requirement(requirement, defects)
 
     return {
         "Requirement": requirement,
@@ -68,6 +67,22 @@ def analyze_requirement(requirement):
         "Rewritten": rewritten
     }
 
+# Function to rewrite requirement
+def rewrite_requirement(requirement, defects):
+    if "no defects" in defects.lower():
+        return "No modification needed."
+
+    # If defects are found, generate a clearer and more complete requirement
+    prompt = f"""Rewrite the following requirement to address the defects listed below. Ensure the rewritten requirement is clear, complete, and addresses all issues:
+    
+    Original Requirement: {requirement}
+    
+    Defects: {defects}
+    
+    Rewritten Requirement:"""
+    response = call_mistral_api(prompt)
+    return response.strip()
+
 # Function to generate a PDF report
 def generate_pdf_report(results):
     pdf = FPDF()