Update app.py

app.py

@@ -325,16 +325,21 @@ def bePrepare1():
     return "I am ready, ask me questions with model google flan-t5."
 
 def ask_question(query):
+    msg="Blank question! "
     if not qa_chain:
-        return "❌ Please clik the button to get the udated resources with tiny Lama."
-    response = qa_chain.invoke({"query": query})
-    return response["result"]
-
+        msg="❌ Please clik the button to get the udated resources with tiny Lama."
+    if query:
+        response = qa_chain.invoke({"query": query})
+        msg= response["result"]
+    return msg
 def ask_question1(query):
+    msg="Blank question!"
     if not qa_chain1:
-        return "❌ Please clik the button to get the udated resources google flan-t5."
-    response1 = qa_chain1.invoke({"query": query})
-    return response1["result"]
+        msg="❌ Please clik the button to get the udated resources google flan-t5."
+    if query:
+        response1 = qa_chain1.invoke({"query": query})
+        msg=response1["result"]
+    return msg
 #====================
 # Gradio UI
 with gr.Blocks() as demo:
@@ -342,7 +347,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         # LEFT COLUMN: Document Management
         with gr.Column(scale=1):
-            gr.Markdown("## 🧠 Using TinyLama Model")
+            gr.Markdown("## 🧠 Using heavy TinyLama Model")
             with gr.Row():
                 Index_processing_output=gr.Textbox(label="📁 Status for tiny lama", interactive=False)
             with gr.Row():
@@ -351,11 +356,12 @@ with gr.Blocks() as demo:
             with gr.Row():
                 query_input = gr.Textbox(label="❓ Your Question pls")
                 query_btn = gr.Button("🧠 Get Answer")
+            with gr.Row():
                 answer_output = gr.Textbox(label="✅ Answer", lines=4)
                 query_btn.click(ask_question, inputs=query_input, outputs=answer_output)
         # RIGHT COLUMN: Document Management
         with gr.Column(scale=2):
-            gr.Markdown("## 🧠 Using google flan-t5")
+            gr.Markdown("## 🧠 Using ligth model - google flan-t5")
             Index_processing_output1=gr.Textbox(label="📁 Status for google flan-t5", interactive=False)
             Index_processing_btn1 = gr.Button("🔄 Clik to get the udated resources with google flan-t5")
             Index_processing_btn1.click(bePrepare1, inputs=None, outputs=Index_processing_output1)