Update app.py

app.py

@@ -56,7 +56,8 @@ def analyze_requirement(requirement):
     defects = call_mistral_api(defects_prompt)
 
     # Rewrite requirement
-    rewritten = rewrite_requirement(requirement, defects)
+    rewritten_prompt = f"""Rewrite the following requirement in 1-2 sentences to address the defects:\n\n{requirement}\n\nRewritten:"""
+    rewritten = call_mistral_api(rewritten_prompt).strip()
 
     return {
         "Requirement": requirement,