Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+import streamlit as st
 import requests
 from fpdf import FPDF
 import pdfplumber
@@ -56,28 +56,6 @@ def analyze_requirement(requirement):
         "Rewritten": rewritten
     }
 
-# Function to process multiple requirements
-def process_requirements(input_text, uploaded_file):
-    requirements = []
-
-    # If a file is uploaded, extract text from the PDF
-    if uploaded_file:
-        with pdfplumber.open(uploaded_file) as pdf:
-            for page in pdf.pages:
-                requirements.extend(page.extract_text().split("\n"))
-
-    # If text input is provided, split into individual requirements
-    if input_text:
-        requirements.extend(input_text.split("\n"))
-
-    # Analyze each requirement
-    results = []
-    for req in requirements:
-        if req.strip():  # Ignore empty lines
-            results.append(analyze_requirement(req.strip()))
-
-    return results
-
 # Function to generate a PDF report
 def generate_pdf_report(results):
     pdf = FPDF()
@@ -97,46 +75,58 @@ def generate_pdf_report(results):
     pdf.output(pdf_output)
     return pdf_output
 
-# Gradio interface
-def app_interface(input_text, uploaded_file):
-    results = process_requirements(input_text, uploaded_file)
-    pdf_report = generate_pdf_report(results)
+# Streamlit app
+def main():
+    st.title("Requirement Defects Detective Tool")
+    st.markdown("**Team Name:** Sadia, Areeba, Rabbia, Tesmia")
+    st.markdown("**LLM Model:** Mistral")
 
-    # Display results in a clean format
-    output_text = ""
-    for result in results:
-        output_text += f"Requirement: {result['Requirement']}\n"
-        output_text += f"Type: {result['Type']}\n"
-        output_text += f"Stakeholders: {result['Stakeholders']}\n"
-        output_text += f"Domain: {result['Domain']}\n"
-        output_text += f"Defects: {result['Defects']}\n"
-        output_text += f"Rewritten: {result['Rewritten']}\n"
-        output_text += "-" * 50 + "\n\n"
-
-    return output_text, pdf_report
-
-# Gradio app
-title = "Requirement Analysis App"
-description = """
-This app analyzes software requirements using the Mistral model. 
-You can either enter requirements manually or upload a PDF file containing requirements.
-"""
-team_name = "Team Innovators"
-
-interface = gr.Interface(
-    fn=app_interface,
-    inputs=[
-        gr.Textbox(label="Enter Requirements (one per line)"),
-        gr.File(label="Upload PDF File (optional)")
-    ],
-    outputs=[
-        gr.Textbox(label="Analysis Results"),
-        gr.File(label="Download PDF Report")
-    ],
-    title=title,
-    description=description,
-    theme="soft"
-)
-
-# Add team name to the interface
-interface.launch()
+    # 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()