Update app.py

app.py

@@ -523,18 +523,19 @@ class MultilingualTranslator:
 def notify_wordpress_completion_and_charge(request_id: str, character_count: int, 
                                          translation_length: int, source_lang: str, 
                                          target_lang: str):
-    """ارسال مستقیم نتیجه ترجمه به وردپرس"""
+    """Send direct translation result to WordPress with improved error handling"""
     try:
-        # ابتدا نتیجه ترجمه را پیدا کن
+        # Get translation result
         if request_id not in translator.completed_translations:
             print(f"❌ Translation result not found for {request_id}")
             return False
         
         translation_data = translator.completed_translations[request_id]
         
-        # استفاده از URL از تنظیمات محیطی
+        # Use environment variable for WordPress URL
         wordpress_base_url = os.getenv("WORDPRESS_NOTIFICATION_URL", "https://echovizio.us.to")
         
+        # Send complete translation data in the notification
         payload = {
             'request_id': request_id,
             'character_count': character_count,
@@ -543,44 +544,53 @@ def notify_wordpress_completion_and_charge(request_id: str, character_count: int
             'target_lang': target_lang,
             'completed_at': datetime.now().isoformat(),
             'status': 'completed',
-            # ارسال مستقیم نتیجه ترجمه
+            # Include the actual translation result
             'translated_text': translation_data['result']['translated_text'],
             'processing_time': translation_data['result']['processing_time'],
-            'from_cache': translation_data['result'].get('from_cache', False)
+            'from_cache': translation_data['result'].get('from_cache', False),
+            'chunks_count': translation_data['result'].get('chunks_count', 1)
         }
         
         headers = {
             'Content-Type': 'application/json',
-            'User-Agent': 'MLT-Server/1.0'
+            'User-Agent': 'MLT-Server/1.0',
+            'Accept': 'application/json'
         }
         
         notification_url = wordpress_base_url.rstrip('/') + '/wp-json/mlt/v1/notification'
         
-        print(f"🔔 Sending translation result to: {notification_url}")
+        print(f"📡 Sending complete translation to: {notification_url}")
+        print(f"🎯 Payload size: {len(payload['translated_text'])} characters")
         
-        try:
-            response = requests.post(
-                notification_url, 
-                json=payload, 
-                headers=headers,
-                timeout=30,
-                verify=False
-            )
-            
-            if response.status_code == 200:
+        response = requests.post(
+            notification_url, 
+            json=payload, 
+            headers=headers,
+            timeout=45,  # Increased timeout
+            verify=False
+        )
+        
+        if response.status_code == 200:
+            response_data = response.json()
+            if response_data.get('success'):
                 print(f"✅ WordPress notification successful for {request_id}")
                 return True
             else:
-                print(f"❌ WordPress notification failed: {response.status_code}")
-                print(f"❌ Response: {response.text}")
+                print(f"❌ WordPress rejected notification: {response.text}")
                 return False
-                
-        except Exception as e:
-            print(f"❌ Request error: {str(e)}")
+        else:
+            print(f"❌ WordPress notification failed with status {response.status_code}")
+            print(f"📄 Response: {response.text}")
             return False
             
+    except requests.exceptions.Timeout:
+        print(f"⏰ WordPress notification timeout for {request_id}")
+        return False
+    except requests.exceptions.ConnectionError:
+        print(f"🔌 WordPress connection error for {request_id}")
+        return False
     except Exception as e:
-        print(f"❌ Notification setup error: {str(e)}")
+        print(f"💥 Notification error for {request_id}: {str(e)}")
         return False
 
 def process_heavy_translation_background(text: str, source_lang: str, target_lang: str,