hackerloi45 commited on
Commit
3055619
Β·
1 Parent(s): 2b16ad8

fixed the bugs

Browse files
Files changed (1) hide show
  1. app.py +13 -4
app.py CHANGED
@@ -125,7 +125,7 @@ def add_item(mode: str, uploaded_image, text_description: str):
125
  # -------------------------
126
  # App logic: search
127
  # -------------------------
128
- def search_items(query_image, query_text, limit: int = 5):
129
  if query_image is not None:
130
  qvec = embed_image_pil(query_image).tolist()
131
  elif query_text and len(query_text.strip()) > 0:
@@ -143,11 +143,18 @@ def search_items(query_image, query_text, limit: int = 5):
143
 
144
  results = []
145
  for h in hits:
146
- payload = h.payload or {}
147
  score = getattr(h, "score", None)
 
 
 
 
148
  results.append(
149
- f"id:{h.id} | score:{float(score) if score else None:.4f} | mode:{payload.get('mode','')} | tags:{payload.get('tags','')} | text:{payload.get('text','')}"
150
  )
 
 
 
 
151
  return "\n\n".join(results)
152
 
153
  # -------------------------
@@ -166,11 +173,13 @@ with gr.Blocks(title="Lost & Found β€” Simple Helper") as demo:
166
  gr.Markdown("### πŸ” Search")
167
  query_img = gr.Image(type="pil", label="Search by image (optional)")
168
  query_text = gr.Textbox(lines=2, label="Search by text (optional)")
 
 
169
  search_btn = gr.Button("πŸ”Ž Search")
170
  search_out = gr.Textbox(label="Search results", interactive=False)
171
 
172
  add_btn.click(add_item, inputs=[mode, upload_img, text_desc], outputs=[add_out])
173
- search_btn.click(search_items, inputs=[query_img, query_text], outputs=[search_out])
174
 
175
  if __name__ == "__main__":
176
  demo.launch(server_name="0.0.0.0", server_port=7860)
 
125
  # -------------------------
126
  # App logic: search
127
  # -------------------------
128
+ def search_items(query_image, query_text, limit: int = 5, min_score: float = 0.90):
129
  if query_image is not None:
130
  qvec = embed_image_pil(query_image).tolist()
131
  elif query_text and len(query_text.strip()) > 0:
 
143
 
144
  results = []
145
  for h in hits:
 
146
  score = getattr(h, "score", None)
147
+ if score is None or score < min_score: # filter by threshold
148
+ continue
149
+
150
+ payload = h.payload or {}
151
  results.append(
152
+ f"id:{h.id} | score:{score:.4f} | mode:{payload.get('mode','')} | tags:{payload.get('tags','')} | text:{payload.get('text','')}"
153
  )
154
+
155
+ if not results:
156
+ return f"No results above similarity threshold {min_score:.2f}"
157
+
158
  return "\n\n".join(results)
159
 
160
  # -------------------------
 
173
  gr.Markdown("### πŸ” Search")
174
  query_img = gr.Image(type="pil", label="Search by image (optional)")
175
  query_text = gr.Textbox(lines=2, label="Search by text (optional)")
176
+ limit_slider = gr.Slider(1, 10, value=5, step=1, label="Max results")
177
+ score_slider = gr.Slider(0.0, 1.0, value=0.90, step=0.01, label="Min similarity score")
178
  search_btn = gr.Button("πŸ”Ž Search")
179
  search_out = gr.Textbox(label="Search results", interactive=False)
180
 
181
  add_btn.click(add_item, inputs=[mode, upload_img, text_desc], outputs=[add_out])
182
+ search_btn.click(search_items, inputs=[query_img, query_text, limit_slider, score_slider], outputs=[search_out])
183
 
184
  if __name__ == "__main__":
185
  demo.launch(server_name="0.0.0.0", server_port=7860)