-02 ver

app.py

@@ -82,6 +82,13 @@ def gen_tags_from_image_file(image_bytes: io.BytesIO) -> str:
         print("Error generating tags:", e)
         return ""
 
+def decode_image_from_b64(b64_str: str):
+    try:
+        img_bytes = base64.b64decode(b64_str)
+        return Image.open(io.BytesIO(img_bytes))
+    except Exception:
+        return None
+
 # -------------------------
 # App logic: add item
 # -------------------------
@@ -131,31 +138,41 @@ def search_items(query_image, query_text, limit: int = 5, min_score: float = 0.9
     elif query_text and len(query_text.strip()) > 0:
         qvec = embed_text(query_text).tolist()
     else:
-        return "⚠️ Please provide a query image or some query text."
+        return [], "⚠️ Please provide a query image or some query text."
 
     try:
         hits = qclient.search(collection_name=COLLECTION, query_vector=qvec, limit=limit)
     except Exception as e:
-        return f"❌ Error querying Qdrant: {e}"
+        return [], f"❌ Error querying Qdrant: {e}"
 
     if not hits:
-        return "No results found."
+        return [], "No results found."
 
-    results = []
+    images, captions = [], []
     for h in hits:
         score = getattr(h, "score", None)
-        if score is None or score < min_score:  # filter by threshold
+        if score is None or score < min_score:
             continue  
 
         payload = h.payload or {}
-        results.append(
-            f"id:{h.id} | score:{score:.4f} | mode:{payload.get('mode','')} | tags:{payload.get('tags','')} | text:{payload.get('text','')}"
-        )
+        caption = f"ID: {h.id}\nScore: {score:.4f}\nMode: {payload.get('mode','')}\nTags: {payload.get('tags','')}\nText: {payload.get('text','')}"
+        captions.append(caption)
+
+        if payload.get("has_image") and payload.get("image_b64"):
+            img = decode_image_from_b64(payload["image_b64"])
+            if img:
+                images.append(img)
+            else:
+                images.append(Image.new("RGB", (200,200), color="gray"))
+        else:
+            # Placeholder for text-only entries
+            img = Image.new("RGB", (200,200), color="lightblue")
+            images.append(img)
 
-    if not results:
-        return f"No results above similarity threshold {min_score:.2f}"
+    if not images:
+        return [], f"No results above similarity threshold {min_score:.2f}"
 
-    return "\n\n".join(results)
+    return list(zip(images, captions)), ""
 
 # -------------------------
 # Gradio UI
@@ -176,10 +193,11 @@ with gr.Blocks(title="Lost & Found — Simple Helper") as demo:
             limit_slider = gr.Slider(1, 10, value=5, step=1, label="Max results")
             score_slider = gr.Slider(0.0, 1.0, value=0.90, step=0.01, label="Min similarity score")
             search_btn = gr.Button("🔎 Search")
-            search_out = gr.Textbox(label="Search results", interactive=False)
+            gallery = gr.Gallery(label="Search Results", show_label=True, elem_id="gallery").style(grid=[2], height="auto")
+            search_msg = gr.Textbox(label="Message", interactive=False)
 
     add_btn.click(add_item, inputs=[mode, upload_img, text_desc], outputs=[add_out])
-    search_btn.click(search_items, inputs=[query_img, query_text, limit_slider, score_slider], outputs=[search_out])
+    search_btn.click(search_items, inputs=[query_img, query_text, limit_slider, score_slider], outputs=[gallery, search_msg])
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)