Update mailchimp_client.py

mailchimp_client.py

@@ -1,47 +1,47 @@
 # mailchimp_client.py
+
+import os
 import mailchimp_marketing as MailchimpMarketing
 from mailchimp_marketing.api_client import ApiClientError
-import os
 
-MAILCHIMP_API_KEY = os.getenv("MAILCHIMP_API_KEY")
-MAILCHIMP_SERVER = os.getenv("MAILCHIMP_SERVER")  # e.g., "us19"
-LIST_ID = os.getenv("MAILCHIMP_LIST_ID")          # create a list and paste its ID here
+# Read credentials from env (set these in HF Space Settings → Secrets)
+API_KEY   = os.getenv("MAILCHIMP_API_KEY")
+SERVER    = os.getenv("MAILCHIMP_SERVER")    # e.g. "us19"
+LIST_ID   = os.getenv("MAILCHIMP_LIST_ID")
+
+if not all([API_KEY, SERVER, LIST_ID]):
+    raise ValueError("Missing one of MAILCHIMP_API_KEY, MAILCHIMP_SERVER, or MAILCHIMP_LIST_ID")
 
 client = MailchimpMarketing.Client()
 client.set_config({
-    "api_key": MAILCHIMP_API_KEY,
-    "server": MAILCHIMP_SERVER
+    "api_key": API_KEY,
+    "server":  SERVER
 })
 
-def create_campaign(subject: str, html_content: str):
+def create_and_send_campaign(subject: str, html_content: str):
+    """
+    Create a Mailchimp campaign and send it immediately.
+    """
     try:
+        # 1. Create campaign
         campaign = client.campaigns.create({
             "type": "regular",
             "recipients": {"list_id": LIST_ID},
             "settings": {
                 "subject_line": subject,
-                "title": subject,
-                "from_name": "AutoExec AI",
-                "reply_to": "no‑reply@yourdomain.com"
+                "title":        subject,
+                "from_name":    "AutoExec AI",
+                "reply_to":     "no-reply@yourdomain.com"
             }
         })
-        client.campaigns.set_content(campaign["id"], {"html": html_content})
-        return campaign["id"]
-    except ApiClientError as error:
-        print("Error creating campaign: {}".format(error.text))
-        raise
+        camp_id = campaign["id"]
 
-def send_campaign(campaign_id: str):
-    try:
-        client.campaigns.send(campaign_id)
-    except ApiClientError as error:
-        print("Error sending campaign: {}".format(error.text))
-        raise
-
-if __name__ == "__main__":
-    camp_id = create_campaign(
-        subject="Welcome to Your AI Fitness Experience!",
-        html_content="<h1>Welcome!</h1><p>Let’s get you moving with AI‑driven workouts.</p>"
-    )
-    send_campaign(camp_id)
-    print("Campaign sent with ID:", camp_id)
+        # 2. Set campaign content
+        client.campaigns.set_content(camp_id, {"html": html_content})
+
+        # 3. Send campaign
+        client.campaigns.send(camp_id)
+        return camp_id
+
+    except ApiClientError as e:
+        raise RuntimeError(f"Mailchimp error: {e.text}")