Update shopify_client.py

shopify_client.py

@@ -3,36 +3,40 @@
 import os
 import shopify
 
-# Read credentials from env (set these in HF Space Settings → Secrets)
+# ─── Configuration ─────────────────────────────────────────────────────────────
 API_KEY      = os.getenv("SHOPIFY_API_KEY")
 ADMIN_TOKEN  = os.getenv("SHOPIFY_ADMIN_TOKEN")
 STORE_DOMAIN = os.getenv("SHOPIFY_STORE_DOMAIN")  # e.g. "jc23eb-c7.myshopify.com"
-API_VERSION  = os.getenv("SHOPIFY_API_VERSION", "2023-10")
+API_VERSION  = "2023-10"
 
-if not all([API_KEY, ADMIN_TOKEN, STORE_DOMAIN]):
-    raise ValueError("Missing one of SHOPIFY_API_KEY, SHOPIFY_ADMIN_TOKEN, or SHOPIFY_STORE_DOMAIN")
+if not (API_KEY and ADMIN_TOKEN and STORE_DOMAIN):
+    raise RuntimeError("Shopify credentials not set in environment variables")
 
-# Build the session URL
-session_url = f"https://{API_KEY}:{ADMIN_TOKEN}@{STORE_DOMAIN}/admin/api/{API_VERSION}"
-shopify.ShopifyResource.set_site(session_url)
+# Build the base URL for the Admin API
+SHOP_URL = f"https://{API_KEY}:{ADMIN_TOKEN}@{STORE_DOMAIN}/admin/api/{API_VERSION}"
+shopify.ShopifyResource.set_site(SHOP_URL)
 
-def create_product(title: str, body_html: str, price: str, image_url: str):
+# ─── Public Interface ───────────────────────────────────────────────────────────
+def create_shopify_product(title: str, description: str, price: str, image_url: str = None) -> str:
     """
-    Create a new Shopify product.
+    Creates a new product in Shopify with the given title, description, and price.
+    Optionally uploads a single image via URL.
+    Returns the public product URL on your store.
     """
+    # 1) Build the new product object
     product = shopify.Product()
     product.title = title
-    product.body_html = body_html
+    product.body_html = description
     product.variants = [shopify.Variant({"price": price})]
-    product.images   = [shopify.Image({"src": image_url})]
+    if image_url:
+        product.images = [shopify.Image({"src": image_url})]
+
+    # 2) Save to Shopify
     success = product.save()
     if not success:
-        raise Exception(f"Shopify error: {product.errors.full_messages()}")
-    return product.to_dict()
+        errors = product.errors.full_messages()
+        raise Exception(f"Shopify error: {errors}")
 
-def fetch_recent_orders(limit: int = 5):
-    """
-    Return a list of the most recent orders.
-    """
-    orders = shopify.Order.find(limit=limit, status="any")
-    return [o.to_dict() for o in orders]
+    # 3) Construct the storefront URL from the product handle
+    handle = product.handle
+    return f"https://{STORE_DOMAIN}/products/{handle}"