Update app.py

app.py

@@ -63,17 +63,11 @@ 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 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()
+    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 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()
+    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,
@@ -157,7 +151,7 @@ st.markdown("""
             margin-bottom: 1.5rem;
             transition: transform 0.2s;
             color: #333333;
-            
+
         }
         .requirement-card:hover {
             transform: translateY(-3px);
@@ -168,15 +162,6 @@ 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;
@@ -224,12 +209,12 @@ def main():
             with st.spinner("🔍 Analyzing requirements..."):
                 requirements = [req.strip() for req in input_text.replace("\n", ".").split(".") if req.strip()]
                 results = []
-                
+
                 progress_bar = st.progress(0)
                 for i, req in enumerate(requirements):
                     results.append(analyze_requirement(req))
                     progress_bar.progress((i+1)/len(requirements))
-                
+
                 st.success("✅ Analysis Completed!")
                 time.sleep(0.5)
                 st.session_state.results = results
@@ -250,15 +235,15 @@ def main():
                                     🏷️ Domain: {result['Domain']}
                                 </div>
                             </div>
-                            
+
                             <div style="margin: 1rem 0;">
                                 <h4>🔎 Identified Issues</h4>
                                 <div style="display: flex; gap: 0.5rem; flex-wrap: wrap;">
-                                    {''.join([f'<div class="analysis-badge defect-badge">⚠️ {d.strip()}</div>'  
-                                            for d in result['Defects'].split(', ') if d.strip()])
+                                    {''.join([f'<div class="analysis-badge defect-badge">⚠️ {d}</div>'
+                                            for d in result['Defects'].split(', ')])}
                                 </div>
                             </div>
-                            
+
                             <div style="margin: 1rem 0;">
                                 <h4>✨ Improved Version</h4>
                                 <div class="analysis-badge improved-badge">