Upload folder using huggingface_hub

src/orchestration/flows.py

@@ -11,19 +11,7 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-WEBHOOK_URL = os.getenv("WEBHOOK_URL")
-
-def notify_discord(message: str):
-    """Sends a notification to Discord."""
-    if not WEBHOOK_URL:
-        print("Warning: WEBHOOK_URL not set. Skipping notification.")
-        return
-    
-    data = {"content": message}
-    try:
-        requests.post(WEBHOOK_URL, json=data)
-    except Exception as e:
-        print(f"Failed to send notification: {e}")
+from src.orchestration.notifications import notify_discord
 
 @task(retries=3, retry_delay_seconds=60)
 def fetch_stock_data(symbol: str):

src/orchestration/notifications.py

@@ -0,0 +1,23 @@
+import os
+import requests
+from dotenv import load_dotenv
+
+load_dotenv()
+
+WEBHOOK_URL = os.getenv("WEBHOOK_URL")
+
+def notify_discord(message: str):
+    """Sends a notification to Discord."""
+    if not WEBHOOK_URL:
+        print("Warning: WEBHOOK_URL not set. Skipping notification.")
+        return
+    
+    data = {"content": message}
+    try:
+        response = requests.post(WEBHOOK_URL, json=data)
+        response.raise_for_status()
+    except requests.exceptions.RequestException as e:
+        print(f"Failed to send notification: {e}")
+        if 'response' in locals() and response is not None:
+             print(f"Response status: {response.status_code}")
+             print(f"Response body: {response.text}")

src/orchestration/test_notification.py

@@ -0,0 +1,15 @@
+from src.orchestration.notifications import notify_discord
+import os
+from dotenv import load_dotenv
+
+# Force load .env if available
+load_dotenv()
+
+if __name__ == "__main__":
+    print("Testing Discord notification...")
+    webhook = os.getenv("WEBHOOK_URL")
+    if not webhook:
+        print("ERROR: WEBHOOK_URL environment variable is NOT set.")
+    else:
+        print(f"WEBHOOK_URL is set (starts with {webhook[:10]}...)")
+        notify_discord("Test notification from Stock Prediction System debugging script")