Update app.py

app.py

@@ -57,17 +57,17 @@ def call_groq_api(prompt):
 def analyze_requirement(requirement):
     # Use Mistral for classification and domain identification
     type_prompt = f"Classify the following requirement as Functional or Non-Functional:\n\n{requirement}\n\nType:"
-    req_type = call_mistral_api(type_prompt)
+    req_type = call_mistral_api(type_prompt).strip()
 
     domain_prompt = f"Classify the domain for the following requirement (e.g., Bank, Healthcare, etc.):\n\n{requirement}\n\nDomain:"
-    domain = call_mistral_api(domain_prompt)
+    domain = call_mistral_api(domain_prompt).strip()
 
     # Use Groq for defect analysis and rewriting
     defects_prompt = f"""Analyze the following requirement and identify ONLY MAJOR defects (e.g., Ambiguity, Incompleteness, etc.). 
     If the requirement is clear and complete, respond with 'No defects.' 
     Requirement: {requirement}
     Defects:"""
-    defects = call_groq_api(defects_prompt)
+    defects = call_groq_api(defects_prompt).strip()
 
     rewritten = rewrite_requirement(requirement, defects)