Update app.py

app.py

@@ -5,34 +5,52 @@ import pdfplumber
 # Load the pre-trained question-answering model
 qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
 
-def answer_question(file, question, answer_button):
-    if answer_button:
-        try:
-            # Read and extract text from the uploaded PDF file
-            with pdfplumber.open(file) as pdf:
-                text = ""
-                for page in pdf.pages:
-                    text += page.extract_text()
+# Shared variable to store uploaded PDF text
+pdf_text = ""
 
-            # Ask the user's question using the question-answering model
-            answer = qa_pipeline({"context": text, "question": question})
-            return answer["answer"]
+# Function to load the PDF and store its text
+def load_pdf(file):
+    global pdf_text
+    try:
+        with pdfplumber.open(file) as pdf:
+            pdf_text = ""
+            for page in pdf.pages:
+                pdf_text += page.extract_text()
+        return "PDF loaded successfully."
+    except Exception as e:
+        return f"Error processing PDF: {str(e)}"
 
-        except Exception as e:
-            # Handle exceptions, e.g., invalid PDF or other errors
-            return f"Error processing PDF: {str(e)}"
+# Function to answer the user's question based on the loaded PDF
+def answer_question(question):
+    if not pdf_text:
+        return "No PDF loaded. Upload a PDF first."
+    
+    try:
+        # Ask the user's question using the question-answering model
+        answer = qa_pipeline({"context": pdf_text, "question": question})
+        return answer["answer"]
+    except Exception as e:
+        return f"Error answering question: {str(e)}"
 
-iface = gr.Interface(
+# Interface for uploading the PDF
+pdf_interface = gr.Interface(
+    fn=load_pdf,
+    inputs=gr.File(label="Upload PDF"),
+    outputs="text",
+    live=True,
+    title="PDF Uploader",
+    description="Upload a PDF to load its content.",
+)
+
+# Interface for answering questions based on the loaded PDF
+qa_interface = gr.Interface(
     fn=answer_question,
-    inputs=[
-        gr.File(label="Upload PDF"),
-        gr.Textbox(label="Enter Question", type="text"),
-        gr.Button("Answer"),
-    ],
+    inputs=gr.Textbox(label="Enter Question", type="text"),
     outputs="text",
     live=True,
     title="PDF Question-Answering",
-    description="Upload a PDF, enter a question, and click 'Answer' to get the answer.",
+    description="Enter a question to get an answer based on the loaded PDF.",
 )
 
-iface.launch()
+pdf_interface.launch()
+qa_interface.launch()