Update main.py

main.py

@@ -1,9 +1,7 @@
-# main.py
 from fastapi import FastAPI
 from fastapi.responses import RedirectResponse
 import gradio as gr
 
-# ✅ Import your QA functions from app.py and appImage.py
 from app import answer_question_from_doc
 from appImage import answer_question_from_image
 
@@ -15,7 +13,8 @@ with gr.Blocks() as doc_interface:
     doc_file = gr.File(label="Upload File (PDF, DOCX, PPTX, XLSX)")
     doc_question = gr.Textbox(label="Ask a question")
     doc_answer = gr.Textbox(label="Answer")
-    doc_file.change(fn=answer_question_from_doc, inputs=[doc_file, doc_question], outputs=doc_answer)
+    doc_submit = gr.Button("Get Answer")
+    doc_submit.click(fn=answer_question_from_doc, inputs=[doc_file, doc_question], outputs=doc_answer)
 
 # === Image QA Tab ===
 with gr.Blocks() as img_interface:
@@ -23,7 +22,8 @@ with gr.Blocks() as img_interface:
     img_input = gr.Image(label="Upload an Image")
     img_question = gr.Textbox(label="Ask a question")
     img_answer = gr.Textbox(label="Answer")
-    img_input.change(fn=answer_question_from_image, inputs=[img_input, img_question], outputs=img_answer)
+    img_submit = gr.Button("Get Answer")
+    img_submit.click(fn=answer_question_from_image, inputs=[img_input, img_question], outputs=img_answer)
 
 # === Combine Tabs ===
 demo = gr.TabbedInterface([doc_interface, img_interface], tab_names=["Document QA", "Image QA"])