Update app.py

app.py

@@ -59,9 +59,25 @@ def analyze_requirement(requirement):
     type_prompt = f"Classify the following requirement as Functional or Non-Functional in one word:\n\n{requirement}\n\nType:"
     req_type = call_mistral_api(type_prompt).strip()
 
-    domain_prompt = f"Classify the domain for the following requirement in one word (e.g., E-commerce, Education, etc.):\n\n{requirement}\n\nDomain:"
+    defects_prompt = f"""List ONLY the major defects in this requirement (EXACTLY 3 bullet points, 
+    each defect 1-2 words maximum, no explanations, format exactly like this example):
+    - Ambiguity
+    - Vagueness
+    - Incompleteness
+    
+    Requirement: {requirement}
+    Defects:"""
     domain = call_mistral_api(domain_prompt).strip()
 
+    # Process defects response
+    defects = []
+    if "API Error" not in defects_response:
+        defects = [line.strip()[2:] for line in defects_response.split("\n") 
+                  if line.strip().startswith("- ")]
+    
+    if not defects:
+        defects = ["No major defects found"]
+
     # Use Groq for defect analysis and rewriting
     defects_prompt = f"""List ONLY the major defects in the following requirement (e.g., Ambiguity, Incompleteness, etc.) in 1-2 words each:\n\n{requirement}\n\nDefects:"""
     defects = call_groq_api(defects_prompt).strip()
@@ -73,7 +89,7 @@ def analyze_requirement(requirement):
         "Requirement": requirement,
         "Type": req_type,
         "Domain": domain,
-        "Defects": defects,
+        "Defects": defects,  # This is now a clean list
         "Rewritten": rewritten
     }
 
@@ -240,14 +256,14 @@ def main():
                                 <h4>🔎 Identified Issues</h4>
                                 <div style="display: flex; gap: 0.5rem; flex-wrap: wrap;">
                                     {''.join([f'<div class="analysis-badge defect-badge">⚠️ {d}</div>' 
-                                            for d in result['Defects'].split(', ')])}
+                                            for d in result["Defects"]])}
                                 </div>
                             </div>
                             
                             <div style="margin: 1rem 0;">
                                 <h4>✨ Improved Version</h4>
                                 <div class="analysis-badge improved-badge">
-                                    📝 {result['Rewritten']}
+                                    📝 {result["Rewritten"]}
                                 </div>
                             </div>
                         </div>