Update app.py

app.py

@@ -6,13 +6,75 @@ import time
 from datetime import datetime
 import groq
 
-# API keys (replace with your actual keys)
-mistral_api_key = os.getenv("MISTRAL_API_KEY", "your_mistral_key_here")
-groq_api_key = os.getenv("GROQ_API_KEY", "your_groq_key_here")
+# API keys (replace with your keys or use environment variables)
+mistral_api_key = os.getenv("MISTRAL_API_KEY", "gz6lDXokxgR6cLY72oomALWcm7vhjRzQ")
+groq_api_key = os.getenv("GROQ_API_KEY", "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")
 
 # Initialize Groq client
 groq_client = groq.Client(api_key=groq_api_key)
 
+# Custom CSS for professional styling
+st.markdown("""
+    <style>
+        .main {
+            background-color: #f8f9fa;
+        }
+        .stTextArea textarea {
+            border: 2px solid #4a4e69;
+            border-radius: 10px;
+            padding: 15px !important;
+        }
+        .stButton button {
+            background-color: #4a4e69;
+            color: white;
+            border: none;
+            padding: 12px 30px;
+            border-radius: 8px;
+            font-size: 16px;
+            font-weight: bold;
+            transition: all 0.3s ease;
+            width: 100%;
+        }
+        .stButton button:hover {
+            background-color: #3a3e59;
+            transform: scale(1.02);
+        }
+        .report-title {
+            color: #4a4e69;
+            font-size: 2.5em;
+            text-align: center;
+            padding: 20px;
+            border-bottom: 3px solid #4a4e69;
+            margin-bottom: 30px;
+        }
+        .requirement-card {
+            background: white;
+            border-radius: 15px;
+            padding: 25px;
+            margin: 15px 0;
+            box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+            border-left: 5px solid #4a4e69;
+        }
+        .sidebar .sidebar-content {
+            background-color: #4a4e69;
+            color: white;
+        }
+        .metric-box {
+            background: white;
+            padding: 15px;
+            border-radius: 10px;
+            margin: 10px 0;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05);
+        }
+        .stSpinner > div {
+            display: flex;
+            justify-content: center;
+            align-items: center;
+        }
+    </style>
+""", unsafe_allow_html=True)
+
+# Function to call Mistral API
 def call_mistral_api(prompt):
     url = "https://api.mistral.ai/v1/chat/completions"
     headers = {
@@ -21,38 +83,53 @@ def call_mistral_api(prompt):
     }
     payload = {
         "model": "mistral-medium",
-        "messages": [{"role": "user", "content": prompt}]
+        "messages": [
+            {"role": "user", "content": prompt}
+        ]
     }
     try:
         response = requests.post(url, headers=headers, json=payload)
-        response.raise_for_status()
+        response.raise_for_status()  # Raise an error for bad status codes
         return response.json()['choices'][0]['message']['content']
+    except requests.exceptions.HTTPError as err:
+        if response.status_code == 429:  # Rate limit exceeded
+            st.warning("Rate limit exceeded. Please wait a few seconds and try again.")
+            time.sleep(5)  # Wait for 5 seconds before retrying
+            return call_mistral_api(prompt)  # Retry the request
+        return f"HTTP Error: {err}"
     except Exception as err:
         return f"Error: {err}"
 
+# Function to call Groq API
 def call_groq_api(prompt):
     try:
         response = groq_client.chat.completions.create(
-            model="llama-3.3-70b-versatile",
-            messages=[{"role": "user", "content": prompt}]
+            model="llama-3.3-70b-versatile",  # Correct model name
+            messages=[
+                {"role": "user", "content": prompt}
+            ]
         )
         return response.choices[0].message.content
     except Exception as err:
+        st.error(f"Error: {err}")
         return f"Error: {err}"
 
+# Function to analyze a single requirement using both models
 def analyze_requirement(requirement):
-    type_prompt = f"Classify as Functional or Non-Functional in one word:\n{requirement}\nType:"
+    # Use Mistral for classification and domain identification
+    type_prompt = f"Classify the following requirement as Functional or Non-Functional in one word:\n\n{requirement}\n\nType:"
     req_type = call_mistral_api(type_prompt).strip()
-    
-    domain_prompt = f"Classify domain in one word:\n{requirement}\nDomain:"
+
+    domain_prompt = f"Classify the domain for the following requirement in one word (e.g., E-commerce, Education, etc.):\n\n{requirement}\n\nDomain:"
     domain = call_mistral_api(domain_prompt).strip()
-    
-    defects_prompt = f"List major defects (1-2 words each):\n{requirement}\nDefects:"
+
+    # 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 = call_groq_api(defects_prompt).strip()
-    
-    rewritten_prompt = f"Rewrite to address defects:\n{requirement}\nRewritten:"
+
+    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,
         "Type": req_type,
@@ -61,148 +138,169 @@ def analyze_requirement(requirement):
         "Rewritten": rewritten
     }
 
+# Function to generate a PDF report
 def generate_pdf_report(results):
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", size=12)
-    
-    # PDF content (same as previous)
-    # ... [Keep the existing PDF generation code] ...
-    
+
+    # 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.set_text_color(0, 0, 0)  # Black color for title
+    pdf.cell(200, 10, txt="AI Powered Requirement Analysis and Defect Detection", 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)  # 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:  # 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 and Defect Detection", 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)  # 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)
+        pdf.multi_cell(200, 10, txt=f"Type: {result['Type']}", 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
+
     pdf_output = "requirements_report.pdf"
     pdf.output(pdf_output)
     return pdf_output
 
+# Streamlit app with enhanced UI
 def main():
-    st.markdown("""
-        <style>
-            @import url('https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;600&display=swap');
-            * { font-family: 'Poppins', sans-serif; }
-            .main-container { max-width: 800px; margin: 0 auto; padding: 2rem; }
-            .gradient-header { 
-                background: linear-gradient(45deg, #4F46E5, #1E40AF);
-                color: white; 
-                padding: 2rem; 
-                border-radius: 15px;
-                margin-bottom: 2rem;
-            }
-            .team-card {
-                background: #f8f9fa;
-                padding: 1.5rem;
-                border-radius: 15px;
-                margin: 1rem 0;
-                box-shadow: 0 2px 8px rgba(0,0,0,0.1);
-            }
-            .result-card {
-                background: white;
-                padding: 1.5rem;
-                border-radius: 15px;
-                margin: 1rem 0;
-                box-shadow: 0 2px 8px rgba(0,0,0,0.1);
-            }
-            .badge {
-                padding: 0.5rem 1rem;
-                border-radius: 20px;
-                font-size: 0.9rem;
-                margin-right: 0.5rem;
-                display: inline-block;
-            }
-            .type-badge { background: #4F46E510; color: #4F46E5; }
-            .domain-badge { background: #10B98110; color: #10B981; }
-            .defect-badge { background: #EF444410; color: #EF4444; }
-            .animate-hover { transition: transform 0.2s; }
-            .animate-hover:hover { transform: translateY(-2px); }
-            .styled-textarea {
-                border: 2px solid #e0e0e0 !important;
-                border-radius: 10px !important;
-                padding: 1rem !important;
-            }
-        </style>
-    """, unsafe_allow_html=True)
-
-    st.markdown("""
-        <div class="main-container">
-            <div class="gradient-header">
-                <h1 style="margin: 0; font-weight: 600;">🧠 AI Requirement Analyzer</h1>
-                <p style="margin: 0.5rem 0 0; opacity: 0.9;">Smart Analysis & Quality Enhancement System</p>
+    # Sidebar with team info
+    with st.sidebar:
+        st.markdown("<h1 style='color: white; text-align: center;'>Team Details</h1>", unsafe_allow_html=True)
+        st.markdown("""
+            <div style='background: rgba(255,255,255,0.1); padding: 15px; border-radius: 10px; margin: 15px 0;'>
+                <h3 style='color: white; margin-bottom: 10px;'>👩💻 Team Members</h3>
+                <p style='color: white;'>Sadia<br>Areeba<br>Rabbia<br>Tesmia</p>
             </div>
-            
-            <div class="team-card">
-                <h3 style="margin: 0 0 1rem;">👥 Team Members</h3>
-                <div style="display: grid; grid-template-columns: repeat(2, 1fr); gap: 1rem;">
-                    <div style="background: #4F46E510; padding: 0.5rem; border-radius: 10px;">
-                        <p style="margin: 0; color: #4F46E5;">Sadia</p>
-                    </div>
-                    <div style="background: #10B98110; padding: 0.5rem; border-radius: 10px;">
-                        <p style="margin: 0; color: #10B981;">Areeba</p>
-                    </div>
-                    <div style="background: #EF444410; padding: 0.5rem; border-radius: 10px;">
-                        <p style="margin: 0; color: #EF4444;">Rabbia</p>
-                    </div>
-                    <div style="background: #3B82F610; padding: 0.5rem; border-radius: 10px;">
-                        <p style="margin: 0; color: #3B82F6;">Tesmia</p>
-                    </div>
-                </div>
+        """, unsafe_allow_html=True)
+        st.markdown("""
+            <div style='background: rgba(255,255,255,0.1); padding: 15px; border-radius: 10px;'>
+                <h3 style='color: white; margin-bottom: 10px;'>🤖 AI Models</h3>
+                <p style='color: white;'>Mistral (Classification)<br>Groq (Defect Analysis)</p>
             </div>
+        """, unsafe_allow_html=True)
+
+    # Main content area
+    col1, col2, col3 = st.columns([1,6,1])
+    with col2:
+        st.markdown("<h1 class='report-title'>AI Requirement Analyzer</h1>", unsafe_allow_html=True)
+        
+        # Input section
+        with st.container():
+            st.markdown("### 📝 Enter Requirements")
+            input_text = st.text_area(
+                "Enter your requirements (one per line or separated by periods):",
+                height=200,
+                label_visibility="collapsed"
+            )
             
-            <div style="margin: 2rem 0;">
-                <h2>📥 Input Requirements</h2>
-                <p style="opacity: 0.8;">Enter multiple requirements separated by new lines or periods</p>
-    """, unsafe_allow_html=True)
-
-    input_text = st.text_area("", height=200, key="input_area", 
-                            help="Example:\n1. The system must handle 1000 users\n2. User interface should be responsive")
-
-    if st.button("🚀 Start Analysis", key="analyze_btn", use_container_width=True):
-        if not input_text.strip():
-            st.warning("⚠️ Please enter requirements to analyze")
-        else:
-            requirements = [req.strip() for req in input_text.replace("\n", ".").split(".") if req.strip()]
-            results = []
-            progress_bar = st.progress(0)
-            
-            with st.spinner("🔍 Analyzing requirements..."):
-                for i, req in enumerate(requirements):
-                    results.append(analyze_requirement(req))
-                    progress_bar.progress((i+1)/len(requirements))
-                    time.sleep(0.1)
-
-            st.success("✅ Analysis completed!")
-            st.markdown("---")
-            
-            for i, result in enumerate(results, 1):
-                st.markdown(f"""
-                    <div class="result-card animate-hover">
-                        <h3 style="margin: 0 0 1rem;">🔖 Requirement R{i}</h3>
-                        <p style="font-size: 1.1rem; margin-bottom: 1.5rem;">{result['Requirement']}</p>
-                        
-                        <div style="display: flex; gap: 1rem; margin-bottom: 1rem;">
-                            <span class="badge type-badge">📝 {result['Type']}</span>
-                            <span class="badge domain-badge">🌍 {result['Domain']}</span>
-                            <span class="badge defect-badge">⚠️ {result['Defects']}</span>
-                        </div>
-                        
-                        <div style="background: #f8f9fa; padding: 1rem; border-radius: 10px;">
-                            <p style="margin: 0; color: #3B82F6;">✏️ Optimized Version:</p>
-                            <p style="margin: 0.5rem 0 0; font-weight: 500;">{result['Rewritten']}</p>
-                        </div>
-                    </div>
-                """, unsafe_allow_html=True)
-
-            st.markdown("---")
-            st.markdown("### 📄 Generate 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_analysis.pdf",
-                    mime="application/pdf",
-                    use_container_width=True,
-                    key="download_btn"
-                )
-
-    st.markdown("</div>", unsafe_allow_html=True)
+        # Analysis button
+        if st.button("🔍 Analyze Requirements", key="analyze_btn"):
+            if not input_text.strip():
+                st.warning("⚠️ Please enter requirements before analyzing")
+            else:
+                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.strip()))
+                        progress_bar.progress((i+1)/len(requirements))
+                    
+                    # Display results
+                    st.success("✅ Analysis Complete!")
+                    st.markdown("---")
+                    st.markdown("## 📊 Analysis Results")
+                    
+                    # Summary metrics
+                    col1, col2, col3 = st.columns(3)
+                    with col1:
+                        st.markdown(f"""
+                            <div class='metric-box'>
+                                <h3>📄 Total Requirements</h3>
+                                <h2>{len(results)}</h2>
+                            </div>
+                        """, unsafe_allow_html=True)
+                    with col2:
+                        types = [res['Type'] for res in results]
+                        st.markdown(f"""
+                            <div class='metric-box'>
+                                <h3>📊 Functional/Non-Functional</h3>
+                                <h2>{types.count('Functional')}/{types.count('Non-Functional')}</h2>
+                            </div>
+                        """, unsafe_allow_html=True)
+                    with col3:
+                        domains = len(set([res['Domain'] for res in results]))
+                        st.markdown(f"""
+                            <div class='metric-box'>
+                                <h3>🌍 Unique Domains</h3>
+                                <h2>{domains}</h2>
+                            </div>
+                        """, unsafe_allow_html=True)
+                    
+                    # Detailed results
+                    st.markdown("---")
+                    for i, result in enumerate(results, start=1):
+                        with st.container():
+                            st.markdown(f"""
+                                <div class='requirement-card'>
+                                    <div style="display: flex; justify-content: space-between; align-items: center; margin-bottom: 15px;">
+                                        <h3>🔖 Requirement R{i}</h3>
+                                        <span style="font-size: 0.9em; color: #666;">{result['Domain']} Domain</span>
+                                    </div>
+                                    <p style="color: #444; margin-bottom: 15px;">{result['Requirement']}</p>
+                                    <div style="display: grid; grid-template-columns: repeat(3, 1fr); gap: 15px;">
+                                        <div style="background: #f8f9fa; padding: 10px; border-radius: 8px;">
+                                            <h4>📌 Type</h4>
+                                            <p>{result['Type']}</p>
+                                        </div>
+                                        <div style="background: #f8f9fa; padding: 10px; border-radius: 8px;">
+                                            <h4>⚠️ Defects</h4>
+                                            <p>{result['Defects']}</p>
+                                        </div>
+                                        <div style="background: #f8f9fa; padding: 10px; border-radius: 8px;">
+                                            <h4>✏️ Improved Version</h4>
+                                            <p>{result['Rewritten']}</p>
+                                        </div>
+                                    </div>
+                                </div>
+                            """, unsafe_allow_html=True)
+                    
+                    # PDF Download
+                    st.markdown("---")
+                    st.markdown("## 📤 Export Results")
+                    pdf_report = generate_pdf_report(results)
+                    with open(pdf_report, "rb") as f:
+                        st.download_button(
+                            label="📥 Download Full Report (PDF)",
+                            data=f,
+                            file_name="requirements_report.pdf",
+                            mime="application/pdf",
+                            help="Download comprehensive PDF report with all analysis details"
+                        )
 
 if __name__ == "__main__":
     main()