Update app.py

app.py

@@ -53,11 +53,6 @@ def call_groq_api(prompt):
         st.error(f"Error: {err}")
         return f"Error: {err}"
 
-def clean_text(text):
-    # Remove markdown code fences and any trailing newlines/spaces
-    cleaned = text.replace("```", "").strip()
-    return cleaned
-
 # Function to analyze a single requirement
 def analyze_requirement(requirement):
     # Use Mistral for classification and domain identification
@@ -68,13 +63,17 @@ def analyze_requirement(requirement):
     domain = call_mistral_api(domain_prompt).strip()
 
     # 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_raw = call_groq_api(defects_prompt)
-    defects = clean_text(defects_raw)
+    defects_prompt = f"""List ONLY the major defects in the following requirement as comma-separated values (1-2 words each). Valid options: Ambiguity, Vagueness, Incompleteness, Volatility, Unverifiability. Do NOT add explanations:
+    {requirement}
+    Defects:"""
+    defects = defects.split(":")[-1].strip() if ":" in defects else defects
+    defects = defects.replace("Defects", "").strip()
 
-    rewritten_prompt = f"""Rewrite the following requirement in 1-2 sentences to address the defects:\n\n{requirement}\n\nRewritten:"""
-    rewritten_raw = call_groq_api(rewritten_prompt)
-    rewritten = clean_text(rewritten_raw)
+    rewritten_prompt = f"""Rewrite the following requirement in 1-2 sentences to address defects. Start directly with "The project shall..." without preamble:
+    {requirement}
+    Improved:"""
+    rewritten = rewritten.split(":")[-1].strip() if ":" in rewritten else rewritten
+    rewritten = rewritten.replace("Rewritten", "").replace("Improved", "").strip()
 
     return {
         "Requirement": requirement,
@@ -169,6 +168,15 @@ st.markdown("""
             font-weight: 500;
             margin: 0.3rem;
         }
+        .defect-badge { 
+            white-space: nowrap;
+            margin: 0.2rem 0.3rem !important;
+        }
+        .improved-badge {
+            line-height: 1.5;
+            text-align: left;
+            width: 100%;
+        }
         .download-btn {
             width: 100%;
             padding: 1rem;
@@ -247,7 +255,7 @@ 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'].split(', ') if d.strip()])
                                 </div>
                             </div>