Update app.py

app.py

@@ -55,44 +55,56 @@ def call_groq_api(prompt):
 
 # Function to analyze a single requirement
 def analyze_requirement(requirement):
-    # Use Mistral for classification and domain identification
-    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()
-
-    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()
-
-    rewritten_prompt = f"""Rewrite the following requirement in 1-2 sentences to address the defects:\n\n{requirement}\n\nRewritten:"""
-    rewritten = call_groq_api(rewritten_prompt).strip()
-
-    return {
-        "Requirement": requirement,
-        "Type": req_type,
-        "Domain": domain,
-        "Defects": defects,  # This is now a clean list
-        "Rewritten": rewritten
-    }
-
+    try:
+        # 1. Classify requirement type
+        type_prompt = f"Classify this requirement as Functional or Non-Functional (one word only):\n{requirement}"
+        req_type = call_mistral_api(type_prompt).strip()
+        
+        # 2. Identify domain
+        domain_prompt = f"Identify the domain for this requirement (one word only, e.g., Healthcare, Finance):\n{requirement}"
+        domain = call_mistral_api(domain_prompt).strip()
+        
+        # 3. Detect defects
+        defects_prompt = f"""List ONLY the major defects in this requirement 
+        (EXACTLY 3 bullet points, each defect 1-2 words maximum):
+        - Defect1
+        - Defect2
+        - Defect3
+        
+        Requirement: {requirement}
+        Defects:"""
+        defects_response = call_groq_api(defects_prompt).strip()
+        
+        # Process defects
+        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"]
+        
+        # 4. Rewrite requirement
+        rewrite_prompt = f"""Rewrite this requirement to fix defects while keeping it concise 
+        (1-2 sentences maximum):\n{requirement}"""
+        rewritten = call_groq_api(rewrite_prompt).strip()
+        
+        return {
+            "Requirement": requirement,
+            "Type": req_type if "HTTP Error" not in req_type else "Unknown",
+            "Domain": domain if "HTTP Error" not in domain else "Unknown",
+            "Defects": defects,
+            "Rewritten": rewritten if "HTTP Error" not in rewritten else requirement
+        }
+        
+    except Exception as e:
+        st.error(f"Analysis failed: {str(e)}")
+        return {
+            "Requirement": requirement,
+            "Type": "Error",
+            "Domain": "Error",
+            "Defects": ["Analysis Failed"],
+            "Rewritten": requirement
+        }
 # Function to generate a PDF report
 def generate_pdf_report(results):
     pdf = FPDF()