fixed

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 import uuid
+import base64
+import io
 from qdrant_client import QdrantClient
 from qdrant_client.models import PointStruct, VectorParams, Distance
 from sentence_transformers import SentenceTransformer
@@ -8,8 +10,8 @@ from PIL import Image
 # --------------------------
 # Qdrant Cloud Connection
 # --------------------------
-QDRANT_URL = "https://ff4da494-27b1-413c-ba58-d5ea14932fe1.europe-west3-0.gcp.cloud.qdrant.io:6333"  
-QDRANT_API_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhY2Nlc3MiOiJtIn0.jjeB1JgnUSlb1hOOKMdRpVvMrUER57-udT-X1AWXT1E"                    
+QDRANT_URL = "https://ff4da494-27b1-413c-ba58-d5ea14932fe1.europe-west3-0.gcp.cloud.qdrant.io:6333"  # 🔑 Replace with your cluster URL
+QDRANT_API_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhY2Nlc3MiOiJtIn0.jjeB1JgnUSlb1hOOKMdRpVvMrUER57-udT-X1AWXT1E"                    # 🔑 Replace with your API key
 COLLECTION_NAME = "lost_and_found"
 
 # CLIP model (text + image embeddings)
@@ -35,6 +37,17 @@ qclient.recreate_collection(
 # Helper Functions
 # --------------------------
 
+def image_to_base64(img: Image.Image) -> str:
+    """Convert PIL image to base64 string"""
+    buf = io.BytesIO()
+    img.save(buf, format="PNG")
+    return base64.b64encode(buf.getvalue()).decode("utf-8")
+
+def base64_to_image(b64_str: str) -> Image.Image:
+    """Convert base64 string back to PIL image"""
+    img_bytes = base64.b64decode(b64_str)
+    return Image.open(io.BytesIO(img_bytes))
+
 def embed_text(text: str):
     return embedder.encode(text).tolist()
 
@@ -48,12 +61,14 @@ def add_item(image, description, finder_name, finder_phone):
 
     # Encode image
     embedding = embed_image(image)
+    img_b64 = image_to_base64(image)
 
-    # Store metadata
+    # Store metadata including image
     metadata = {
         "description": description,
         "finder_name": finder_name if finder_name.strip() else "NA",
         "finder_phone": finder_phone if finder_phone.strip() else "NA",
+        "image_b64": img_b64
     }
 
     # Insert into Qdrant
@@ -98,7 +113,13 @@ def search_items(query_text, query_image):
         name = r.payload.get("finder_name", "NA")
         phone = r.payload.get("finder_phone", "NA")
         output_text += f"- **{desc}** (Finder: {name}, Phone: {phone})\n"
-        gallery.append(Image.new("RGB", (200, 200), color="gray"))  # Placeholder (no storage for images)
+
+        if "image_b64" in r.payload:
+            try:
+                img = base64_to_image(r.payload["image_b64"])
+                gallery.append(img)
+            except Exception:
+                pass
 
     return output_text, gallery