7.0v

app.py

@@ -1,14 +1,19 @@
 import gradio as gr
 import uuid
+import os
 from qdrant_client import QdrantClient
 from qdrant_client.models import PointStruct
 from sentence_transformers import SentenceTransformer
 from PIL import Image
 import numpy as np
 
+# Create uploads folder
+UPLOAD_DIR = "uploads"
+os.makedirs(UPLOAD_DIR, exist_ok=True)
+
 # Connect to Qdrant
 COLLECTION = "lost_and_found"
-qclient = QdrantClient(":memory:")  # use in-memory for demo, replace with host/port for persistence
+qclient = QdrantClient(":memory:")  # for demo, replace with your Qdrant server for persistence
 
 # Load CLIP model
 model = SentenceTransformer("sentence-transformers/clip-ViT-B-32-multilingual-v1")
@@ -39,9 +44,19 @@ def add_item(mode, text, image, name, phone):
             "text": text,
             "has_image": image is not None,
         }
+
+        # Save image file if uploaded
+        if image is not None:
+            img = Image.open(image).convert("RGB")
+            fname = f"{uuid.uuid4().hex}.png"
+            fpath = os.path.join(UPLOAD_DIR, fname)
+            img.save(fpath)
+            payload["image_path"] = fpath
+
         if mode == "found":
             payload["finder_name"] = name
             payload["finder_phone"] = phone
+
         qclient.upsert(
             collection_name=COLLECTION,
             points=[PointStruct(id=str(uuid.uuid4()), vector=vector.tolist(), payload=payload)]
@@ -73,8 +88,10 @@ def search_items(query_image, query_text, limit, min_score):
             if pl.get("mode") == "found":
                 info += f" | found by: {pl.get('finder_name','?')} ({pl.get('finder_phone','?')})"
             out_texts.append(info)
-            if pl.get("has_image"):
-                out_imgs.append(query_image)  # just echo search image (or store actual image paths if needed)
+
+            if pl.get("image_path"):
+                out_imgs.append(pl["image_path"])
+
         return "\n".join(out_texts) if out_texts else "No matches.", out_imgs
     except Exception as e:
         return f"❌ Error: {e}", []
@@ -82,11 +99,13 @@ def search_items(query_image, query_text, limit, min_score):
 # Clear all images
 def clear_all_images():
     try:
+        # Clear uploads folder
+        for f in os.listdir(UPLOAD_DIR):
+            os.remove(os.path.join(UPLOAD_DIR, f))
+        # Clear Qdrant
         qclient.delete(
             collection_name=COLLECTION,
-            points_selector={
-                "filter": {"must": [{"key": "has_image", "match": {"value": True}}]}
-            }
+            points_selector={"filter": {"must": [{"key": "has_image", "match": {"value": True}}]}}
         )
         return "🗑️ All image items cleared!"
     except Exception as e: