Update app.py

app.py

@@ -1,24 +1,41 @@
 import streamlit as st
 from transformers import pipeline
 import fitz  # PyMuPDF
+import tempfile
+import os
 
 # Load the QA model
 qa_model = pipeline("question-answering", "timpal0l/mdeberta-v3-base-squad2")
 
 # Function to extract text from a PDF file
 def extract_text_from_pdf(uploaded_file):
+    temp_file = None
     try:
-        doc = fitz.open(stream=uploaded_file, filetype="pdf")
+        # Save the uploaded PDF as a temporary file
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+        temp_file.write(uploaded_file.read())
+
+        # Open the temporary PDF file and extract text
+        doc = fitz.open(temp_file.name)
         text = ""
         for page_num in range(doc.page_count):
             page = doc[page_num]
             text += page.get_text()
         doc.close()
+
         return text
+
     except Exception as e:
         st.error(f"Error extracting text from PDF: {str(e)}")
         return None
 
+    finally:
+        # Remove the temporary file
+        if temp_file:
+            temp_file.close()
+            # Uncomment the line below if you want to delete the temporary file after use
+            # os.remove(temp_file.name)
+
 # Streamlit app
 def main():
     st.title("PDF Question Answering App")