Update app.py

app.py

@@ -16,13 +16,19 @@ DATASET_FILENAME = "video_usage_data.json"
 USAGE_LIMIT = 5
 HF_TOKEN = os.environ.get("HF_TOKEN")
 TEMP_DIR = "/app/tmp" 
+
+# ✅✅✅ آدرس ورودی اصلی (مسیریاب) که از متغیرهای راز خوانده می‌شود
 ROUTER_WORKER_URL = os.environ.get("ROUTER_WORKER_URL")
 
 # --- تنظیمات لاگینگ ---
+# فقط پیام لاگ نمایش داده می‌شود
 for handler in logging.root.handlers[:]:
     logging.root.removeHandler(handler)
 logging.basicConfig(level=logging.INFO, format='%(message)s')
+
+# ساکت کردن لاگ‌های کتابخانه‌های پرحرف
 logging.getLogger('huggingface_hub').setLevel(logging.ERROR)
+logging.getLogger('gunicorn.error').setLevel(logging.WARNING)
 logging.getLogger('werkzeug').setLevel(logging.ERROR)
 logging.getLogger("urllib3").setLevel(logging.WARNING)
 
@@ -39,11 +45,13 @@ api = None
 if HF_TOKEN:
     api = HfApi(token=HF_TOKEN)
 
+# بررسی وجود آدرس مسیریاب در هنگام شروع به کار
 if not ROUTER_WORKER_URL:
     logging.error("❌ CRITICAL ERROR: The ROUTER_WORKER_URL environment variable is not set!")
 else:
     logging.info(f"✅ Main app is configured to use the router at: {ROUTER_WORKER_URL}")
 
+# --- توابع مدیریت داده (بدون تغییر) ---
 def load_initial_data():
     global usage_data_cache
     with cache_lock:
@@ -51,13 +59,18 @@ def load_initial_data():
         try:
             with tempfile.TemporaryDirectory(dir=TEMP_DIR) as tmp_download_dir:
                 local_path = hf_hub_download(
-                    repo_id=DATASET_REPO, filename=DATASET_FILENAME, repo_type="dataset",
-                    token=HF_TOKEN, force_download=True, cache_dir=tmp_download_dir
+                    repo_id=DATASET_REPO, 
+                    filename=DATASET_FILENAME, 
+                    repo_type="dataset",
+                    token=HF_TOKEN,
+                    force_download=True,
+                    cache_dir=tmp_download_dir
                 )
                 with open(local_path, 'r', encoding='utf-8') as f:
                     content = f.read()
                     usage_data_cache = json.loads(content) if content else []
         except Exception:
+            # در صورت بروز خطا، با لیست خالی شروع می‌کنیم
             usage_data_cache = []
 
 def persist_data_to_hub():
@@ -72,10 +85,14 @@ def persist_data_to_hub():
                 temp_filepath = temp_f.name
                 json.dump(data_to_write, temp_f, ensure_ascii=False, indent=2)
             api.upload_file(
-                path_or_fileobj=temp_filepath, path_in_repo=DATASET_FILENAME,
-                repo_id=DATASET_REPO, repo_type="dataset", commit_message="Update animation usage data"
+                path_or_fileobj=temp_filepath,
+                path_in_repo=DATASET_FILENAME,
+                repo_id=DATASET_REPO,
+                repo_type="dataset",
+                commit_message="Update animation usage data"
             )
         except Exception:
+            # اگر آپلود ناموفق بود، دوباره پرچم را تنظیم می‌کنیم تا در تلاش بعدی ارسال شود
             data_changed.set()
         finally:
             if temp_filepath and os.path.exists(temp_filepath):
@@ -101,7 +118,7 @@ def enhance_animation_prompt():
     if not image_file:
         return jsonify({"error": "فایل تصویر الزامی است."}), 400
 
-    # 🧠🧠🧠 شاه پرامپت کامل و نهایی برای ساخت ویدیو 🧠🧠🧠
+    # 🧠🧠🧠 شاه پرامپت کامل در اینجا ساخته می‌شود 🧠🧠🧠
     gemini_master_prompt = f"""
 You are an expert AI Animation Planner. Your absolute highest priority is to faithfully and creatively execute the user's specific request. You are not just an artist; you are a technical problem solver.
 
@@ -166,41 +183,42 @@ Based on your decision, generate the following two keys in English.
 }}
 """
     
+    # خواندن فایل تصویر در حافظه
     image_file.seek(0)
     image_bytes = image_file.read()
     
     endpoint = f"{ROUTER_WORKER_URL}/v1/request"
-    logging.info(f"Relaying video prompt request to router at {endpoint}")
+    logging.info(f"Relaying request to router at {endpoint}")
 
     try:
         files = {'image': (image_file.filename, image_bytes, image_file.mimetype)}
-        data = {'prompt': gemini_master_prompt}
+        data = {'prompt': gemini_master_prompt} # شاه پرامپت کامل به مسیریاب ارسال می‌شود
         
+        # ارسال درخواست با تایم‌اوت طولانی‌تر برای اطمینان
         response = requests.post(endpoint, files=files, data=data, timeout=100)
-        response.raise_for_status()
-        
-        router_response = response.json()
         
-        if router_response.get("type") == "json" and "content" in router_response:
-            logging.info("✅ Successfully received and unwrapped JSON content from router.")
-            return jsonify(router_response["content"])
-        else:
-            logging.error(f"❌ Unexpected response format from router: {router_response}")
-            return jsonify({"error": "پاسخ دریافت شده از سرور پردازش، معتبر نبود."}), 502
+        # اگر پاسخ خطا بود (مثلاً 5xx)، استثنا ایجاد می‌کند
+        response.raise_for_status()
+
+        logging.info("✅ Successfully received response from router.")
+        return jsonify(response.json())
 
     except requests.exceptions.Timeout:
-        logging.error(f"❌ Timeout error from router.")
-        return jsonify({"error": "سرور پردازش با تاخیر مواجه است. لطفاً دوباره تلاش کنید."}), 504
+        logging.error("❌ Timeout error: The request to the router timed out.")
+        return jsonify({"error": "سرور پردازش با تاخیر مواجه است. لطفاً چند لحظه دیگر دوباره تلاش کنید."}), 504
     except requests.exceptions.RequestException as e:
-        error_detail = "سرویس هوش مصنوعی با مشکل مواجه شده است."
+        error_detail = "سرویس هوش مصنوعی با مشکل مواجه شده است. لطفاً بعداً تلاش کنید."
         if e.response is not None:
             try:
+                # تلاش برای استخراج پیام خطای اصلی از مسیریاب
                 router_error = e.response.json().get("detail", e.response.text)
                 error_detail = f"خطا از سرور پردازش: {router_error}"
-            except: pass
-        logging.error(f"❌ RequestException from router: {error_detail}")
+            except json.JSONDecodeError:
+                error_detail = f"سرور پردازش با خطای {e.response.status_code} پاسخ داد."
+        logging.error(f"❌ RequestException when contacting router: {error_detail}")
         return jsonify({"error": error_detail}), 503
 
+# --- روت‌های مدیریت اعتبار (بدون تغییر) ---
 def get_user_identifier(data):
     fingerprint = data.get('fingerprint')
     if fingerprint: return str(fingerprint)
@@ -260,10 +278,12 @@ def use_credit():
         
     return jsonify({"status": "success", "credits_remaining": credits_remaining})
 
+# --- روت بررسی سلامت (Health Check) ---
 @app.route('/health', methods=['GET'])
 def health_check():
     return "OK", 200
 
+# --- راه‌اندازی برنامه (بدون تغییر) ---
 if __name__ != '__main__':
     try:
         load_initial_data()