Spaces:
Sleeping
Sleeping
| 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() |