Update app.py

app.py

@@ -1,13 +1,11 @@
-!pip install gradio 
-
 import os
 import streamlit as st
 from transformers import pipeline
 from PyPDF2 import PdfReader
 import tempfile
-import gradio as gr
 
 # Function to perform question-answering
+@st.cache_data(show_spinner=False)
 def question_answering(questions, pdf_text):
     # Perform question-answering using Hugging Face's Transformers
     question_answerer = pipeline("question-answering", model="distilbert-base-cased-distilled-squad", tokenizer="distilbert-base-cased-distilled-squad")
@@ -15,8 +13,15 @@ def question_answering(questions, pdf_text):
 
     return answers
 
-def main(uploaded_file, question):
-    if uploaded_file is not None:
+def main():
+    st.title("Question Answering on PDF Files")
+
+    uploaded_file = st.file_uploader("Upload a PDF file:", type=["pdf"])
+
+    st.write("Enter your question(s) below (separate multiple questions with new lines):")
+    questions = st.text_area("Questions").split('\n')
+
+    if st.button("Answer") and uploaded_file is not None:
         pdf_path = os.path.join(tempfile.gettempdir(), uploaded_file.name)
         with open(pdf_path, "wb") as f:
             f.write(uploaded_file.read())
@@ -25,15 +30,14 @@ def main(uploaded_file, question):
         pdf_reader = PdfReader(pdf_path)
         pdf_text = "\n".join([pdf_page.extract_text() for pdf_page in pdf_reader.pages])
 
-        # Perform question-answering
-        answers = question_answering([question], pdf_text)
+        # Perform question-answering in batches
+        answers = question_answering(questions, pdf_text)
 
-        result = f"Question: '{question}'\nAnswer: {answers[0]['answer']}\nScore: {answers[0]['score']:.2f}"
-        return result
+        st.write("Questions and Answers:")
+        for i, (question, answer) in enumerate(zip(questions, answers)):
+            st.write(f"Question {i + 1}: '{question}'")
+            st.write("Answer:", answer['answer'])
+            st.write("Score:", answer['score'])
 
-iface = gr.Interface(fn=main, 
-                     inputs=[gr.inputs.File(type="file", label="Upload a PDF file"), 
-                             gr.inputs.Textbox(label="Enter a question:")],
-                     outputs=gr.outputs.Textbox(label="Question Answering Results"),
-                     live=True)
-iface.launch()
+if __name__ == "__main__":
+    main()