Update app.py

app.py

@@ -2,15 +2,16 @@ import streamlit as st
 import requests
 from fpdf import FPDF
 import pdfplumber
+import os
 
-# Mistral API key (replace with your key)
-API_KEY = "gz6lDXokxgR6cLY72oomALWcm7vhjzQ"
-MISTRAL_API_URL = "https://api.mistral.ai/v1/chat/completions"
+# Mistral API key (replace with your key or use environment variable)
+api_key = os.getenv("MISTRAL_API_KEY", "gz6lDXokxgR6cLY72oomALWcm7vhjRzQ")
 
 # Function to call Mistral API
 def call_mistral_api(prompt):
+    url = "https://api.mistral.ai/v1/chat/completions"
     headers = {
-        "Authorization": f"Bearer {API_KEY}",
+        "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json"
     }
     payload = {
@@ -19,11 +20,14 @@ def call_mistral_api(prompt):
             {"role": "user", "content": prompt}
         ]
     }
-    response = requests.post(MISTRAL_API_URL, headers=headers, json=payload)
-    if response.status_code == 200:
+    try:
+        response = requests.post(url, headers=headers, json=payload)
+        response.raise_for_status()  # Raise an error for bad status codes
         return response.json()['choices'][0]['message']['content']
-    else:
-        return f"Error: {response.status_code}, {response.text}"
+    except requests.exceptions.HTTPError as err:
+        return f"HTTP Error: {err}"
+    except Exception as err:
+        return f"Error: {err}"
 
 # Function to analyze a single requirement
 def analyze_requirement(requirement):
@@ -40,7 +44,10 @@ def analyze_requirement(requirement):
     domain = call_mistral_api(domain_prompt)
 
     # Detect defects
-    defects_prompt = f"Detect any defects in the following requirement (e.g., Ambiguity, Incompleteness, etc.):\n\n{requirement}\n\nDefects:"
+    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_mistral_api(defects_prompt)
 
     # Rewrite requirement
@@ -77,9 +84,9 @@ def generate_pdf_report(results):
 
 # Streamlit app
 def main():
-    st.title("Requirement Defects Detective Tool")
-    st.markdown("**Team Name:** Sadia, Areeba, Rabbia, Tesmia")
-    st.markdown("**LLM Model:** Mistral")
+    st.title("Requirement Analysis Tool")
+    st.markdown("**Team Name:** Team Innovators")
+    st.markdown("**Model:** Mistral")
 
     # Input options
     input_option = st.radio("Choose input method:", ("Enter Requirements", "Upload PDF"))