Update app.py

app.py

@@ -52,24 +52,25 @@ def generate_text(prompt, max_length=50):
     return tokenizer.decode(outputs[0])
 
 # Gradio Interface
+input_file = gr.File(label="Upload PDF File")
+input_question = gr.Textbox(label="Ask about the document")
+output_text_gemini = gr.Textbox(label="Answer - GeminiPro")
+input_prompt = gr.Textbox(label="Enter prompt for text completion")
+output_text_mistral = gr.Textbox(label="Completed Text - Mistral")
+
 def pdf_qa_wrapper(file, question):
     return asyncio.run(pdf_qa(file, question))
 
-with gr.Blocks() as demo:
-    gr.Markdown("# Combined PDF QA and Text Completion System")
-    
-    with gr.Tab("PDF Question Answering"):
-        input_file = gr.File(label="Upload PDF File")
-        input_question = gr.Textbox(label="Ask about the document")
-        output_text_gemini = gr.Textbox(label="Answer - GeminiPro")
-        pdf_qa_button = gr.Button("Ask Question")
-    
-    with gr.Tab("Text Completion"):
-        input_prompt = gr.Textbox(label="Enter prompt for text completion")
-        output_text_mistral = gr.Textbox(label="Completed Text - Mistral")
-        complete_text_button = gr.Button("Complete Text")
-
-    pdf_qa_button.click(pdf_qa_wrapper, inputs=[input_file, input_question], outputs=output_text_gemini)
-    complete_text_button.click(generate_text, inputs=input_prompt, outputs=output_text_mistral)
+# Create Gradio Interface
+iface = gr.Interface(
+    fn=[pdf_qa_wrapper, generate_text],
+    inputs=[
+        [input_file, input_question],
+        input_prompt
+    ],
+    outputs=[output_text_gemini, output_text_mistral],
+    title="Combined PDF QA and Text Completion System",
+    description="Upload a PDF file to ask questions about its content, or enter a prompt for text completion."
+)
 
-demo.launch()
+iface.launch()