hide categories that aren't sold abroad

app.py

@@ -19,6 +19,7 @@ with open("items.json", "r", encoding="utf-8") as f:
     items_data = json.load(f)["items"]
 
 ITEM_TO_TYPE = {v["name"]: v["type"].lower() for v in items_data.values() if "name" in v and "type" in v}
+ALL_ITEMS = list(ITEM_TO_TYPE.keys())
 ALL_CATEGORIES = sorted(set(ITEM_TO_TYPE.values()))
 ITEM_FILE_MTIME = os.path.getmtime("items.json")
 
@@ -177,6 +178,17 @@ def fetch_yata(force_refresh=False):
         print(f"❌ Fetch error: {e}")
         return {"stocks": {}}, "Fetch failed"
 
+def get_live_categories(data):
+    """Return sorted list of categories present in YATA data."""
+    live_cats = set()
+    for _, cdata in data.get("stocks", {}).items():
+        for item in cdata.get("stocks", []):
+            name = item.get("name")
+            cat = ITEM_TO_TYPE.get(name)
+            if cat:
+                live_cats.add(cat.lower())
+    return sorted(live_cats)
+
 # ---------------- Core logic ----------------
 def query_inventory(query_text="", category="", country_name="", capacity=10, refresh=False):
     print(f"🔎 Query received: '{query_text}', category='{category}', country='{country_name}', cap={capacity}")
@@ -184,7 +196,6 @@ def query_inventory(query_text="", category="", country_name="", capacity=10, re
     stocks = data.get("stocks", {})
     rows = []
 
-    # Parse query
     parsed_item, parsed_country = parse_freeform_query(query_text)
     if not country_name and parsed_country:
         country_name = parsed_country
@@ -193,7 +204,6 @@ def query_inventory(query_text="", category="", country_name="", capacity=10, re
     semantic_items = semantic_result["items"]
     semantic_category = semantic_result["category"]
 
-    # Country gating (strict)
     user_code = normalize_country_query(country_name)
 
     for code_raw, cdata in stocks.items():
@@ -210,14 +220,12 @@ def query_inventory(query_text="", category="", country_name="", capacity=10, re
         update_ts = cdata.get("update")
         update_str = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(update_ts)) if update_ts else "Unknown"
 
-        # ONLY iterate this country's actual items (no global catalog)
         for item in cdata.get("stocks", []):
             iname = item.get("name", "")
             itype = ITEM_TO_TYPE.get(iname, "").lower()
             qty = item.get("quantity", 0)
             cost = item.get("cost", 0)
 
-            # Item filter
             if item_term:
                 item_ok = (
                     item_term.lower() in iname.lower()
@@ -236,7 +244,6 @@ def query_inventory(query_text="", category="", country_name="", capacity=10, re
                     "Category": itype.title() if itype else "",
                     "Quantity": qty,
                     "Cost": cost,
-                    # IMPORTANT: use the function arg 'capacity' directly
                     "Max Capacity Cost": cost * capacity,
                     "Updated": update_str,
                 })
@@ -250,16 +257,15 @@ def query_inventory(query_text="", category="", country_name="", capacity=10, re
             df[col] = df[col].apply(lambda x: f"{x:,.0f}" if isinstance(x, (int, float)) and x != "" else x)
     return df, f"Last update: {last_update}"
 
-# ---------------- Wrapper ----------------
+# ---------------- Wrappers ----------------
 def run_query(query_text, category, country, capacity, refresh):
-    return query_inventory(query_text, category, country, capacity, refresh)
-
-# ---------------- Gradio UI ----------------
-def run_query(query_text, category, country, capacity, refresh):
-    return query_inventory(query_text, category, country, capacity, refresh)
+    data, _ = fetch_yata(force_refresh=refresh)
+    df, ts = query_inventory(query_text, category, country, capacity, refresh)
+    live_categories = get_live_categories(data)
+    return df, ts, gr.Dropdown.update(choices=[""] + live_categories)
 
 def run_multi(queries, capacity, refresh):
-    """Execute multiple queries and merge all results."""
+    data, _ = fetch_yata(force_refresh=refresh)
     dfs = []
     for qtext in queries:
         print(f"⚡ Running convenience subquery: {qtext}")
@@ -267,24 +273,24 @@ def run_multi(queries, capacity, refresh):
         if "Result" not in df.columns:
             dfs.append(df)
     if not dfs:
-        return pd.DataFrame([{"Result": "No results for that set."}]), "No results."
-    merged = pd.concat(dfs, ignore_index=True)
-    merged = merged.sort_values(by=["Country", "Item"])
+        return pd.DataFrame([{"Result": "No results for that set."}]), "No results.", gr.Dropdown.update()
+    merged = pd.concat(dfs, ignore_index=True).sort_values(by=["Country", "Item"])
     for col in ["Quantity", "Cost", "Max Capacity Cost"]:
         if col in merged.columns:
             merged[col] = merged[col].apply(lambda x: f"{x:,.0f}" if isinstance(x, (int, float)) and x != "" else x)
-    return merged, f"Last update: {time.strftime('%Y-%m-%d %H:%M:%S')}"
+    live_categories = get_live_categories(data)
+    return merged, f"Last update: {time.strftime('%Y-%m-%d %H:%M:%S')}", gr.Dropdown.update(choices=[""] + live_categories)
 
+# ---------------- Gradio UI ----------------
 with gr.Blocks(title="🧳 Torn Inventory Viewer") as iface:
     gr.Markdown("## 🧳 Torn Inventory Viewer")
     gr.Markdown("_Search Torn YATA travel stocks with semantic matching._")
 
-    # --- Convenience Buttons ---
     with gr.Row():
         btn_short = gr.Button("🌸 Flushies (Short Haul)")
         btn_med = gr.Button("✈️ Flushies (Medium Haul)")
         btn_long = gr.Button("🛫 Flushies (Long Haul)")
-        btn_xanax = gr.Button("💊 Xanax (SA)")
+        btn_xanax = gr.Button("💊 Xanax")
         btn_temps = gr.Button("💣 Temps")
 
     query_box = gr.Textbox(label="Search (semantic, e.g. 'flowers in England')", elem_id="qbox")
@@ -293,7 +299,6 @@ with gr.Blocks(title="🧳 Torn Inventory Viewer") as iface:
     capacity_slider = gr.Number(label="Travel Capacity", value=10, minimum=5, maximum=88, precision=0, elem_id="cap_num")
     refresh_check = gr.Checkbox(label="Force refresh (ignore cache)", value=False)
 
-    # Hidden bridge for JS sync
     cap_bridge = gr.Textbox(visible=False, elem_id="cap_bridge")
 
     def apply_bridge(cap_text):
@@ -313,77 +318,38 @@ with gr.Blocks(title="🧳 Torn Inventory Viewer") as iface:
 
     run_btn.click(run_query,
                   inputs=[query_box, category_drop, country_box, capacity_slider, refresh_check],
-                  outputs=[result_df, meta_box])
-
-    # --- Convenience button handlers ---
-    btn_short.click(
-        fn=lambda capacity, refresh: run_multi(
-            [
-                "flowers in mexico",
-                "flowers in cayman islands",
-                "flowers in canada",
-                "plushies in mexico",
-                "plushies in cayman islands",
-                "plushies in canada",
-            ],
-            capacity, refresh),
-        inputs=[capacity_slider, refresh_check],
-        outputs=[result_df, meta_box],
-    )
-
-    btn_med.click(
-        fn=lambda capacity, refresh: run_multi(
-            [
-                "flowers in hawaii",
-                "flowers in united kingdom",
-                "flowers in argentina",
-                "flowers in switzerland",
-                "flowers in japan",
-                "plushies in hawaii",
-                "plushies in united kingdom",
-                "plushies in argentina",
-                "plushies in switzerland",
-                "plushies in japan",
-            ],
-            capacity, refresh),
-        inputs=[capacity_slider, refresh_check],
-        outputs=[result_df, meta_box],
-    )
-
-    btn_long.click(
-        fn=lambda capacity, refresh: run_multi(
-            [
-                "flowers in uae",
-                "flowers in china",
-                "flowers in south africa",
-                "plushies in uae",
-                "plushies in china",
-                "plushies in south africa",
-            ],
-            capacity, refresh),
-        inputs=[capacity_slider, refresh_check],
-        outputs=[result_df, meta_box],
-    )
-
-    btn_xanax.click(
-        fn=lambda capacity, refresh: run_multi(["xanax in south africa"], capacity, refresh),
-        inputs=[capacity_slider, refresh_check],
-        outputs=[result_df, meta_box],
-    )
-
-    btn_temps.click(
-        fn=lambda capacity, refresh: run_multi(
-            [
-                "tear gas in argentina",
-                "smoke grenade in south africa",
-                "flash grenade in switzerland",
-            ],
-            capacity, refresh),
-        inputs=[capacity_slider, refresh_check],
-        outputs=[result_df, meta_box],
-    )
-
-    # --- JS: restore capacity ---
+                  outputs=[result_df, meta_box, category_drop])
+
+    # Convenience buttons
+    btn_short.click(lambda c, r: run_multi(
+        ["flowers in mexico","flowers in cayman islands","flowers in canada",
+         "plushies in mexico","plushies in cayman islands","plushies in canada"],
+        c, r),
+        inputs=[capacity_slider, refresh_check], outputs=[result_df, meta_box, category_drop])
+
+    btn_med.click(lambda c, r: run_multi(
+        ["flowers in hawaii","flowers in united kingdom","flowers in argentina",
+         "flowers in switzerland","flowers in japan",
+         "plushies in hawaii","plushies in united kingdom","plushies in argentina",
+         "plushies in switzerland","plushies in japan"],
+        c, r),
+        inputs=[capacity_slider, refresh_check], outputs=[result_df, meta_box, category_drop])
+
+    btn_long.click(lambda c, r: run_multi(
+        ["flowers in uae","flowers in china","flowers in south africa",
+         "plushies in uae","plushies in china","plushies in south africa"],
+        c, r),
+        inputs=[capacity_slider, refresh_check], outputs=[result_df, meta_box, category_drop])
+
+    btn_xanax.click(lambda c, r: run_multi(["xanax in south africa"], c, r),
+        inputs=[capacity_slider, refresh_check], outputs=[result_df, meta_box, category_drop])
+
+    btn_temps.click(lambda c, r: run_multi(
+        ["tear gas in argentina","smoke grenade in south africa","flash grenade in switzerland"],
+        c, r),
+        inputs=[capacity_slider, refresh_check], outputs=[result_df, meta_box, category_drop])
+
+    # --- JS for capacity persistence ---
     gr.HTML("""
     <script>
     (function () {
@@ -392,29 +358,28 @@ with gr.Blocks(title="🧳 Torn Inventory Viewer") as iface:
         return app && app.shadowRoot ? app.shadowRoot : document;
       }
       function el(sel) { return root().querySelector(sel); }
-
-      function restoreCapacity() {
+      function restore() {
         const saved = localStorage.getItem('travel_capacity');
-        const capField = el('#cap_num input[type=number]');
+        const num = el('#cap_num input[type=number]');
         const bridge = el('#cap_bridge textarea');
-        if (saved && capField) {
-          capField.value = saved;
-          capField.dispatchEvent(new Event('input', { bubbles: true }));
+        if (saved && num) {
+          num.value = saved;
+          num.dispatchEvent(new Event('input', { bubbles: true }));
           if (bridge) {
             bridge.value = saved;
             bridge.dispatchEvent(new Event('input', { bubbles: true }));
             bridge.dispatchEvent(new Event('change', { bubbles: true }));
           }
         }
-        if (capField && !capField.dataset.synced) {
-          capField.dataset.synced = '1';
-          capField.addEventListener('input', () => {
-            localStorage.setItem('travel_capacity', capField.value);
+        if (num && !num.dataset.synced) {
+          num.dataset.synced = '1';
+          num.addEventListener('input', () => {
+            localStorage.setItem('travel_capacity', num.value);
           });
         }
       }
-      window.addEventListener('DOMContentLoaded', () => setTimeout(restoreCapacity, 500));
-      const obs = new MutationObserver(() => setTimeout(restoreCapacity, 100));
+      window.addEventListener('DOMContentLoaded', () => setTimeout(restore, 500));
+      const obs = new MutationObserver(() => setTimeout(restore, 100));
       obs.observe(document.documentElement, { childList: true, subtree: true });
     })();
     </script>