Update app.py

app.py

@@ -52,25 +52,24 @@ 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))
 
-# 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."
-)
+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)
 
-iface.launch()
+demo.launch()