Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import time
 import json
+import tempfile
 from typing import Dict, Any, List, Generator, Tuple
 
 import requests
@@ -41,10 +42,22 @@ def _log(logs: List[str], msg: str):
     logs.append(f"[{_now()}] {msg}")
 
 
-def _find_first_valid_pair(logs: List[str]) -> Dict[str, Any]:
+def _list_shops(logs: List[str]) -> List[Dict[str, Any]]:
+    _log(logs, "Listing shops")
+    shops = _req("GET", "/v1/shops.json")
+    if not isinstance(shops, list):
+        raise RuntimeError("Unexpected shops response.")
+    _log(logs, f"FOUND {len(shops)} shops")
+    return shops
+
+
+def _collect_phase_a(shop_id: int, logs: List[str]) -> Dict[str, Any]:
     _log(logs, "Listing blueprints")
     blueprints = _req("GET", "/v1/catalog/blueprints.json")
 
+    templates: List[Dict[str, Any]] = []
+    scanned_pairs = 0
+
     for bp in blueprints:
         bp_id = str(bp["id"])
         _log(logs, f"Blueprint {bp_id}: fetching providers")
@@ -52,6 +65,7 @@ def _find_first_valid_pair(logs: List[str]) -> Dict[str, Any]:
         providers = _req("GET", f"/v1/catalog/blueprints/{bp_id}/print_providers.json")
         for p in providers:
             p_id = str(p["id"])
+            scanned_pairs += 1
             _log(logs, f"Blueprint {bp_id} / Provider {p_id}: fetching variants")
 
             vr = _req(
@@ -59,22 +73,32 @@ def _find_first_valid_pair(logs: List[str]) -> Dict[str, Any]:
                 f"/v1/catalog/blueprints/{bp_id}/print_providers/{p_id}/variants.json?show-out-of-stock=1",
             )
             variants = vr.get("variants")
-            if isinstance(variants, list) and variants:
-                _log(logs, f"FOUND {len(variants)} variants")
-
-                return {
-                    "blueprint": bp,
-                    "provider": p,
-                    "variants": variants,
-                    "blueprintDetails": _req("GET", f"/v1/catalog/blueprints/{bp_id}.json"),
-                    "providerDetails": _req("GET", f"/v1/catalog/print_providers/{p_id}.json"),
-                    "shippingInfo": _req(
-                        "GET",
-                        f"/v1/catalog/blueprints/{bp_id}/print_providers/{p_id}/shipping.json",
-                    ),
-                }
-
-    raise RuntimeError("No blueprint/provider pair with variants found.")
+            if not (isinstance(variants, list) and variants):
+                continue
+
+            _log(logs, f"FOUND {len(variants)} variants")
+
+            blob = {
+                "blueprint": bp,
+                "provider": p,
+                "variants": variants,
+                "blueprintDetails": _req("GET", f"/v1/catalog/blueprints/{bp_id}.json"),
+                "providerDetails": _req("GET", f"/v1/catalog/print_providers/{p_id}.json"),
+                "shippingInfo": _req(
+                    "GET",
+                    f"/v1/catalog/blueprints/{bp_id}/print_providers/{p_id}/shipping.json",
+                ),
+            }
+
+            templates.append({
+                "shop_id": shop_id,
+                "blueprint_id": int(bp_id),
+                "print_provider_id": int(p_id),
+                "raw": blob,
+            })
+
+    _log(logs, f"PHASE_A_DONE templates={len(templates)} scanned_pairs={scanned_pairs}")
+    return {"shop_id": shop_id, "templates": templates}
 
 
 def _build_product(blob: Dict[str, Any], currency: str, logs: List[str]) -> Dict[str, Any]:
@@ -91,6 +115,7 @@ def _build_product(blob: Dict[str, Any], currency: str, logs: List[str]) -> Dict
             "color": (v.get("options") or {}).get("color"),
             "priceCents": cents,
             "price": round(cents / 100, 2) if cents is not None else None,
+            "placeholders": v.get("placeholders") or [],
         })
 
     _log(logs, f"VARIANT_SNAPSHOT sample={json.dumps(snapshot[:20])}")
@@ -125,21 +150,60 @@ def _build_product(blob: Dict[str, Any], currency: str, logs: List[str]) -> Dict
     }
 
 
-def run(currency: str) -> Generator[Tuple[str, str], None, None]:
+def fetch_shops() -> Tuple[str, gr.Dropdown]:
+    logs: List[str] = []
+    shops = _list_shops(logs)
+
+    choices = []
+    default_val = None
+    for s in shops:
+        sid = s.get("id")
+        title = s.get("title") or "untitled"
+        chan = s.get("sales_channel") or "unknown"
+        label = f"{title} ({chan}) — {sid}"
+        val = str(sid)
+        choices.append((label, val))
+        if (title or "").strip().lower() == "atheria":
+            default_val = val
+
+    return "\n".join(logs), gr.Dropdown.update(choices=choices, value=default_val)
+
+
+def run(currency: str, shop_id: str) -> Generator[Tuple[str, str, str], None, None]:
     logs: List[str] = []
     result: Dict[str, Any] = {}
+    dl_path = ""
 
     def flush():
-        return "\n".join(logs), json.dumps(result, indent=2)
+        return "\n".join(logs), json.dumps(result, indent=2), dl_path
 
     try:
         _log(logs, "START")
         yield flush()
 
-        blob = _find_first_valid_pair(logs)
+        if not shop_id:
+            raise RuntimeError("Missing shop_id. Click Fetch Shops and select Atheria.")
+
+        sid = int(str(shop_id).strip())
+        _log(logs, f"USING_SHOP_ID {sid}")
+        yield flush()
+
+        phase_a = _collect_phase_a(sid, logs)
         yield flush()
 
-        result = _build_product(blob, currency or "USD", logs)
+        templates = phase_a.get("templates") or []
+        if templates:
+            first_raw = (templates[0].get("raw") or {})
+            result = _build_product(first_raw, currency or "USD", logs)
+        else:
+            result = {"shop_id": sid, "templates": []}
+
+        fd, path = tempfile.mkstemp(prefix="phase_a_", suffix=".json")
+        os.close(fd)
+        with open(path, "w", encoding="utf-8") as f:
+            json.dump(phase_a, f, indent=2)
+        dl_path = path
+
         _log(logs, "DONE")
         yield flush()
 
@@ -153,10 +217,17 @@ with gr.Blocks(title="Printify Catalog Probe") as demo:
     gr.Markdown("Extract and normalize ONE Printify blueprint/provider into a structured JSON object.")
 
     currency = gr.Textbox(label="Currency", value="USD")
+
+    fetch_btn = gr.Button("Fetch Shops")
+    shops_log = gr.Textbox(label="Shop Logs", lines=6)
+    shop = gr.Dropdown(label="Shop (select Atheria)", choices=[])
+
     btn = gr.Button("Run")
     logs = gr.Textbox(label="Logs", lines=18)
     out = gr.Textbox(label="Output JSON", lines=18)
+    dl = gr.File(label="Download Phase A JSON")
 
-    btn.click(run, inputs=[currency], outputs=[logs, out])
+    fetch_btn.click(fetch_shops, inputs=[], outputs=[shops_log, shop])
+    btn.click(run, inputs=[currency, shop], outputs=[logs, out, dl])
 
 demo.queue(concurrency_count=1).launch()