try fix again

app.py

@@ -186,6 +186,12 @@ def query_inventory(query_text="", category="", country_name="", capacity=10, re
     stocks = data.get("stocks", {})
     rows = []
 
+    # --- Build per-country catalogs from live payload ---
+    COUNTRY_CATALOG = {
+        code.upper(): {i["name"] for i in cdata.get("stocks", [])}
+        for code, cdata in stocks.items()
+    }
+
     parsed_item, parsed_country = parse_freeform_query(query_text)
     if not country_name and parsed_country:
         country_name = parsed_country
@@ -200,7 +206,7 @@ def query_inventory(query_text="", category="", country_name="", capacity=10, re
         code = code_raw.upper()
         cname = COUNTRY_NAMES.get(code, code)
 
-        # Filter by requested country
+        # Strict country filter
         if country_name:
             if user_code:
                 if code != user_code:
@@ -212,10 +218,10 @@ def query_inventory(query_text="", category="", country_name="", capacity=10, re
         update_str = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(update_ts)) if update_ts else "Unknown"
 
         live_lookup = {i["name"]: i for i in cdata.get("stocks", [])}
+        candidate_names = sorted(COUNTRY_CATALOG.get(code, set()))  # ONLY this country's items
 
-        # Merge known items (include 0-qty)
-        for iname, itype in ITEM_TO_TYPE.items():
-            itype = itype.lower()
+        for iname in candidate_names:
+            itype = ITEM_TO_TYPE.get(iname, "").lower()
             item_data = live_lookup.get(iname, {"quantity": 0, "cost": 0})
             qty = item_data.get("quantity", 0)
             cost = item_data.get("cost", 0)
@@ -235,7 +241,7 @@ def query_inventory(query_text="", category="", country_name="", capacity=10, re
                 rows.append({
                     "Country": cname,
                     "Item": iname,
-                    "Category": itype.title(),
+                    "Category": itype.title() if itype else "",
                     "Quantity": qty,
                     "Cost": cost,
                     "Max Capacity Cost": cost * capacity,
@@ -243,10 +249,10 @@ def query_inventory(query_text="", category="", country_name="", capacity=10, re
                 })
 
     if not rows:
+        print(f"⚠️ No matches for '{item_term or category or 'all items'}' in {country_name or 'ALL'}")
         return pd.DataFrame([{"Result": "No inventory found for that query."}]), f"Last update: {last_update}"
 
-    df = pd.DataFrame(rows)
-    df = df.sort_values(by=["Country", "Item"])
+    df = pd.DataFrame(rows).sort_values(by=["Country", "Item"])
     for col in ["Quantity", "Cost", "Max Capacity Cost"]:
         if col in df.columns:
             df[col] = df[col].apply(lambda x: f"{x:,.0f}" if isinstance(x, (int, float)) and x != "" else x)
@@ -278,24 +284,42 @@ with gr.Blocks(title="🧳 Torn Inventory Viewer") as iface:
     # --- JS fix for persistent capacity ---
     gr.HTML("""
     <script>
-    window.addEventListener('DOMContentLoaded', () => {
-      const saved = localStorage.getItem('travel_capacity');
-      const capField = document.querySelector('input[type=number]');
-      if (saved && capField) {
-        capField.value = saved;
-        capField.dispatchEvent(new Event('input', { bubbles: true }));
+    (function () {
+      function root() {
+        const app = document.querySelector('gradio-app');
+        return app && app.shadowRoot ? app.shadowRoot : document;
+      }
+      function setCapacity(val) {
+        const r = root();
+        const field = r.querySelector('input[type=number]');
+        if (!field) return false;
+        field.value = val;
+        field.dispatchEvent(new Event('input', { bubbles: true }));
+        return true;
       }
-      const observer = new MutationObserver(() => {
-        const field = document.querySelector('input[type=number]');
-        if (field && !field.dataset.synced) {
-          field.dataset.synced = "1";
-          field.addEventListener('input', () => {
-            localStorage.setItem('travel_capacity', field.value);
-          });
-        }
+      function restoreOnce() {
+        const saved = localStorage.getItem('travel_capacity');
+        if (!saved) return;
+        if (setCapacity(saved)) console.log('🧳 capacity restored:', saved);
+      }
+      function attachSaver() {
+        const r = root();
+        const field = r.querySelector('input[type=number]');
+        if (!field || field.dataset.tcListen === '1') return;
+        field.dataset.tcListen = '1';
+        field.addEventListener('input', () => {
+          localStorage.setItem('travel_capacity', field.value);
+        });
+      }
+      window.addEventListener('DOMContentLoaded', () => {
+        setTimeout(() => { restoreOnce(); attachSaver(); }, 500);
+      });
+      const obs = new MutationObserver(() => {
+        restoreOnce();
+        attachSaver();
       });
-      observer.observe(document.body, { childList: true, subtree: true });
-    });
+      obs.observe(document.documentElement, { childList: true, subtree: true });
+    })();
     </script>
     """)