fixed V

app.py

@@ -14,28 +14,32 @@ os.makedirs(UPLOAD_DIR, exist_ok=True)
 
 COLLECTION = "lost_and_found"
 
-# Qdrant client
-qclient = qdrant_client.QdrantClient(":memory:")  # In-memory for Hugging Face
+# Qdrant client (in-memory for Hugging Face)
+qclient = qdrant_client.QdrantClient(":memory:")
 encoder = SentenceTransformer("clip-ViT-B-32")
 
-# Create collection
+# Create collection if not exists
 if not qclient.collection_exists(COLLECTION):
     qclient.create_collection(
         collection_name=COLLECTION,
         vectors_config=models.VectorParams(size=512, distance=models.Distance.COSINE),
     )
 
+
 # ===============================
 # Encode Function (Text or Image)
 # ===============================
 def encode_data(text=None, image=None):
-    if image is not None:
+    if isinstance(image, Image.Image):  # Image is already PIL
+        return encoder.encode(image.convert("RGB"))
+    elif isinstance(image, str):  # Path to image
         return encoder.encode(Image.open(image).convert("RGB"))
     elif text:
         return encoder.encode([text])[0]
     else:
         return None
 
+
 # ===============================
 # Add Item
 # ===============================
@@ -44,16 +48,13 @@ def add_item(text, image, uploader_name, uploader_phone):
         img_path = None
         vector = None
 
-        if image:
+        if isinstance(image, Image.Image):  # PIL image
             img_id = str(uuid.uuid4())
             img_path = os.path.join(UPLOAD_DIR, f"{img_id}.png")
             image.save(img_path)
-
-            # ✅ Always store image embedding if available
-            vector = encode_data(image=img_path)
+            vector = encode_data(image=image)
 
         elif text:
-            # ✅ Fallback: text embedding
             vector = encode_data(text=text)
 
         if vector is None:
@@ -78,13 +79,14 @@ def add_item(text, image, uploader_name, uploader_phone):
     except Exception as e:
         return f"❌ Error: {e}"
 
+
 # ===============================
 # Search Function
 # ===============================
 def search_items(text, image, max_results, min_score):
     try:
         vector = None
-        if image:
+        if isinstance(image, Image.Image):  # Search with PIL
             vector = encode_data(image=image)
         elif text:
             vector = encode_data(text=text)
@@ -119,6 +121,7 @@ def search_items(text, image, max_results, min_score):
     except Exception as e:
         return f"❌ Error: {e}", []
 
+
 # ===============================
 # Delete All
 # ===============================
@@ -130,12 +133,14 @@ def clear_database():
     )
     return "🗑️ Database cleared!"
 
+
 # ===============================
 # Gradio UI
 # ===============================
 with gr.Blocks() as demo:
     gr.Markdown("## 🗝️ Lost & Found - Database")
 
+    # --- Add Item Tab ---
     with gr.Tab("➕ Add Item"):
         with gr.Row():
             text_input = gr.Textbox(label="Description (optional)")
@@ -152,6 +157,7 @@ with gr.Blocks() as demo:
             outputs=add_output,
         )
 
+    # --- Search Tab ---
     with gr.Tab("🔍 Search"):
         with gr.Row():
             search_text = gr.Textbox(label="Search by text (optional)")
@@ -169,6 +175,7 @@ with gr.Blocks() as demo:
             outputs=[search_text_out, search_gallery],
         )
 
+    # --- Admin Tab ---
     with gr.Tab("🗑️ Admin"):
         clear_btn = gr.Button("Clear Database")
         clear_out = gr.Textbox(label="Status")