Update main.py

main.py

@@ -1,26 +1,36 @@
+# main.py
+from fastapi import FastAPI
+from fastapi.responses import RedirectResponse
 import gradio as gr
-from app import answer_question_from_doc  # Assuming `answer_question_from_doc` is in app.py
-from appImage import answer_question_from_image  # Assuming `answer_question_from_image` is in appImage.py
 
-# Create tabs for Document QA and Image QA
-with gr.Blocks() as demo:
-    with gr.Tab("Document QA"):
-        gr.Markdown("### Document Question Answering")
-        file_input = gr.File(label="Upload PDF")
-        question_input = gr.Textbox(label="Ask a Question")
-        file_output = gr.Textbox(label="Answer")
-        file_input.change(answer_question_from_doc, [file_input, question_input], file_output)
-    
-    with gr.Tab("Image QA"):
-        gr.Markdown("### Image Question Answering")
-        img_input = gr.Image(label="Upload Image")
-        img_question_input = gr.Textbox(label="Ask a Question")
-        img_output = gr.Textbox(label="Answer")
-        img_input.change(answer_question_from_image, [img_input, img_question_input], img_output)
+# ✅ Import your QA functions from app.py and appImage.py
+from app import answer_question_from_doc
+from appImage import answer_question_from_image
 
+app = FastAPI()
+
+# === Document QA Tab ===
+with gr.Blocks() as doc_interface:
+    gr.Markdown("## 📄 Document QA")
+    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)
+
+# === Image QA Tab ===
+with gr.Blocks() as img_interface:
+    gr.Markdown("## 🖼️ Image QA")
+    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)
+
+# === Combine Tabs ===
+demo = gr.TabbedInterface([doc_interface, img_interface], tab_names=["Document QA", "Image QA"])
+
+# Mount Gradio App
 app = gr.mount_gradio_app(app, demo, path="/")
 
-# Root redirect to Gradio app
 @app.get("/")
-def home():
-    return RedirectResponse(url="/")
+def root():
+    return RedirectResponse(url="/")