Update app.py

app.py

@@ -4,49 +4,50 @@ from fpdf import FPDF
 import os
 import time
 from datetime import datetime
+import groq
 
-# Groq API key (replace with your actual key)
-groq_api_key = "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"
+# Groq API key (replace with your key or use environment variable)
+api_key = os.getenv("GROQ_API_KEY", "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")
 
-# Function to call Groq Llama API
+# Initialize Groq client
+groq_client = groq.Client(api_key=api_key)
+
+# Function to call Groq API with rate limit handling
 def call_groq_api(prompt):
-    url = "https://api.groq.com/openai/v1/chat/completions"
-    
-    headers = {
-        "Authorization": f"Bearer {groq_api_key}",
-        "Content-Type": "application/json"
-    }
-    payload = {
-        "model": "llama-2-13b-chat",
-        "messages": [{"role": "user", "content": prompt}]
-    }
     try:
-        response = requests.post(url, headers=headers, json=payload)
-        response.raise_for_status()  
-        return response.json()['choices'][0]['message']['content']
-    except requests.exceptions.HTTPError as err:
-        return f"HTTP Error: {err}"
+        response = groq_client.chat.completions.create(
+            model="llama2-70b-4096",
+            messages=[
+                {"role": "user", "content": prompt}
+            ]
+        )
+        return response.choices[0].message.content
     except Exception as err:
         return f"Error: {err}"
 
-# Function to analyze requirements
-def analyze_requirement_groq(requirement):
+# Function to analyze a single requirement
+def analyze_requirement(requirement):
+    # Detect requirement type
     type_prompt = f"Classify the following requirement as Functional or Non-Functional:\n\n{requirement}\n\nType:"
     req_type = call_groq_api(type_prompt)
 
+    # Identify stakeholders
     stakeholders_prompt = f"Identify the stakeholders for the following requirement:\n\n{requirement}\n\nStakeholders:"
     stakeholders = call_groq_api(stakeholders_prompt)
 
+    # Classify domain
     domain_prompt = f"Classify the domain for the following requirement (e.g., Bank, Healthcare, etc.):\n\n{requirement}\n\nDomain:"
     domain = call_groq_api(domain_prompt)
 
+    # Detect defects
     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_groq_api(defects_prompt)
 
-    rewritten = rewrite_requirement_groq(requirement, defects)
+    # Rewrite requirement
+    rewritten = rewrite_requirement(requirement, defects)
 
     return {
         "Requirement": requirement,
@@ -58,10 +59,11 @@ def analyze_requirement_groq(requirement):
     }
 
 # Function to rewrite requirement concisely
-def rewrite_requirement_groq(requirement, defects):
+def rewrite_requirement(requirement, defects):
     if "no defects" in defects.lower():
         return "No modification needed."
 
+    # If defects are found, generate a concise and clear rewritten requirement
     prompt = f"""Rewrite the following requirement to address the defects listed below. Ensure the rewritten requirement is clear, concise, and free of defects. It should be no more than 1-2 sentences.
     
     Original Requirement: {requirement}
@@ -72,27 +74,40 @@ def rewrite_requirement_groq(requirement, defects):
     response = call_groq_api(prompt)
     return response.strip()
 
-# Function to generate a PDF report
-def generate_pdf_report_groq(results):
+# Function to generate a PDF report with professional formatting
+def generate_pdf_report(results):
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", size=12)
 
+    # Add watermark
+    pdf.set_font("Arial", 'B', 50)
+    pdf.set_text_color(230, 230, 230)  # Light gray color for watermark
+    pdf.rotate(45)  # Rotate the text for watermark effect
+    pdf.text(60, 150, "AI Powered Requirement Analysis")
+    pdf.rotate(0)  # Reset rotation
+
+    # Add title and date/time
     pdf.set_font("Arial", 'B', 16)
-    pdf.cell(200, 10, txt="AI Powered Requirement Analysis - Groq Llama", ln=True, align='C')
+    pdf.set_text_color(0, 0, 0)  # Black color for title
+    pdf.cell(200, 10, txt="AI Powered Requirement Analysis and Defect Detection using LLaMA Model", ln=True, align='C')
+    pdf.set_font("Arial", size=12)
     pdf.cell(200, 10, txt=f"Report Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}", ln=True, align='C')
-    pdf.ln(10)
+    pdf.ln(10)  # Add some space
 
+    # Add requirements analysis
     pdf.set_font("Arial", size=12)
     for i, result in enumerate(results, start=1):
-        if pdf.get_y() > 250:
-            pdf.add_page()
+        # Check if we need a new page
+        if pdf.get_y() > 250:  # If the content is near the bottom of the page
+            pdf.add_page()  # Add a new page
             pdf.set_font("Arial", 'B', 16)
-            pdf.cell(200, 10, txt="AI Powered Requirement Analysis - Groq Llama", ln=True, align='C')
+            pdf.cell(200, 10, txt="AI Powered Requirement Analysis and Defect Detection using LLaMA Model", ln=True, align='C')
             pdf.set_font("Arial", size=12)
             pdf.cell(200, 10, txt=f"Report Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}", ln=True, align='C')
-            pdf.ln(10)
+            pdf.ln(10)  # Add some space
 
+        # Add requirement details
         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)
@@ -102,32 +117,36 @@ def generate_pdf_report_groq(results):
         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)
+        pdf.ln(5)  # Add some space between requirements
 
-    pdf_output = "requirements_report_groq.pdf"
+    pdf_output = "requirements_report.pdf"
     pdf.output(pdf_output)
     return pdf_output
 
 # Streamlit app
 def main():
-    st.title("AI Requirement Analysis - Groq Llama")
+    st.title("AI Powered Requirement Analysis and Defect Detection using Large Language Model LLaMA")
     st.markdown("**Team Name:** Sadia, Areeba, Rabbia, Tesmia")
-    st.markdown("**Model:** Groq Llama")
+    st.markdown("**Model:** LLaMA")
 
+    # Input requirements manually
     input_text = st.text_area("Enter your requirements (one per line or separated by periods):")
     requirements = []
     if input_text:
+        # Split by periods or newlines
         requirements = [req.strip() for req in input_text.replace("\n", ".").split(".") if req.strip()]
 
+    # Analyze requirements
     if st.button("Analyze Requirements"):
         if not requirements:
             st.warning("Please enter requirements.")
         else:
             results = []
             for req in requirements:
-                if req.strip():
-                    results.append(analyze_requirement_groq(req.strip()))
+                if req.strip():  # Ignore empty lines
+                    results.append(analyze_requirement(req.strip()))
 
+            # Display results
             st.subheader("Analysis Results")
             for i, result in enumerate(results, start=1):
                 st.write(f"### Requirement R{i}: {result['Requirement']}")
@@ -138,14 +157,16 @@ def main():
                 st.write(f"**Rewritten:** {result['Rewritten']}")
                 st.write("---")
 
-            pdf_report = generate_pdf_report_groq(results)
+            # Generate and download PDF report
+            pdf_report = generate_pdf_report(results)
             with open(pdf_report, "rb") as f:
                 st.download_button(
                     label="Download PDF Report",
                     data=f,
-                    file_name="requirements_report_groq.pdf",
+                    file_name="requirements_report.pdf",
                     mime="application/pdf"
                 )
 
+# Run the app
 if __name__ == "__main__":
-    main()
+    main()