Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import requests
 from fpdf import FPDF
+import pdfplumber
 import os
 import time
 
@@ -109,21 +110,33 @@ def main():
 
     # Input options
     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)
+    #input_option = st.radio("Choose input method:", ("Enter Requirements", "Upload PDF"))
+    input_option == "Enter Requirements"
 
     requirements = []
-    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]
+    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}")
 
     # Analyze requirements
     if st.button("Analyze Requirements"):
         if not requirements:
-            st.warning("Please enter at least one requirement.")
+            st.warning("Please enter your requirements.")
         else:
             results = []
             for req in requirements:
@@ -133,13 +146,15 @@ def main():
             # Display results
             st.subheader("Analysis Results")
             for i, result in enumerate(results, start=1):
-                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("---")
+                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("---")
+
+                
 
             # Generate and download PDF report
             pdf_report = generate_pdf_report(results)