Update app.py

app.py

@@ -7,7 +7,10 @@ import google.generativeai as genai
 import gradio as gr
 
 # Function for initialization
-def initialize(file_path, question):
+def initialize(pdf_file, question):
+    with open("/tmp/uploaded_file.pdf", "wb") as f:
+        f.write(pdf_file.read())
+    file_path = "/tmp/uploaded_file.pdf"
     genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
     model = genai.GenerativeModel('gemini-pro')
     model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.3)
@@ -28,18 +31,21 @@ def initialize(file_path, question):
     else:
         return "Error: Unable to process the document. Please ensure the PDF file is valid."
 
-def chatbot(file):
-    file_path = "/tmp/uploaded_file.pdf"
-    with open(file_path, "wb") as f:
-        f.write(file.read())
-    question_input = input("Ask about the document: ")
-    return initialize(file_path, question_input)
-
-# Create Gradio interface
-inputs = gr.inputs.File(label="Upload PDF", type="file")
-outputs = gr.outputs.Textbox(label="Answer - GeminiPro")
+# Create a Gradio interface for file upload and question input
+iface = gr.Interface(
+    fn=initialize,
+    inputs=[
+        gr.inputs.File(label="Upload PDF", type="file"),
+        gr.inputs.Textbox(label="Question")
+    ],
+    outputs="text",
+    title="GeminiPro Q&A Bot",
+    description="Ask questions about the uploaded PDF document.",
+)
+
+# Launch the interface
+iface.launch()
 
-gr.Interface(fn=chatbot, inputs=inputs, outputs=outputs, title="GeminiPro Q&A Bot", description="Ask questions about the uploaded PDF document.").launch()