Update app.py

app.py

@@ -7,25 +7,31 @@ from datetime import datetime
 
 # Streamlit UI
 st.title("Generate Meal Request PDF")
-st.write("Upload a template PDF and provide an input file containing data for the meal request.")
+st.write("Upload two PDF files and provide a Meal Request template.")
 
-# Input Fields
-input_file = st.file_uploader("Upload the Input File (PDF or Text)", type=["pdf", "txt"])
+# Input Fields: Allow uploading multiple PDFs
+uploaded_files = st.file_uploader("Upload Two PDF Files", type=["pdf"], accept_multiple_files=True)
+
+# Template PDF: Upload template PDF
 template_file = st.file_uploader("Upload the Meal Request Template PDF", type=["pdf"])
 
-# Function to extract data from the uploaded input file
-def extract_data_from_file(file):
-    if file.type == "application/pdf":
-        # Extract text from PDF
-        reader = PdfReader(file)
-        text = " ".join([page.extract_text() for page in reader.pages if page.extract_text()])
-        return text
-    elif file.type == "text/plain":
-        # If the file is a text file, just read the content
-        return file.read().decode("utf-8")
-    else:
-        st.warning("Unsupported file format. Please upload a PDF or text file.")
-        return None
+# Ensure two PDFs are uploaded
+if len(uploaded_files) != 2:
+    st.error("Please upload exactly two PDF files.")
+else:
+    st.success("Two PDF files uploaded successfully!")
+
+# Function to extract data from the uploaded input files
+def extract_data_from_files(files):
+    extracted_text = []
+    
+    for file in files:
+        if file.type == "application/pdf":
+            reader = PdfReader(file)
+            text = " ".join([page.extract_text() for page in reader.pages if page.extract_text()])
+            extracted_text.append(text)
+    
+    return extracted_text
 
 # Function to populate the template with extracted data
 def generate_meal_request_pdf(template_pdf, data):
@@ -66,18 +72,17 @@ def generate_meal_request_pdf(template_pdf, data):
 
 # Process when the button is pressed
 if st.button("Generate Meal Request PDF"):
-    if not input_file or not template_file:
-        st.error("Please provide both the input file and the template PDF.")
+    if not template_file:
+        st.error("Please provide the template PDF.")
     else:
-        # Step 1: Extract data from the uploaded input file
-        data_from_file = extract_data_from_file(input_file)
+        # Step 1: Extract data from the uploaded files
+        data_from_files = extract_data_from_files(uploaded_files)
         
-        if data_from_file:
-            # Simulating data extraction from the uploaded file
-            # For demonstration purposes, assuming extracted data is text with specific info
+        if data_from_files:
+            # Simulating extraction of relevant data (replace this with actual logic)
             extracted_data = {
-                "name": "Alice",  # Normally, extract from the input text
-                "meal_type": "Vegan",  # Normally, extract from the input text
+                "name": "Alice",  # Normally extracted from the PDF
+                "meal_type": "Vegan",  # Normally extracted from the PDF
                 "date": datetime.today().strftime("%B %d, %Y"),
             }