Update app.py

app.py

@@ -24,6 +24,7 @@ MAX_CHUNK_SIZE = 350
 MAX_WORKERS = 3
 CLEANUP_INTERVAL = 300  # 5 دقیقه
 
+translations = {}
 
 os.environ["HF_HOME"] = "/tmp/huggingface"
 os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
@@ -991,22 +992,26 @@ async def check_auto_charge_status(request: AutoChargeStatusRequest):
         "status": req_data.get('status', 'unknown')
     }
 
+app = FastAPI()
+
+# دیکشنری سراسری برای وضعیت ترجمه‌ها
+translations = {}
+
 @app.post("/api/translate/heavy")
 async def heavy_translate(request: Request):
     data = await request.json()
-    
-    # اگر وردپرس request_id فرستاده بود، همونو استفاده کن
+
+    # اگر وردپرس request_id فرستاده، همونو استفاده کن
     request_id = data.get("request_id")
     if not request_id:
-        request_id = str(uuid.uuid4())   # فقط در صورت نبودن request_id
-    
+        request_id = str(uuid.uuid4())
+
     text = data.get("text")
     source_lang = data.get("source_lang")
     target_lang = data.get("target_lang")
     auto_charge = data.get("auto_charge", False)
     notification_url = data.get("notification_url")
 
-    # ذخیره وضعیت اولیه (processing)
     translations[request_id] = {
         "status": "processing",
         "progress": 0,
@@ -1014,7 +1019,6 @@ async def heavy_translate(request: Request):
         "message": "Translation in progress..."
     }
 
-    # اجرای async برای شبیه‌سازی ترجمه
     asyncio.create_task(run_translation_job(request_id, text, source_lang, target_lang, notification_url))
 
     return {"success": True, "request_id": request_id, "message": "Background translation started."}
@@ -1022,13 +1026,11 @@ async def heavy_translate(request: Request):
 
 async def run_translation_job(request_id, text, source_lang, target_lang, notification_url):
     try:
-        # شبیه‌سازی پیشرفت
         for i in range(1, 10):
-            await asyncio.sleep(5)  # مثلا هر ۵ ثانیه
+            await asyncio.sleep(5)
             translations[request_id]["progress"] = i * 10
             translations[request_id]["elapsed_time"] += 5
-        
-        # در پایان، متن ترجمه‌شده
+
         translated_text = f"[{target_lang}] {text}"
         translations[request_id] = {
             "status": "completed",
@@ -1038,14 +1040,13 @@ async def run_translation_job(request_id, text, source_lang, target_lang, notifi
             "result": translated_text
         }
 
-        # ارسال notify به وردپرس
         if notification_url:
+            import requests
             payload = {
                 "request_id": request_id,
                 "status": "completed",
                 "result": translated_text
             }
-            import requests
             try:
                 r = requests.post(notification_url, json=payload, verify=False, timeout=10)
                 print(f"📡 Sent notify for {request_id}, status={r.status_code}, body={r.text[:200]}")