add ui when search

app.py

@@ -95,6 +95,8 @@ latest_query_result = {"query": "", "result": "", "raw_query": "", "time": ""}
 
 # Search Function
 def search_product(query):
+    yield gr.update(value="🔄 กำลังค้นหา..."), ""
+
     start_time = time.time()
     latest_query_result["raw_query"] = query
 
@@ -110,9 +112,9 @@ def search_product(query):
             limit=50
         ).points
     except Exception as e:
-        return f"<p>❌ Qdrant error: {str(e)}</p>"
+        yield gr.update(value="❌ Qdrant error"), f"<p>❌ Qdrant error: {str(e)}</p>"
+        return
 
-    # ✅ Rerank Top 10
     if len(result) > 0:
         topk = 10
         docs = [r.payload.get("name", "") for r in result[:topk]]
@@ -158,7 +160,6 @@ def search_product(query):
 
     if not found:
         html_output += '<div style="text-align: center; font-size: 18px; color: #a00; padding: 30px;">❌ ไม่พบสินค้าที่เกี่ยวข้องกับคำค้นนี้</div>'
-        return html_output
 
     latest_query_result.update({
         "query": corrected_query,
@@ -166,7 +167,7 @@ def search_product(query):
         "time": elapsed,
     })
 
-    return html_output
+    yield gr.update(value="✅ ค้นหาเสร็จแล้ว!"), html_output
 
 # Feedback Function
 def log_feedback(feedback):
@@ -191,6 +192,7 @@ with gr.Blocks() as demo:
 
     query_input = gr.Textbox(label="พิมพ์คำค้นหา")
     result_output = gr.HTML(label="📋 ผลลัพธ์")
+    status_output = gr.Textbox(label="🕒 สถานะ", interactive=False)
 
     with gr.Row():
         match_btn = gr.Button("✅ ตรง")
@@ -198,7 +200,11 @@ with gr.Blocks() as demo:
 
     feedback_status = gr.Textbox(label="📬 สถานะ Feedback")
 
-    query_input.submit(search_product, inputs=[query_input], outputs=result_output)
+    query_input.submit(
+        search_product,
+        inputs=[query_input],
+        outputs=[status_output, result_output]
+    )
     match_btn.click(fn=lambda: log_feedback("match"), outputs=feedback_status)
     not_match_btn.click(fn=lambda: log_feedback("not_match"), outputs=feedback_status)