Update app.py

app.py

@@ -36,15 +36,6 @@ def call_mistral_api(prompt):
 
 # Function to analyze a single requirement
 def analyze_requirement(requirement):
-    # Define test case expectations
-    expected_outcomes = {
-        "Type": "Functional or Non-Functional",
-        "Stakeholders": "List of stakeholders for the requirement",
-        "Domain": "Classify the domain (e.g., Bank, Healthcare)",
-        "Defects": "List of major defects or 'No defects' if none",
-        "Rewritten": "Concise, defect-free rewritten requirement"
-    }
-
     # Detect requirement type
     type_prompt = f"Classify the following requirement as Functional or Non-Functional:\n\n{requirement}\n\nType:"
     req_type = call_mistral_api(type_prompt)
@@ -67,14 +58,13 @@ def analyze_requirement(requirement):
     # Rewrite requirement
     rewritten = rewrite_requirement(requirement, defects)
 
-    # Return both the result and expected outcome comparison
     return {
         "Requirement": requirement,
-        "Type": {"Expected": expected_outcomes["Type"], "Actual": req_type},
-        "Stakeholders": {"Expected": expected_outcomes["Stakeholders"], "Actual": stakeholders},
-        "Domain": {"Expected": expected_outcomes["Domain"], "Actual": domain},
-        "Defects": {"Expected": expected_outcomes["Defects"], "Actual": defects},
-        "Rewritten": {"Expected": expected_outcomes["Rewritten"], "Actual": rewritten}
+        "Type": req_type,
+        "Stakeholders": stakeholders,
+        "Domain": domain,
+        "Defects": defects,
+        "Rewritten": rewritten
     }
 
 # Function to rewrite requirement concisely
@@ -130,11 +120,11 @@ def generate_pdf_report(results):
         pdf.set_font("Arial", 'B', 14)
         pdf.multi_cell(200, 10, txt=f"Requirement R{i}: {result['Requirement']}", align='L')
         pdf.set_font("Arial", size=12)
-        
-        # Display expected vs actual results
-        for key in ["Type", "Stakeholders", "Domain", "Defects", "Rewritten"]:
-            pdf.multi_cell(200, 10, txt=f"{key}:\nExpected: {result[key]['Expected']}\nActual: {result[key]['Actual']}", align='L')
-        
+        pdf.multi_cell(200, 10, txt=f"Type: {result['Type']}", align='L')
+        pdf.multi_cell(200, 10, txt=f"Stakeholders: {result['Stakeholders']}", align='L')
+        pdf.multi_cell(200, 10, txt=f"Domain: {result['Domain']}", align='L')
+        pdf.multi_cell(200, 10, txt=f"Defects: {result['Defects']}", align='L')
+        pdf.multi_cell(200, 10, txt=f"Rewritten: {result['Rewritten']}", align='L')
         pdf.multi_cell(200, 10, txt="-" * 50, align='L')
         pdf.ln(5)  # Add some space between requirements
 
@@ -169,8 +159,11 @@ def main():
             st.subheader("Analysis Results")
             for i, result in enumerate(results, start=1):
                 st.write(f"### Requirement R{i}: {result['Requirement']}")
-                for key in ["Type", "Stakeholders", "Domain", "Defects", "Rewritten"]:
-                    st.write(f"**{key}:** Expected: {result[key]['Expected']} | Actual: {result[key]['Actual']}")
+                st.write(f"**Type:** {result['Type']}")
+                st.write(f"**Stakeholders:** {result['Stakeholders']}")
+                st.write(f"**Domain:** {result['Domain']}")
+                st.write(f"**Defects:** {result['Defects']}")
+                st.write(f"**Rewritten:** {result['Rewritten']}")
                 st.write("---")
 
             # Generate and download PDF report
@@ -185,4 +178,4 @@ def main():
 
 # Run the app
 if __name__ == "__main__":
-    main()
+    main()