Fix CLIrrr2 model issue in appetete333.py

app.py

@@ -31,6 +31,9 @@ print("Loading CLIP model (this may take 20-60s the first time)...")
 MODEL_ID = "sentence-transformers/clip-ViT-B-32-multilingual-v1"
 clip_model = SentenceTransformer(MODEL_ID)
 
+# Detect correct embedding size dynamically
+VECTOR_SIZE = clip_model.get_sentence_embedding_dimension()
+
 genai_client = genai.Client(api_key=GEMINI_API_KEY) if GEMINI_API_KEY else None
 
 if not QDRANT_URL:
@@ -38,7 +41,6 @@ if not QDRANT_URL:
 
 qclient = QdrantClient(url=QDRANT_URL, api_key=QDRANT_API_KEY)
 COLLECTION = "lost_found_items"
-VECTOR_SIZE = 512
 
 # Create collection if missing
 try:
@@ -75,7 +77,8 @@ def gen_tags_from_image_file(image_bytes: io.BytesIO) -> str:
             contents=[prompt_text, file_obj],
         )
         return response.text.strip()
-    except Exception:
+    except Exception as e:
+        print("Error generating tags:", e)
         return ""
 
 # -------------------------
@@ -85,38 +88,39 @@ def add_item(mode: str, uploaded_image, text_description: str):
     item_id = str(uuid.uuid4())
     payload = {"mode": mode, "text": text_description}
 
-    if uploaded_image is not None:
-        img_bytes = io.BytesIO()
-        uploaded_image.convert("RGB").save(img_bytes, format="PNG")
-        img_bytes.seek(0)
+    try:
+        if uploaded_image is not None:
+            img_bytes = io.BytesIO()
+            uploaded_image.convert("RGB").save(img_bytes, format="PNG")
+            img_bytes.seek(0)
 
-        vec = embed_image_pil(uploaded_image).tolist()
-        payload["has_image"] = True
+            vec = embed_image_pil(uploaded_image).tolist()
+            payload["has_image"] = True
 
-        payload["tags"] = gen_tags_from_image_file(img_bytes)
-        payload["image_b64"] = base64.b64encode(img_bytes.getvalue()).decode("utf-8")
-    else:
-        vec = embed_text(text_description).tolist()
-        payload["has_image"] = False
-        if genai_client:
-            try:
-                resp = genai_client.models.generate_content(
-                    model="gemini-2.5-flash",
-                    contents=f"Give 4 short, comma-separated tags for this item described as: {text_description}. Reply only with tags."
-                )
-                payload["tags"] = resp.text.strip()
-            except Exception:
-                payload["tags"] = ""
+            payload["tags"] = gen_tags_from_image_file(img_bytes)
+            payload["image_b64"] = base64.b64encode(img_bytes.getvalue()).decode("utf-8")
         else:
-            payload["tags"] = ""
+            vec = embed_text(text_description).tolist()
+            payload["has_image"] = False
+            if genai_client:
+                try:
+                    resp = genai_client.models.generate_content(
+                        model="gemini-2.5-flash",
+                        contents=f"Give 4 short, comma-separated tags for this item described as: {text_description}. Reply only with tags."
+                    )
+                    payload["tags"] = resp.text.strip()
+                except Exception:
+                    payload["tags"] = ""
+            else:
+                payload["tags"] = ""
 
-    try:
+        # Upsert into Qdrant
         point = PointStruct(id=item_id, vector=vec, payload=payload)
         qclient.upsert(collection_name=COLLECTION, points=[point], wait=True)
-    except Exception as e:
-        return f"Error saving to Qdrant: {e}"
 
-    return f"Saved item id: {item_id}\nTags: {payload.get('tags','')}"
+        return f"✅ Saved item id: {item_id}\nTags: {payload.get('tags','')}"
+    except Exception as e:
+        return f"❌ Error saving to Qdrant: {e}"
 
 # -------------------------
 # App logic: search
@@ -127,22 +131,22 @@ def search_items(query_image, query_text, limit: int = 5):
     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."
+        return "No results found."
 
     results = []
     for h in hits:
         payload = h.payload or {}
         score = getattr(h, "score", None)
         results.append(
-            f"id:{h.id} score:{float(score) if score else None} mode:{payload.get('mode','')} tags:{payload.get('tags','')} text:{payload.get('text','')}"
+            f"id:{h.id} | score:{float(score) if score else None:.4f} | mode:{payload.get('mode','')} | tags:{payload.get('tags','')} | text:{payload.get('text','')}"
         )
     return "\n\n".join(results)
 
@@ -150,19 +154,19 @@ def search_items(query_image, query_text, limit: int = 5):
 # Gradio UI
 # -------------------------
 with gr.Blocks(title="Lost & Found — Simple Helper") as demo:
-    gr.Markdown("## Lost & Found Helper (image/text search) — upload items, then search by image or text.")
+    gr.Markdown("## 🧳 Lost & Found Helper — Upload items, then search by image or text.")
     with gr.Row():
         with gr.Column():
             mode = gr.Radio(choices=["lost", "found"], value="lost", label="Add as")
             upload_img = gr.Image(type="pil", label="Item photo (optional)")
             text_desc = gr.Textbox(lines=2, placeholder="Short description (e.g. 'black backpack with blue zipper')", label="Description (optional)")
-            add_btn = gr.Button("Add item")
+            add_btn = gr.Button("➕ Add item")
             add_out = gr.Textbox(label="Add result", interactive=False)
         with gr.Column():
-            gr.Markdown("### Search")
+            gr.Markdown("### 🔍 Search")
             query_img = gr.Image(type="pil", label="Search by image (optional)")
             query_text = gr.Textbox(lines=2, label="Search by text (optional)")
-            search_btn = gr.Button("Search")
+            search_btn = gr.Button("🔎 Search")
             search_out = gr.Textbox(label="Search results", interactive=False)
 
     add_btn.click(add_item, inputs=[mode, upload_img, text_desc], outputs=[add_out])