app.py

import streamlit as st
import requests
from fpdf import FPDF
import os
import time
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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")

# Initialize Groq client
groq_client = groq.Client(api_key=groq_api_key)

# Function to call Mistral API
def call_mistral_api(prompt):
    url = "https://api.mistral.ai/v1/chat/completions"
    headers = {
        "Authorization": f"Bearer {mistral_api_key}",
        "Content-Type": "application/json"
    }
    payload = {
        "model": "mistral-medium",
        "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:
        if response.status_code == 429:
            st.warning("Rate limit exceeded. Please wait a few seconds and try again.")
            time.sleep(5)
            return call_mistral_api(prompt)
        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}
            ]
        )
        return response.choices[0].message.content
    except Exception as err:
        st.error(f"Error: {err}")
        return f"Error: {err}"

# Function to analyze requirement
def analyze_requirement(requirement):
    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()

    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 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,
        "Domain": domain,
        "Defects": defects,
        "Rewritten": rewritten
    }

# Function to generate PDF report
def generate_pdf_report(results):
    pdf = FPDF()
    pdf.add_page()
    pdf.set_font("Arial", size=12)

    pdf.set_font("Arial", 'B', 50)
    pdf.set_text_color(230, 230, 230)
    pdf.rotate(45)
    pdf.text(60, 150, "AI Powered Requirement Analysis")
    pdf.rotate(0)

    pdf.set_font("Arial", 'B', 16)
    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)

    pdf.set_font("Arial", size=12)
    for i, result in enumerate(results, start=1):
        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)

        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)

    pdf_output = "requirements_report.pdf"
    pdf.output(pdf_output)
    return pdf_output

# Streamlit 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; }
            .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()]
            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()