Update app.py

app.py

@@ -8,8 +8,8 @@ from datetime import datetime, timedelta
 from concurrent.futures import ThreadPoolExecutor
 from typing import Dict, List, Optional, Any
 import json
-
-from fastapi import FastAPI, HTTPException, BackgroundTasks
+import uuid
+from fastapi import FastAPI, HTTPException, BackgroundTasks, Request
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel, Field
@@ -992,31 +992,72 @@ async def check_auto_charge_status(request: AutoChargeStatusRequest):
     }
 
 @app.post("/api/translate/heavy")
-async def translate_heavy_text(request: TranslationRequest, background_tasks: BackgroundTasks):
-    """ترجمه متن‌های سنگین در پس‌زمینه"""
-    import uuid
-    
-    # تولید شناسه منحصر به فرد
-    request_id = str(uuid.uuid4())
+async def heavy_translate(request: Request):
+    data = await request.json()
     
-    # اضافه کردن به وظایف پس‌زمینه
-    background_tasks.add_task(
-        process_heavy_translation_background,
-        request.text,
-        request.source_lang,
-        request.target_lang,
-        request_id,
-        request.auto_charge
-    )
+    # اگر وردپرس request_id فرستاده بود، همونو استفاده کن
+    request_id = data.get("request_id")
+    if not request_id:
+        request_id = str(uuid.uuid4())   # فقط در صورت نبودن request_id
     
-    return {
-        "success": True,
-        "message": "ترجمه در پس‌زمینه آغاز شد",
-        "request_id": request_id,
-        "estimated_time": len(request.text) // 100,  # تخمین بر اساس طول متن
-        "auto_charge": request.auto_charge
+    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,
+        "elapsed_time": 0,
+        "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."}
+
+
+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)  # مثلا هر ۵ ثانیه
+            translations[request_id]["progress"] = i * 10
+            translations[request_id]["elapsed_time"] += 5
+        
+        # در پایان، متن ترجمه‌شده
+        translated_text = f"[{target_lang}] {text}"
+        translations[request_id] = {
+            "status": "completed",
+            "progress": 100,
+            "elapsed_time": translations[request_id]["elapsed_time"],
+            "message": "Translation completed successfully.",
+            "result": translated_text
+        }
+
+        # ارسال notify به وردپرس
+        if notification_url:
+            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]}")
+            except Exception as e:
+                print(f"❌ Notify failed for {request_id}: {e}")
+
+    except Exception as e:
+        translations[request_id] = {
+            "status": "failed",
+            "message": f"Error: {e}"
+        }
+
 @app.post("/api/translate/session")
 async def translate_with_session(request: TranslationRequest):
     """ترجمه با session برای پیگیری پیشرفت"""