Update app.py

app.py

@@ -5,32 +5,47 @@ import pdfplumber
 # Load the pre-trained question-answering model
 qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
 
-def answer_question(file, ques: str):
-    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()
+class PDFQuestionAnswering:
+    def __init__(self):
+        self.text = ""
 
-        # Ask a default question
-        question = ques
+    def load_pdf(self, file):
+        try:
+            # Read and extract text from the uploaded PDF file
+            with pdfplumber.open(file) as pdf:
+                self.text = ""
+                for page in pdf.pages:
+                    self.text += page.extract_text()
 
-        # Ask the question using the question-answering model
-        answer = qa_pipeline({"context": text, "question": question})
+            return "PDF loaded successfully."
+
+        except Exception as e:
+            # Handle exceptions, e.g., invalid PDF or other errors
+            return f"Error processing PDF: {str(e)}"
 
+    def answer_question(self, question):
+        if not self.text:
+            return "No PDF loaded. Upload a PDF first."
+
+        # Ask the question using the question-answering model
+        answer = qa_pipeline({"context": self.text, "question": question})
         return answer["answer"]
 
-    except Exception as e:
-        return f"Error processing PDF: {str(e)}"
+pdf_qa = PDFQuestionAnswering()
 
 iface = gr.Interface(
-    fn=answer_question,
-    inputs=gr.File(label="Upload PDF"),
+    fn_map={
+        "Load PDF": pdf_qa.load_pdf,
+        "Answer Question": pdf_qa.answer_question,
+    },
+    inputs=[
+        gr.File(label="Upload PDF", type="file"),
+        gr.Textbox(label="Enter Question", type="text"),
+    ],
     outputs="text",
     live=True,
-    title="PDF Documents Question-Answering",
-    description="Ask a question about the contents of the uploaded PDF file.",
+    title="PDF Question-Answering",
+    description="Upload a PDF, load it, and ask questions about its contents.",
 )
 
 iface.launch()