Update app.py

app.py

@@ -30,6 +30,8 @@ os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
 cache_dir = "/tmp/huggingface"
 os.makedirs(cache_dir, mode=0o777, exist_ok=True)
 
+# Set WordPress notification URL from environment or default
+WORDPRESS_BASE_URL = os.getenv("WORDPRESS_NOTIFICATION_URL", "http://localhost")
 
 # نگاشت زبان‌ها
 LANGUAGE_MAP = {
@@ -523,16 +525,20 @@ def notify_wordpress_completion_and_charge(request_id: str, character_count: int
                                          target_lang: str):
     """Enhanced WordPress notification with multiple URL attempts"""
     try:
-        # Get WordPress URL from settings or environment
+        # Get WordPress URL from environment or settings
         wordpress_base_url = os.getenv("WORDPRESS_NOTIFICATION_URL")
+        
+        # If no environment variable, try to get from settings or use default
         if not wordpress_base_url:
-            wordpress_base_url = "http://localhost"  # Default fallback
-            
-        wordpress_urls = [
-            f"{wordpress_base_url}/wp-json/mlt/v1/notification",
-            "http://localhost/wp-json/mlt/v1/notification",
-            "http://127.0.0.1/wp-json/mlt/v1/notification"
-        ]
+            # Try common WordPress URLs - replace with your actual WordPress URL
+            wordpress_urls = [
+                "https://your-wordpress-site.com",  # Replace with your actual WordPress URL
+                "http://your-wordpress-site.com",   # Replace with your actual WordPress URL
+                "http://localhost:8080",            # If using different port
+                "http://localhost"                  # Last resort
+            ]
+        else:
+            wordpress_urls = [wordpress_base_url]
         
         payload = {
             'request_id': request_id,
@@ -544,12 +550,18 @@ def notify_wordpress_completion_and_charge(request_id: str, character_count: int
             'status': 'completed'
         }
         
-        for wordpress_url in wordpress_urls:
-            if not wordpress_url:
+        for base_url in wordpress_urls:
+            if not base_url:
                 continue
                 
+            # Ensure URL ends with notification endpoint
+            if not base_url.endswith('/wp-json/mlt/v1/notification'):
+                notification_url = base_url.rstrip('/') + '/wp-json/mlt/v1/notification'
+            else:
+                notification_url = base_url
+                
             try:
-                response = requests.post(wordpress_url, json=payload, timeout=10)
+                response = requests.post(notification_url, json=payload, timeout=10)
                 
                 if response.status_code == 200:
                     # Mark as auto charged
@@ -558,10 +570,10 @@ def notify_wordpress_completion_and_charge(request_id: str, character_count: int
                     print(f"WordPress notification successful for request {request_id}")
                     return True
                 else:
-                    print(f"WordPress notification failed: {response.status_code} - {wordpress_url}")
+                    print(f"WordPress notification failed: {response.status_code} - {notification_url}")
                     
             except requests.exceptions.RequestException as e:
-                print(f"WordPress notification error for {wordpress_url}: {str(e)}")
+                print(f"WordPress notification error for {notification_url}: {str(e)}")
                 continue
         
         print(f"All WordPress notification attempts failed for request {request_id}")