Update app.py

app.py

@@ -3,14 +3,8 @@
 # Dataset: ckandemir/amazon-products (Hugging Face)
 # UI: Gradio (English)
 #
-# Requirements (requirements.txt):
-# gradio>=4.44.0
-# datasets>=3.0.0
-# pandas>=2.2.2
-# numpy>=1.26.4
-# sentence-transformers>=3.0.1
-# faiss-cpu>=1.8.0
-# tabulate>=0.9.0
+# Requirements (see requirements.txt):
+# gradio, datasets, pandas, numpy, sentence-transformers, faiss-cpu, tabulate
 
 import os, re, random
 from typing import Dict, List, Tuple
@@ -226,13 +220,14 @@ def recommend_topk_embeddings(profile: Dict, model_key: str, k: int=3) -> pd.Dat
     order = np.argsort(-sims)  # descending similarity
     seen, picks = set(), []
     for gi in idxs[order]:
-        if gi not in df_f.index:  # keep only allowed subset
+        gi = int(gi)
+        if gi not in df_f.index:
             continue
-        nm = CATALOG.loc[int(gi), "name"]
+        nm = CATALOG.loc[gi, "name"]
         if nm in seen:
             continue
         seen.add(nm)
-        picks.append(int(gi))
+        picks.append(gi)
         if len(picks) >= k:
             break
 
@@ -290,27 +285,24 @@ def generate_message(profile: Dict) -> str:
 
 # ========================= Gradio UI =========================
 EXAMPLES = [
-    [["tech","music"], "birthday", [20, 60], "Noa", "adult (18–64)", "MiniLM (384d)", "warm and friendly"],
-    [["home","cooking","practical"], "housewarming", [25, 45], "Daniel", "adult (18–64)", "MiniLM (384d)", "warm"],
-    [["games","photography"], "birthday", [30, 120], "Omer", "teen (13–17)", "MPNet (768d)", "fun"],
-    [["reading","design","aesthetic"], "thank_you", [15, 35], "Maya", "any", "E5-base (768d)", "friendly"],
+    [["tech","music"], "birthday", 20, 60, "Noa", "adult (18–64)", "MiniLM (384d)", "warm and friendly"],
+    [["home","cooking","practical"], "housewarming", 25, 45, "Daniel", "adult (18–64)", "MiniLM (384d)", "warm"],
+    [["games","photography"], "birthday", 30, 120, "Omer", "teen (13–17)", "MPNet (768d)", "fun"],
+    [["reading","design","aesthetic"], "thank_you", 15, 35, "Maya", "any", "E5-base (768d)", "friendly"],
 ]
 
 def safe_markdown_table(df: pd.DataFrame) -> str:
     try:
         return df.to_markdown(index=False)
     except Exception:
-        # fallback if tabulate is missing
         return df.to_string(index=False)
 
-def ui_predict(interests_list: List[str], occasion: str, budget_range, recipient_name: str,
-               age_label: str, model_key: str, tone: str):
+def ui_predict(interests_list: List[str], occasion: str, budget_min: float, budget_max: float,
+               recipient_name: str, age_label: str, model_key: str, tone: str):
     try:
-        # Parse budget range [min, max]
-        if isinstance(budget_range, (list, tuple)) and len(budget_range) == 2:
-            budget_min, budget_max = float(budget_range[0]), float(budget_range[1])
-        else:
-            budget_min, budget_max = 20.0, 60.0
+        # sanity
+        if budget_min is None: budget_min = 20.0
+        if budget_max is None: budget_max = 60.0
         if budget_min > budget_max:
             budget_min, budget_max = budget_max, budget_min
 
@@ -319,9 +311,9 @@ def ui_predict(interests_list: List[str], occasion: str, budget_range, recipient
             "recipient_name": recipient_name or "Friend",
             "interests": interests_list or [],
             "occasion": occasion or "birthday",
-            "budget_min": budget_min,
-            "budget_max": budget_max,
-            "budget_usd": budget_max,
+            "budget_min": float(budget_min),
+            "budget_max": float(budget_max),
+            "budget_usd": float(budget_max),
             "age_range": age_range,
             "tone": tone or "warm and friendly",
         }
@@ -351,7 +343,10 @@ with gr.Blocks() as demo:
         age = gr.Dropdown(label="Age group", choices=list(AGE_OPTIONS.keys()), value="adult (18–64)")
         model = gr.Dropdown(label="Embedding model", choices=list(MODEL_CHOICES.keys()), value="MiniLM (384d)")
 
-    budget = gr.RangeSlider(label="Budget range (USD)", minimum=5, maximum=500, step=1, value=[20, 60])
+    # Two sliders (for older Gradio versions): min + max budget
+    with gr.Row():
+        budget_min = gr.Slider(label="Min budget (USD)", minimum=5, maximum=500, step=1, value=20)
+        budget_max = gr.Slider(label="Max budget (USD)", minimum=5, maximum=500, step=1, value=60)
 
     with gr.Row():
         recipient_name = gr.Textbox(label="Recipient name", value="Noa")
@@ -364,13 +359,13 @@ with gr.Blocks() as demo:
 
     gr.Examples(
         EXAMPLES,
-        [interests, occasion, budget, recipient_name, age, model, tone],
+        [interests, occasion, budget_min, budget_max, recipient_name, age, model, tone],
         label="Quick examples",
     )
 
     go.click(
         ui_predict,
-        [interests, occasion, budget, recipient_name, age, model, tone],
+        [interests, occasion, budget_min, budget_max, recipient_name, age, model, tone],
         [out_top3, out_gen, out_msg]
     )