logic improvisation

app.py

@@ -14,7 +14,7 @@ os.makedirs(UPLOAD_DIR, exist_ok=True)
 qclient = QdrantClient(":memory:")
 COLLECTION = "lost_and_found"
 
-# Create collection (with deprecation fix)
+# Create collection
 if not qclient.collection_exists(COLLECTION):
     qclient.create_collection(
         COLLECTION,
@@ -42,9 +42,9 @@ def encode_data(text=None, image=None):
 
 
 # ==============================
-# Add Item
+# Add Item (finder uploads)
 # ==============================
-def add_item(mode, text, image, uploader_name, uploader_phone):
+def add_item(text, image, uploader_name, uploader_phone):
     try:
         img_path = None
         if image:
@@ -61,7 +61,6 @@ def add_item(mode, text, image, uploader_name, uploader_phone):
                     id=str(uuid.uuid4()),
                     vector=vector.tolist(),
                     payload={
-                        "mode": mode,
                         "text": text or "",
                         "uploader_name": uploader_name or "N/A",
                         "uploader_phone": uploader_phone or "N/A",
@@ -71,13 +70,13 @@ def add_item(mode, text, image, uploader_name, uploader_phone):
                 )
             ],
         )
-        return f"✅ Added successfully as {mode}!"
+        return "✅ Item added to database!"
     except Exception as e:
         return f"❌ Error: {e}"
 
 
 # ==============================
-# Search Items
+# Search Items (for lost things)
 # ==============================
 def search_items(text, image, max_results, min_score):
     try:
@@ -94,14 +93,12 @@ def search_items(text, image, max_results, min_score):
         texts, imgs = [], []
         for r in results:
             p = r.payload
-            desc = (
-                f"id:{r.id} | score:{r.score:.3f} | mode:{p.get('mode','')} | text:{p.get('text','')}"
-            )
+            desc = f"id:{r.id} | score:{r.score:.3f} | text:{p.get('text','')}"
 
-            # Always show uploader details
+            # Show finder details
             uploader_name = p.get("uploader_name", "N/A") or "N/A"
             uploader_phone = p.get("uploader_phone", "N/A") or "N/A"
-            desc += f" | uploader:{uploader_name} ({uploader_phone})"
+            desc += f" | finder:{uploader_name} ({uploader_phone})"
 
             texts.append(desc)
             if p.get("has_image") and "image_path" in p:
@@ -125,24 +122,23 @@ def clear_all():
 # Gradio UI
 # ==============================
 with gr.Blocks() as demo:
-    gr.Markdown("# 🔑 Lost & Found Image/Text Search")
+    gr.Markdown("# 🔑 Lost & Found Database")
 
-    with gr.Tab("➕ Add Item"):
-        mode = gr.Radio(["lost", "found"], label="Mode", value="found")
+    with gr.Tab("➕ Add Found Item"):
         text_in = gr.Textbox(label="Description (optional)")
         img_in = gr.Image(type="pil", label="Upload Image")
-        uploader_name = gr.Textbox(label="Your Name")
-        uploader_phone = gr.Textbox(label="Your Phone")
+        uploader_name = gr.Textbox(label="Finder Name")
+        uploader_phone = gr.Textbox(label="Finder Phone")
         add_btn = gr.Button("Add to Database")
         add_out = gr.Textbox(label="Status")
 
         add_btn.click(
             add_item,
-            inputs=[mode, text_in, img_in, uploader_name, uploader_phone],
+            inputs=[text_in, img_in, uploader_name, uploader_phone],
             outputs=add_out,
         )
 
-    with gr.Tab("🔍 Search"):
+    with gr.Tab("🔍 Search Lost Item"):
         search_text = gr.Textbox(label="Search by text (optional)")
         search_img = gr.Image(type="pil", label="Search by image (optional)")
         max_results = gr.Slider(1, 10, value=5, step=1, label="Max results")