Update app.py

app.py

@@ -113,13 +113,19 @@ def clear_all():
 
 
 # ── Tab 2: Destiny Search ────────────────────────────────────────────────────
-def destiny_search(query: str, filter_1word: bool, filter_2words: bool, top_k: int):
+def destiny_search(query: str, dob: str, filter_1word: bool, filter_2words: bool, top_k: int):
     if not query.strip():
-        return build_search_html([], query, error="⚠️ Please enter a name to search for your destiny!")
+        return build_search_html([], query, error="⚠️ Please enter a name to search!")
 
-    query_vec = model.encode(query.strip(), convert_to_tensor=False, device=device).tolist()
+    # Combine name + dob
+    query_text = query.strip()
+    dob = dob.strip()
+    if dob:
+        query_text = f"{query_text} {dob}"
+
+    query_vec = model.encode(query_text, convert_to_tensor=False, device=device).tolist()
 
-    # Build metadata filter
+    # Metadata filter
     if filter_1word and filter_2words:
         meta_filter = {"num_words": {"$in": [1, 2]}}
     elif filter_1word:
@@ -127,20 +133,33 @@ def destiny_search(query: str, filter_1word: bool, filter_2words: bool, top_k: i
     elif filter_2words:
         meta_filter = {"num_words": {"$eq": 2}}
     else:
-        meta_filter = None  # no filter
+        meta_filter = None
 
     query_kwargs = dict(
         vector=query_vec,
-        top_k=top_k,
+        top_k=top_k + 5,  # small buffer to compensate filtering
         include_metadata=True,
     )
+
     if meta_filter:
         query_kwargs["filter"] = meta_filter
 
     response = index.query(**query_kwargs)
     matches = response.get("matches", [])
 
-    return build_search_html(matches, query)
+    # 🔥 Remove input name from results
+    query_clean = query.strip().lower()
+    filtered_matches = []
+
+    for m in matches:
+        text = m.get("metadata", {}).get("text", "").strip().lower()
+        if text != query_clean:
+            filtered_matches.append(m)
+
+    # Trim back to top_k
+    filtered_matches = filtered_matches[:top_k]
+
+    return build_search_html(filtered_matches, query)
 
 
 def build_search_html(matches: list, query: str, error: str = "") -> str:
@@ -210,7 +229,7 @@ def build_search_html(matches: list, query: str, error: str = "") -> str:
 
 
 def clear_search():
-    return "", gr.update(value=""), True, True, 10
+    return "", "", gr.update(value=""), True, True, 10
 
 
 # ── CSS ──────────────────────────────────────────────────────────────────────
@@ -437,6 +456,12 @@ with gr.Blocks(css=custom_css, title="LatentLove 💕") as demo:
                     lines=1,
                 )
 
+                search_dob = gr.Textbox(
+                    label="Date of Birth (optional)",
+                    placeholder="dd/mm/yyyy",
+                    lines=1,
+                )
+
                 with gr.Row():
                     cb_1word  = gr.Checkbox(label="1-word names", value=True)
                     cb_2words = gr.Checkbox(label="2-word names", value=True)
@@ -488,14 +513,14 @@ with gr.Blocks(css=custom_css, title="LatentLove 💕") as demo:
 
     search_btn.click(
         fn=destiny_search,
-        inputs=[search_query, cb_1word, cb_2words, top_k_slider],
+        inputs=[search_query, search_dob, cb_1word, cb_2words, top_k_slider],
         outputs=[search_results],
     )
 
     clear_search_btn.click(
         fn=clear_search,
         inputs=[],
-        outputs=[search_query, search_results, cb_1word, cb_2words, top_k_slider],
+        outputs=[search_query, search_dob, search_results, cb_1word, cb_2words, top_k_slider],
     )