Update app.py

app.py

@@ -8,25 +8,13 @@ from werkzeug.utils import secure_filename
 from itertools import cycle
 from dotenv import load_dotenv
 
-# بارگذاری متغیرهای محیطی از فایل .env (برای تست لوکال)
 load_dotenv()
 
-# --- 1. تنظیمات اصلی ---
-
-# دریافت لیست کارگرها از متغیر محیطی
-# آدرس‌ها باید با کاما (,) جدا شده باشند
 workers_env = os.getenv("WORKER_NODES", "")
-
 if not workers_env:
-    # اگر متغیر تنظیم نشده باشد، برنامه با خطا متوقف می‌شود تا متوجه شوید
-    raise ValueError("خطا: متغیر محیطی WORKER_NODES تنظیم نشده است. لطفاً آدرس کارگرها را در تنظیمات اضافه کنید.")
+    raise ValueError("WORKER_NODES environment variable is not set")
 
-# تبدیل رشته به لیست و حذف فاصله‌های اضافی
 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"
@@ -40,11 +28,10 @@ lock = threading.Lock()
 def get_next_worker_url():
     return next(worker_pool)
 
-# --- 2. توابع پس‌زمینه ---
 def get_permanent_link(job_id, temp_render_url, worker_url):
     try:
         with lock: jobs[job_id]["status"] = "در حال دائمی‌سازی لینک..."
-        if not HF_TOKEN: raise Exception("توکن HF برای آپلود دائمی یافت نشد.")
+        if not HF_TOKEN: raise Exception("HF_TOKEN not found")
         
         video_full_url = f"{worker_url}{temp_render_url}"
         
@@ -55,7 +42,7 @@ def get_permanent_link(job_id, temp_render_url, worker_url):
         data = response.json()
         
         uploader_proxy_url = data.get("hf_url")
-        if not uploader_proxy_url: raise Exception("پاسخ از سرور آپلودر معتبر نبود.")
+        if not uploader_proxy_url: raise Exception("Invalid response from uploader")
         
         path_part = uploader_proxy_url.split('/proxy/')[1]
         final_proxy_url = f"/proxy/{path_part}"
@@ -65,22 +52,21 @@ def get_permanent_link(job_id, temp_render_url, worker_url):
             jobs[job_id]["result"] = final_proxy_url
             
     except Exception as e:
-        print(f"خطا در دائمی‌سازی لینک برای {job_id}: {e}")
+        print(f"Error making link permanent for {job_id}: {e}")
         with lock:
             jobs[job_id]["status"] = "error"
-            jobs[job_id]["result"] = f"خطا در دائمی‌سازی لینک. لینک موقت: {temp_render_url}"
+            jobs[job_id]["result"] = f"Error making link permanent. Temp link: {temp_render_url}"
 
-# <<< تابع اصلاح شده: حذف تایم‌اوت و افزایش مقاومت در برابر خطای ارتباطی >>>
 def poll_worker_service(job_id, render_job_id, worker_url):
     POLLING_INTERVAL = 15
-    MAX_POLLING_ERRORS = 20 # اگر ۲۰ بار پشت سر هم (حدود ۵ دقیقه) نتوانستیم با کارگر صحبت کنیم، خطا می‌دهیم
+    MAX_POLLING_ERRORS = 20
     error_count = 0
 
-    while True: # حلقه بی‌نهایت، هرگز به خاطر زمان متوقف نمی‌شود
+    while True:
         try:
             response = requests.post(f"{worker_url}/check_job_status", json={"job_id": render_job_id}, timeout=45)
             response.raise_for_status()
-            error_count = 0 # ریست کردن شمارنده خطا در صورت موفقیت
+            error_count = 0
 
             data = response.json()
             with lock:
@@ -90,27 +76,26 @@ def poll_worker_service(job_id, render_job_id, worker_url):
             
             if data.get("status") == "completed":
                 get_permanent_link(job_id, data.get("result"), worker_url)
-                return # خروج از حلقه
+                return
             elif data.get("status") == "error":
                 with lock: 
                     current_job = jobs.get(job_id)
                     if current_job:
-                        current_job["result"] = data.get("result", "خطای نامشخص از کارگر")
-                return # خروج از حلقه
+                        current_job["result"] = data.get("result", "Unknown worker error")
+                return
         except requests.exceptions.RequestException as e:
             error_count += 1
-            print(f"خطای ارتباطی در پیگیری کارگر {worker_url} (تلاش {error_count}/{MAX_POLLING_ERRORS}): {e}")
+            print(f"Connection error polling worker {worker_url} ({error_count}/{MAX_POLLING_ERRORS}): {e}")
             if error_count >= MAX_POLLING_ERRORS:
                 with lock:
                     current_job = jobs.get(job_id)
                     if current_job:
                         current_job["status"] = "error"
-                        current_job["result"] = f"ارتباط با سرور کارگر ({worker_url}) برای مدت طولانی قطع شد. ممکن است کارگر در حال ری‌استارت باشد. لطفاً بعداً وضعیت را مجدداً بررسی کنید."
-                return # خروج از حلقه
+                        current_job["result"] = f"Connection to worker ({worker_url}) lost."
+                return
         
         time.sleep(POLLING_INTERVAL)
 
-# --- 3. API Endpoints ---
 @app.route('/')
 def index():
     return render_template('index.html')
@@ -118,7 +103,7 @@ def index():
 @app.route('/api/submit_job', methods=['POST'])
 def submit_job():
     if 'image_file' not in request.files or 'video_file' not in request.files:
-        return jsonify({"error": "فایل عکس و ویدیو الزامی است."}), 400
+        return jsonify({"error": "Image and video files are required"}), 400
 
     image_file = request.files['image_file']
     video_file = request.files['video_file']
@@ -131,7 +116,7 @@ def submit_job():
     MAX_RETRIES = 3
     for attempt in range(MAX_RETRIES):
         selected_worker_url = get_next_worker_url()
-        print(f"تلاش {attempt + 1}/{MAX_RETRIES}: ارسال کار به کارگر: {selected_worker_url}")
+        print(f"Attempt {attempt + 1}/{MAX_RETRIES}: Sending to worker: {selected_worker_url}")
         try:
             response = requests.post(
                 f"{selected_worker_url}/submit_new_job",
@@ -143,33 +128,33 @@ def submit_job():
             response.raise_for_status()
             render_data = response.json()
             render_job_id = render_data.get("job_id")
-            if not render_job_id: raise Exception("پاسخ معتبر از کارگر دریافت نشد.")
+            if not render_job_id: raise Exception("Invalid response from worker")
             
             internal_job_id = str(uuid.uuid4())
-            with lock: jobs[internal_job_id] = {"status": "ارسال شده به کارگر...", "result": None}
+            with lock: jobs[internal_job_id] = {"status": "Sending to worker...", "result": None}
             
             thread = threading.Thread(target=poll_worker_service, args=(internal_job_id, render_job_id, selected_worker_url))
             thread.start()
             return jsonify({"job_id": internal_job_id})
         except Exception as e:
-            print(f"خطا در تلاش {attempt + 1} برای کارگر {selected_worker_url}: {e}")
+            print(f"Error on attempt {attempt + 1} for worker {selected_worker_url}: {e}")
             time.sleep(2)
 
-    final_error_message = ("<strong>سرورهای پردازش شلوغ هستند!</strong><br>"
-                         "لطفاً چند دقیقه دیگر مجدداً تلاش نمایید.")
+    final_error_message = ("<strong>Servers are busy!</strong><br>"
+                         "Please try again in a few minutes.")
     return jsonify({"error": final_error_message}), 500
 
 @app.route('/api/check_status', methods=['POST'])
 def check_status():
     data = request.get_json()
     job_id = data.get('job_id')
-    if not job_id: return jsonify({"error": "شناسه کار الزامی است."}), 400
+    if not job_id: return jsonify({"error": "Job ID is required"}), 400
     with lock: job = jobs.get(job_id, {"status": "not_found", "result": None})
     return jsonify({"status": job["status"], "result": job["result"]})
 
 @app.route('/proxy/<user>/<repo>/<path:file_path>')
 def file_proxy(user, repo, file_path):
-    if not HF_TOKEN: return "Token سرور برای پراکسی تنظیم نشده است.", 500
+    if not HF_TOKEN: return "HF_TOKEN not set", 500
     repo_id = f"{user}/{repo}"
     file_url = f"https://huggingface.co/datasets/{repo_id}/resolve/main/{file_path}"
     headers = {"Authorization": f"Bearer {HF_TOKEN}"}
@@ -179,11 +164,11 @@ def file_proxy(user, repo, file_path):
         return Response(hf_response.iter_content(chunk_size=8192),
                         content_type=hf_response.headers.get('Content-Type', 'application/octet-stream'))
     except requests.exceptions.HTTPError as e:
-        if e.response.status_code == 404: return "فایل یافت نشد.", 404
-        return f"خطای سرور Hugging Face: {e.response.status_code}", e.response.status_code
+        if e.response.status_code == 404: return "File not found", 404
+        return f"Hugging Face Server Error: {e.response.status_code}", e.response.status_code
     except Exception as e:
-        print(f"خطای پراکسی: {e}")
-        return "خطا در واکشی فایل.", 500
+        print(f"Proxy error: {e}")
+        return "Error fetching file", 500
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))