Spaces:
Sleeping
Sleeping
File size: 4,959 Bytes
cbe7195 fc8383a ce64498 fc8383a ce64498 fc8383a ce64498 fc8383a cbe7195 fc8383a cbe7195 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 |
import streamlit as st
import requests
from fpdf import FPDF
import pdfplumber
# Mistral API key (replace with your key)
API_KEY = "gz6lDXokxgR6cLY72oomALWcm7vhjzQ"
MISTRAL_API_URL = "https://api.mistral.ai/v1/chat/completions"
# Function to call Mistral API
def call_mistral_api(prompt):
headers = {
"Authorization": f"Bearer {API_KEY}",
"Content-Type": "application/json"
}
payload = {
"model": "mistral-medium",
"messages": [
{"role": "user", "content": prompt}
]
}
response = requests.post(MISTRAL_API_URL, headers=headers, json=payload)
if response.status_code == 200:
return response.json()['choices'][0]['message']['content']
else:
return f"Error: {response.status_code}, {response.text}"
# Function to analyze a single requirement
def analyze_requirement(requirement):
# Detect requirement type
type_prompt = f"Classify the following requirement as Functional or Non-Functional:\n\n{requirement}\n\nType:"
req_type = call_mistral_api(type_prompt)
# Identify stakeholders
stakeholders_prompt = f"Identify the stakeholders for the following requirement:\n\n{requirement}\n\nStakeholders:"
stakeholders = call_mistral_api(stakeholders_prompt)
# Classify domain
domain_prompt = f"Classify the domain for the following requirement (e.g., Bank, Healthcare, etc.):\n\n{requirement}\n\nDomain:"
domain = call_mistral_api(domain_prompt)
# Detect defects
defects_prompt = f"Detect any defects in the following requirement (e.g., Ambiguity, Incompleteness, etc.):\n\n{requirement}\n\nDefects:"
defects = call_mistral_api(defects_prompt)
# Rewrite requirement
rewrite_prompt = f"Rewrite the following requirement in a simpler form:\n\n{requirement}\n\nSimplified Requirement:"
rewritten = call_mistral_api(rewrite_prompt)
return {
"Requirement": requirement,
"Type": req_type,
"Stakeholders": stakeholders,
"Domain": domain,
"Defects": defects,
"Rewritten": rewritten
}
# Function to generate a PDF report
def generate_pdf_report(results):
pdf = FPDF()
pdf.add_page()
pdf.set_font("Arial", size=12)
for result in results:
pdf.cell(200, 10, txt=f"Requirement: {result['Requirement']}", ln=True)
pdf.cell(200, 10, txt=f"Type: {result['Type']}", ln=True)
pdf.cell(200, 10, txt=f"Stakeholders: {result['Stakeholders']}", ln=True)
pdf.cell(200, 10, txt=f"Domain: {result['Domain']}", ln=True)
pdf.cell(200, 10, txt=f"Defects: {result['Defects']}", ln=True)
pdf.cell(200, 10, txt=f"Rewritten: {result['Rewritten']}", ln=True)
pdf.cell(200, 10, txt="-" * 50, ln=True)
pdf_output = "requirements_report.pdf"
pdf.output(pdf_output)
return pdf_output
# Streamlit app
def main():
st.title("Requirement Defects Detective Tool")
st.markdown("**Team Name:** Sadia, Areeba, Rabbia, Tesmia")
st.markdown("**LLM Model:** Mistral")
# Input options
input_option = st.radio("Choose input method:", ("Enter Requirements", "Upload PDF"))
requirements = []
if input_option == "Enter Requirements":
input_text = st.text_area("Enter your requirements (one per line):")
if input_text:
requirements = input_text.split("\n")
else:
uploaded_file = st.file_uploader("Upload a PDF file:", type="pdf")
if uploaded_file:
with pdfplumber.open(uploaded_file) as pdf:
for page in pdf.pages:
requirements.extend(page.extract_text().split("\n"))
# Analyze requirements
if st.button("Analyze Requirements"):
if not requirements:
st.warning("Please enter or upload requirements.")
else:
results = []
for req in requirements:
if req.strip(): # Ignore empty lines
results.append(analyze_requirement(req.strip()))
# Display results
st.subheader("Analysis Results")
for result in results:
st.write(f"**Requirement:** {result['Requirement']}")
st.write(f"**Type:** {result['Type']}")
st.write(f"**Stakeholders:** {result['Stakeholders']}")
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"
)
# Run the app
if __name__ == "__main__":
main() |