Update app.py

app.py

@@ -5,47 +5,33 @@ import pdfplumber
 # Load the pre-trained question-answering model
 qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
 
-class PDFQuestionAnswering:
-    def __init__(self):
-        self.text = ""
+def answer_question(file, question):
+    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()
 
-    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 user's 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"]
 
-pdf_qa = PDFQuestionAnswering()
+    except Exception as e:
+        # Handle exceptions, e.g., invalid PDF or other errors
+        return f"Error processing PDF: {str(e)}"
 
 iface = gr.Interface(
-    fn_map={
-        "Load PDF": pdf_qa.load_pdf,
-        "Answer Question": pdf_qa.answer_question,
-    },
+    fn=answer_question,
     inputs=[
-        gr.File(label="Upload PDF", type="file"),
-        gr.Textbox(label="Enter Question", type="text"),
+        gr.File(label="Upload PDF"),
+        gr.Textbox(label="Enter Question", type="text", default="What is the main topic of the document?"),
     ],
     outputs="text",
     live=True,
     title="PDF Question-Answering",
-    description="Upload a PDF, load it, and ask questions about its contents.",
+    description="Upload a PDF and ask questions about its contents.",
 )
 
 iface.launch()