Set up model

app.py

@@ -0,0 +1,126 @@
+import gradio as gr
+import time
+from datetime import datetime
+import pandas as pd
+from sentence_transformers import SentenceTransformer
+from qdrant_client import QdrantClient
+from qdrant_client.models import Filter, FieldCondition, MatchValue
+
+import os
+from qdrant_client import QdrantClient
+
+qdrant_client = QdrantClient(
+    url=os.environ.get("Qdrant_url"),
+    api_key=os.environ.get("Qdrant_api")
+)
+
+# โมเดลที่โหลดล่วงหน้า
+models = {
+    "E5 (intfloat/multilingual-e5-small)": SentenceTransformer('intfloat/multilingual-e5-small'),
+    "MiniLM (paraphrase-multilingual-MiniLM-L12-v2)": SentenceTransformer('paraphrase-multilingual-MiniLM-L12-v2'),
+    "DistilUSE (distiluse-base-multilingual-cased-v1)": SentenceTransformer('sentence-transformers/distiluse-base-multilingual-cased-v1')
+}
+
+# Global memory to hold feedback state
+latest_query_result = {"query": "", "result": "", "model": ""}
+
+
+# 🔍 Search Functions
+def search_with_e5(query):
+    embed = models["E5 (intfloat/multilingual-e5-small)"].encode("query: " + query)
+    return embed
+
+def search_with_minilm(query):
+    embed = models["MiniLM (paraphrase-multilingual-MiniLM-L12-v2)"].encode(query)
+    return embed
+
+def search_with_distiluse(query):
+    embed = models["DistilUSE (distiluse-base-multilingual-cased-v1)"].encode(query)
+    return embed
+
+
+# 🌟 Main search function
+def search_product(query, model_name):
+    start_time = time.time()
+
+    # Choose encoder function
+    if "E5" in model_name:
+        query_embed = search_with_e5(query)
+    elif "MiniLM" in model_name:
+        query_embed = search_with_minilm(query)
+    elif "DistilUSE" in model_name:
+        query_embed = search_with_distiluse(query)
+    else:
+        return "❌ ไม่พบโมเดล"
+
+    # Query Qdrant
+    result = qdrant_client.query_points(
+        collection_name="product_E5",
+        query=query_embed.tolist(),
+        with_payload=True,
+        query_filter=Filter(
+            must=[FieldCondition(key="type", match=MatchValue(value="product"))]
+        )
+    ).points
+
+    elapsed = time.time() - start_time
+
+    # Format result
+    output = f"⏱ Time: {elapsed:.2f}s\n\n📦 ผลลัพธ์:\n"
+    result_summary = ""
+    for res in result:
+        line = f"- {res.payload.get('name', '')} (score: {res.score:.4f})"
+        output += line + "\n"
+        result_summary += line + " | "
+
+    # Save latest query
+    latest_query_result["query"] = query
+    latest_query_result["result"] = result_summary.strip()
+    latest_query_result["model"] = model_name
+
+    return output
+
+
+# 📝 Logging feedback
+def log_feedback(feedback):
+    now = datetime.now().isoformat()
+    log_entry = {
+        "timestamp": now,
+        "model": latest_query_result["model"],
+        "query": latest_query_result["query"],
+        "result": latest_query_result["result"],
+        "feedback": feedback
+    }
+    df = pd.DataFrame([log_entry])
+    df.to_csv("feedback_log.csv", mode='a', header=not pd.io.common.file_exists("feedback_log.csv"), index=False)
+    return f"✅ Feedback saved: {feedback}"
+
+
+# 🎨 Gradio UI
+with gr.Blocks() as demo:
+    gr.Markdown("## 🔎 Product Semantic Search (Vector Search + Qdrant)")
+
+    with gr.Row():
+        model_selector = gr.Dropdown(
+            choices=list(models.keys()),
+            label="เลือกโมเดล",
+            value="E5 (intfloat/multilingual-e5-small)"
+        )
+        query_input = gr.Textbox(label="พิมพ์คำค้นหา")
+
+    result_output = gr.Textbox(label="📋 ผลลัพธ์")
+
+    with gr.Row():
+        match_btn = gr.Button("✅ ตรง")
+        not_match_btn = gr.Button("❌ ไม่ตรง")
+
+    feedback_status = gr.Textbox(label="📬 สถานะ Feedback")
+
+    # Events
+    submit_fn = lambda q, m: search_product(q, m)
+    query_input.submit(submit_fn, inputs=[query_input, model_selector], outputs=result_output)
+    match_btn.click(lambda: log_feedback("match"), outputs=feedback_status)
+    not_match_btn.click(lambda: log_feedback("not_match"), outputs=feedback_status)
+
+# Run app
+demo.launch(share=True)