Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 import requests
 from fpdf import FPDF
-import pdfplumber
 import os
 import time
 
@@ -105,36 +104,26 @@ def generate_pdf_report(results):
 # Streamlit app
 def main():
     st.title("Requirement Analysis Tool")
-    st.markdown("**Team Name:** Team Innovators")
-    st.markdown("**Model:** Mistral")
+    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"))
+    st.markdown("### Enter Requirements")
+    st.markdown("You can enter up to **5 requirements** at a time. Each requirement should be **clear and concise**.")
+    input_text = st.text_area("Enter your requirements (one per line or separated by periods):", height=150)
 
     requirements = []
-    if input_option == "Enter Requirements":
-        input_text = st.text_area("Enter your requirements (one per line or separated by periods):")
-        if input_text:
-            # Split by periods or newlines
-            requirements = [req.strip() for req in input_text.replace("\n", ".").split(".") if req.strip()]
-    else:
-        uploaded_file = st.file_uploader("Upload a PDF file:", type="pdf")
-        if uploaded_file:
-            try:
-                with pdfplumber.open(uploaded_file) as pdf:
-                    requirements = []
-                    for page in pdf.pages:
-                        text = page.extract_text()
-                        if text:
-                            # Split by periods or newlines
-                            requirements.extend([req.strip() for req in text.replace("\n", ".").split(".") if req.strip()])
-            except Exception as e:
-                st.error(f"Error reading PDF file: {e}")
+    if input_text:
+        # Split by periods or newlines
+        requirements = [req.strip() for req in input_text.replace("\n", ".").split(".") if req.strip()]
+        if len(requirements) > 5:
+            st.warning("You have entered more than 5 requirements. Only the first 5 will be analyzed.")
+            requirements = requirements[:5]
 
     # Analyze requirements
     if st.button("Analyze Requirements"):
         if not requirements:
-            st.warning("Please enter or upload requirements.")
+            st.warning("Please enter at least one requirement.")
         else:
             results = []
             for req in requirements:
@@ -144,13 +133,13 @@ def main():
             # 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"**Stakeholders:** {result['Stakeholders']}")
-                st.write(f"**Domain:** {result['Domain']}")
-                st.write(f"**Defects:** {result['Defects']}")
-                st.write(f"**Rewritten:** {result['Rewritten']}")
-                st.write("---")
+                st.markdown(f"### Requirement R{i}: {result['Requirement']}")
+                st.markdown(f"**Type:** {result['Type']}")
+                st.markdown(f"**Stakeholders:** {result['Stakeholders']}")
+                st.markdown(f"**Domain:** {result['Domain']}")
+                st.markdown(f"**Defects:** {result['Defects']}")
+                st.markdown(f"**Rewritten:** {result['Rewritten']}")
+                st.markdown("---")
 
             # Generate and download PDF report
             pdf_report = generate_pdf_report(results)