add convenience buttons

app.py

@@ -255,17 +255,45 @@ 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)
+
+def run_multi(queries, capacity, refresh):
+    """Execute multiple queries and merge all results."""
+    dfs = []
+    for qtext in queries:
+        print(f"⚡ Running convenience subquery: {qtext}")
+        df, _ = query_inventory(qtext, "", "", 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"])
+    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')}"
+
 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")
+        btn_temps = gr.Button("💣 Temps")
+
     query_box = gr.Textbox(label="Search (semantic, e.g. 'flowers in England')", elem_id="qbox")
     category_drop = gr.Dropdown(label="Category (optional exact match)", choices=[""] + ALL_CATEGORIES, elem_id="catdrop")
     country_box = gr.Textbox(label="Country (optional, e.g. UK, Cayman, Japan)", elem_id="countrybox")
     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 capacity → backend init
+    # Hidden bridge for JS sync
     cap_bridge = gr.Textbox(visible=False, elem_id="cap_bridge")
 
     def apply_bridge(cap_text):
@@ -275,9 +303,8 @@ with gr.Blocks(title="🧳 Torn Inventory Viewer") as iface:
                 return gr.Number.update(value=v)
         except:
             pass
-        return gr.Number.update()  # no change
+        return gr.Number.update()
 
-    # When the bridge receives a value, update the slider (so backend sees it)
     cap_bridge.change(apply_bridge, inputs=[cap_bridge], outputs=[capacity_slider])
 
     run_btn = gr.Button("🔍 Search / Refresh")
@@ -288,7 +315,75 @@ with gr.Blocks(title="🧳 Torn Inventory Viewer") as iface:
                   inputs=[query_box, category_drop, country_box, capacity_slider, refresh_check],
                   outputs=[result_df, meta_box])
 
-    # --- Shadow-DOM aware JS: restore capacity & push into hidden bridge BEFORE first search ---
+    # --- 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 ---
     gr.HTML("""
     <script>
     (function () {
@@ -302,21 +397,15 @@ with gr.Blocks(title="🧳 Torn Inventory Viewer") as iface:
         const saved = localStorage.getItem('travel_capacity');
         const capField = el('#cap_num input[type=number]');
         const bridge = el('#cap_bridge textarea');
-
         if (saved && capField) {
-          // Update visible field
           capField.value = saved;
-          // Update frontend state
           capField.dispatchEvent(new Event('input', { bubbles: true }));
-          // Push into hidden bridge -> triggers backend .change -> updates slider value server-side
           if (bridge) {
             bridge.value = saved;
             bridge.dispatchEvent(new Event('input', { bubbles: true }));
             bridge.dispatchEvent(new Event('change', { bubbles: true }));
           }
         }
-
-        // Save on changes
         if (capField && !capField.dataset.synced) {
           capField.dataset.synced = '1';
           capField.addEventListener('input', () => {
@@ -324,12 +413,7 @@ with gr.Blocks(title="🧳 Torn Inventory Viewer") as iface:
           });
         }
       }
-
-      window.addEventListener('DOMContentLoaded', () => {
-        setTimeout(restoreCapacity, 500);
-      });
-
-      // Keep trying on re-renders
+      window.addEventListener('DOMContentLoaded', () => setTimeout(restoreCapacity, 500));
       const obs = new MutationObserver(() => setTimeout(restoreCapacity, 100));
       obs.observe(document.documentElement, { childList: true, subtree: true });
     })();