Update app.py

app.py

@@ -6,11 +6,6 @@ import time
 from datetime import datetime
 import groq
 
-# Load custom CSS
-def local_css(file_name):
-    with open(file_name) as f:
-        st.markdown(f'<style>{f.read()}</style>', unsafe_allow_html=True)
-
 # 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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")
@@ -33,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}"
@@ -48,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="llama-3.3-70b-versatile",
             messages=[
                 {"role": "user", "content": prompt}
             ]
@@ -58,16 +53,14 @@ 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 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:"
     req_type = call_mistral_api(type_prompt).strip()
 
     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()
 
-    # 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()
 
@@ -82,39 +75,35 @@ def analyze_requirement(requirement):
         "Rewritten": rewritten
     }
 
-# Function to generate a PDF report
+# Function to generate PDF report
 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.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)
@@ -123,61 +112,116 @@ 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
+# Streamlit UI
 def main():
-    # Load custom CSS
-    local_css("style.css")
-
-    st.title("AI Powered Requirement Analysis and Defect Detection")
-    st.markdown("**Team Name:** Sadia, Areeba, Rabbia, Tesmia")
-    st.markdown("**Models:** Mistral (Classification & Domain) + Groq (Defects & Rewriting)")
-
-    # Input requirements manually
-    with st.container():
-        st.subheader("Enter Requirements")
-        input_text = st.text_area("Enter your requirements (one per line or separated by periods):", height=200)
-        requirements = []
-        if input_text:
+    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; }
+            .team-card { background: #f8f9fa; padding: 1.5rem; border-radius: 15px; 
+                        margin: 1rem 0; box-shadow: 0 2px 8px rgba(0,0,0,0.1); }
+            .styled-input { border: 2px solid #e0e0e0 !important; border-radius: 10px !important; 
+                          padding: 1rem !important; margin: 1rem 0; }
+            .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.3rem 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); }
+        </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>
+            </div>
+            
+            <div class="team-card">
+                <h3 style="margin: 0 0 1rem;">👥 Team Members</h3>
+                <div style="display: flex; gap: 1rem;">
+                    <div style="flex: 1; text-align: center;">
+                        <div style="background: #4F46E510; padding: 0.5rem; border-radius: 10px;">
+                            <p style="margin: 0; color: #4F46E5;">Sadia</p>
+                        </div>
+                    </div>
+                    <!-- Repeat similar blocks for other team members -->
+                </div>
+            </div>
+    """, unsafe_allow_html=True)
+
+    st.markdown("""
+        <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()]
-
-    # Analyze requirements
-    with st.container():
-        col1, col2 = st.columns([2, 1])
-        with col1:
-            if st.button("Analyze Requirements"):
-                if not requirements:
-                    st.warning("Please enter requirements.")
-                else:
-                    with st.spinner("Analyzing requirements..."):
-                        results = []
-                        for req in requirements:
-                            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']}")
-                            st.write(f"**Type:** {result['Type']}")
-                            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
-                        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.pdf",
-                                mime="application/pdf"
-                            )
-        with col2:
-            st.image
+            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)
+
+if __name__ == "__main__":
+    main()