Update app.py

app.py

@@ -6,15 +6,27 @@ import time
 from flask import Flask, request, jsonify, render_template, Response
 from werkzeug.utils import secure_filename
 from itertools import cycle
+from dotenv import load_dotenv
+
+# بارگذاری متغیرهای محیطی از فایل .env (برای تست لوکال)
+load_dotenv()
 
 # --- 1. تنظیمات اصلی ---
-WORKER_URLS = [
-    "https://ezmary-animat-worker1.hf.space",
-    "https://ezmary-animat-worker2.hf.space",
-    "https://ezmary-animat-worker3.hf.space",
-    "https://ezmary-animat-worker4.hf.space",
-    "https://ezmary-animat-worker5.hf.space",
-]
+
+# دریافت لیست کارگرها از متغیر محیطی
+# آدرس‌ها باید با کاما (,) جدا شده باشند
+workers_env = os.getenv("WORKER_NODES", "")
+
+if not workers_env:
+    # اگر متغیر تنظیم نشده باشد، برنامه با خطا متوقف می‌شود تا متوجه شوید
+    raise ValueError("خطا: متغیر محیطی WORKER_NODES تنظیم نشده است. لطفاً آدرس کارگرها را در تنظیمات اضافه کنید.")
+
+# تبدیل رشته به لیست و حذف فاصله‌های اضافی
+WORKER_URLS = [url.strip() for url in workers_env.split(',') if url.strip()]
+
+if not WORKER_URLS:
+    raise ValueError("خطا: لیست کارگرها خالی است.")
+
 worker_pool = cycle(WORKER_URLS)
 
 UPLOADER_API_URL = "https://hamed744-uploadfile.hf.space/upload"