Update app.py

app.py

@@ -7,73 +7,12 @@ from datetime import datetime
 import groq
 
 # 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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")
+mistral_api_key = os.getenv("MISTRAL_API_KEY", "your_mistral_key_here")
+groq_api_key = os.getenv("GROQ_API_KEY", "your_groq_key_here")
 
 # 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"
@@ -89,13 +28,13 @@ def call_mistral_api(prompt):
     }
     try:
         response = requests.post(url, headers=headers, json=payload)
-        response.raise_for_status()  # Raise an error for bad status codes
+        response.raise_for_status()
         return response.json()['choices'][0]['message']['content']
     except requests.exceptions.HTTPError as err:
-        if response.status_code == 429:  # Rate limit exceeded
+        if response.status_code == 429:
             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
+            time.sleep(5)
+            return call_mistral_api(prompt)
         return f"HTTP Error: {err}"
     except Exception as err:
         return f"Error: {err}"
@@ -104,7 +43,7 @@ def call_mistral_api(prompt):
 def call_groq_api(prompt):
     try:
         response = groq_client.chat.completions.create(
-            model="llama-3.3-70b-versatile",  # Correct model name
+            model="llama3-70b-8192",
             messages=[
                 {"role": "user", "content": prompt}
             ]
@@ -114,7 +53,7 @@ def call_groq_api(prompt):
         st.error(f"Error: {err}")
         return f"Error: {err}"
 
-# Function to analyze a single requirement using both models
+# Function to analyze a single requirement
 def analyze_requirement(requirement):
     # 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:"
@@ -146,31 +85,30 @@ def generate_pdf_report(results):
 
     # 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.set_text_color(230, 230, 230)
+    pdf.rotate(45)
     pdf.text(60, 150, "AI Powered Requirement Analysis")
-    pdf.rotate(0)  # Reset rotation
+    pdf.rotate(0)
 
     # Add title and date/time
     pdf.set_font("Arial", 'B', 16)
-    pdf.set_text_color(0, 0, 0)  # Black color for title
+    pdf.set_text_color(0, 0, 0)
     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
+    pdf.ln(10)
 
     # 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
+        if pdf.get_y() > 250:
+            pdf.add_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
+            pdf.ln(10)
 
-        # 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)
@@ -179,128 +117,167 @@ def generate_pdf_report(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)  # Add some space between requirements
+        pdf.ln(5)
 
     pdf_output = "requirements_report.pdf"
     pdf.output(pdf_output)
     return pdf_output
 
-# Streamlit app with enhanced UI
+# Custom CSS for professional styling
+st.markdown("""
+    <style>
+        .main {
+            background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
+        }
+        .stApp {
+            max-width: 1200px;
+            margin: 0 auto;
+            padding: 2rem;
+        }
+        .header {
+            text-align: center;
+            padding: 2rem;
+            background: white;
+            border-radius: 15px;
+            box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+            margin-bottom: 2rem;
+        }
+        .requirement-card {
+            background: white;
+            padding: 1.5rem;
+            border-radius: 10px;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05);
+            margin-bottom: 1.5rem;
+            transition: transform 0.2s;
+        }
+        .requirement-card:hover {
+            transform: translateY(-3px);
+        }
+        .analysis-badge {
+            padding: 0.5rem 1rem;
+            border-radius: 20px;
+            font-weight: 500;
+            margin: 0.3rem;
+        }
+        .download-btn {
+            width: 100%;
+            padding: 1rem;
+            font-size: 1.1rem;
+            margin-top: 2rem;
+        }
+        .type-functional { background: #e8f5e9; color: #2e7d32; }
+        .type-non-functional { background: #fff3e0; color: #ef6c00; }
+        .defect-badge { background: #ffebee; color: #c62828; }
+        .improved-badge { background: #e3f2fd; color: #1565c0; }
+    </style>
+""", unsafe_allow_html=True)
+
 def main():
-    # 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>
-        """, unsafe_allow_html=True)
+    # Professional Header Section
+    with st.container():
+        st.markdown('<div class="header">', unsafe_allow_html=True)
+        st.title("📝 AI-Powered Requirement Analysis System")
         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 style="margin: 1.5rem 0;">
+                <div style="font-size: 1.2rem; color: #4a4a4a;">Automated Requirement Classification, Defect Detection & Optimization</div>
+                <div style="display: flex; justify-content: center; gap: 1rem; margin: 1rem 0;">
+                    <div style="padding: 0.5rem 1rem; background: #f0f2f6; border-radius: 20px;">Team: Sadia, Areeba, Rabbia, Tesmia</div>
+                    <div style="padding: 0.5rem 1rem; background: #f0f2f6; border-radius: 20px;">Mistral + Groq API</div>
+                </div>
             </div>
         """, unsafe_allow_html=True)
+        st.markdown('</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
+    # Input Section
+    with st.container():
+        st.subheader("📥 Input Requirements")
+        input_text = st.text_area(
+            "Enter your software requirements (one per line or separated by periods):",
+            height=200,
+            placeholder="Example:\nThe system shall allow users to reset their password via email verification.\nThe interface must load within 2 seconds of user interaction...",
+            help="Enter each requirement on a new line or separate with periods."
+        )
+
+    # Analysis Section
+    if st.button("🚀 Start Analysis", type="primary", use_container_width=True):
+        if not input_text.strip():
+            st.warning("⚠️ Please enter requirements to analyze")
+        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))
+                    progress_bar.progress((i+1)/len(requirements))
+                
+                st.success("✅ Analysis Completed!")
+                time.sleep(0.5)
+                st.session_state.results = results
+
+    # Display Results
+    if 'results' in st.session_state:
+        st.subheader("📊 Analysis Results")
         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"
-            )
-            
-        # 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>
+            for i, result in enumerate(st.session_state.results, 1):
+                with st.expander(f"Requirement #{i}: {result['Requirement'][:50]}...", expanded=True):
+                    st.markdown(f"""
+                        <div class="requirement-card">
+                            <div style="margin-bottom: 1.5rem;">
+                                <div class="analysis-badge { 'type-functional' if 'functional' in result['Type'].lower() else 'type-non-functional' }">
+                                    📌 Type: {result['Type']}
+                                </div>
+                                <div class="analysis-badge" style="background: #e8eaf6; color: #303f9f;">
+                                    🏷️ Domain: {result['Domain']}
+                                </div>
                             </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 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}</div>' 
+                                            for d in result['Defects'].split(', ')])}
+                                </div>
                             </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 style="margin: 1rem 0;">
+                                <h4>✨ Improved Version</h4>
+                                <div class="analysis-badge improved-badge">
+                                    📝 {result['Rewritten']}
                                 </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"
-                        )
+                            </div>
+                        </div>
+                    """, unsafe_allow_html=True)
+
+        # PDF Report Section
+        st.subheader("📤 Generate Report")
+        with st.container():
+            col1, col2 = st.columns([3, 2])
+            with col1:
+                st.info("💡 Click below to generate a comprehensive PDF report with all analysis details")
+            with col2:
+                if st.button("📄 Generate PDF Report", type="secondary", use_container_width=True):
+                    with st.spinner("Generating PDF..."):
+                        pdf_report = generate_pdf_report(st.session_state.results)
+                        with open(pdf_report, "rb") as f:
+                            st.download_button(
+                                label="⬇️ Download Full Report",
+                                data=f,
+                                file_name="Requirement_Analysis_Report.pdf",
+                                mime="application/pdf",
+                                use_container_width=True,
+                                type="primary"
+                            )
+
+    # Footer
+    st.markdown("---")
+    st.markdown("""
+        <div style="text-align: center; color: #666; margin-top: 3rem;">
+            <p>AI-Powered Requirement Analysis System • Final Year Project • Computer Science Department</p>
+            <p>🚀 Powered by Mistral AI & Groq • 🛠️ Developed by Team Four</p>
+        </div>
+    """, unsafe_allow_html=True)
 
 if __name__ == "__main__":
     main()