Update main.py

main.py

@@ -311,16 +311,99 @@ def to_english_digits(text):
     return str(text).translate(translation_table)
 
 # ==============================================================================
-# 🟢 پارت 6: تنظیمات وب‌سرور (Flask) و توابع کمکی فایل‌ها
+# 🟢 پارت 6: تنظیمات وب‌سرور (Flask)، دریافت وب‌هوک و توابع کمکی فایل‌ها
 # ==============================================================================
-# --- تنظیمات وب سرور ---
+import requests
+import asyncio
+from flask import Flask, request, jsonify
+
+# --- تنظیمات وب سرور و وب‌هوک ---
 app = Flask(__name__)
 
-@app.route('/')
+# آدرس وب‌هوک (اگر از Cloudflare Worker استفاده کردید، آدرس آن را اینجا بگذارید)
+WEBHOOK_URL = "https://opera8-sadarubikabot.hf.space/"
+
+def set_webhook_on_rubika():
+    bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
+    if not bot_token: 
+        print("❌ توکن روبیکا یافت نشد!")
+        return
+    
+    url = f"https://botapi.rubika.ir/v3/{bot_token}/updateBotEndpoints"
+    
+    payload_update = {"url": WEBHOOK_URL, "type": "ReceiveUpdate"}
+    payload_inline = {"url": WEBHOOK_URL, "type": "ReceiveInlineMessage"}
+    
+    try:
+        r1 = requests.post(url, json=payload_update, timeout=15)
+        print("🟢 Webhook Setup (ReceiveUpdate):", r1.json())
+        
+        r2 = requests.post(url, json=payload_inline, timeout=15)
+        print("🟢 Webhook Setup (InlineMessage):", r2.json())
+    except Exception as e:
+        print("❌ Webhook Setup Failed:", e)
+
+class UpdateWrapper:
+    def __init__(self, d):
+        self._d = d
+        for k, v in d.items():
+            if isinstance(v, dict):
+                setattr(self, k, UpdateWrapper(v))
+            elif isinstance(v, list):
+                setattr(self, k, [UpdateWrapper(i) if isinstance(i, dict) else i for i in v])
+            else:
+                setattr(self, k, v)
+        
+        # 🟢 جادوی تطبیق متغیرهای وب‌هوک با استانداردهای ربات شما (رفع باگ خوانده نشدن پیام‌ها)
+        if hasattr(self, 'new_message'):
+            self.message = self.new_message
+            
+        if hasattr(self, 'message') and isinstance(self.message, UpdateWrapper):
+            if hasattr(self.message, 'sender_id') and not hasattr(self.message, 'author_object_guid'):
+                self.message.author_object_guid = self.message.sender_id
+                
+        if hasattr(self, 'chat_id') and not hasattr(self, 'object_guid'):
+            self.object_guid = self.chat_id
+            self.author_guid = self.chat_id
+    
+    def to_dict(self):
+        return self._d
+        
+    def get(self, key, default=None):
+        return getattr(self, key, default)
+
+@app.route('/', methods=['GET', 'POST'])
 def home():
-    return "ربات یکپارچه آلفا (نسخه پرو + مدیریت اشتراک نامحدود + دعوت دوستان + سیستم تغییر صدا) روشن است! 🚀"
+    if request.method == 'GET':
+        return "ربات یکپارچه آلفا (نسخه پرو + وب‌هوک روبیکا) روشن است! 🚀"
+    
+    if request.method == 'POST':
+        try:
+            data = request.json
+            if data:
+                inner_data = data
+                if "update" in data:
+                    inner_data = data["update"]
+                elif "inline_message" in data:
+                    inner_data = data["inline_message"]
+                    
+                wrapped_update = UpdateWrapper(inner_data)
+                
+                # تسک به سرعت به حلقه اصلی پاس داده می‌شود تا سرور منتظر نماند
+                if 'loop' in globals() and 'main_handler' in globals():
+                    asyncio.run_coroutine_threadsafe(main_handler(bot, wrapped_update), loop)
+                    
+        except Exception as e:
+            print("❌ Error in webhook processing:", e)
+            
+        # همیشه و در کسری از ثانیه 200 OK برمی‌گردانیم تا روبیکا ارتباط را قطع نکند
+        return jsonify({"status": "ok"})
 
 def run_flask():
+    set_webhook_on_rubika()
+    import logging
+    log = logging.getLogger('werkzeug')
+    log.setLevel(logging.ERROR)
     app.run(host="0.0.0.0", port=7860, threaded=True)
 
 # --- توابع کمکی ---
@@ -337,12 +420,15 @@ def sync_read_file(file_name):
         return f.read()
 
 def sync_combine_audio(current_audio, new_bytes):
+    from pydub import AudioSegment
+    import io
     audio_segment = AudioSegment.from_file(io.BytesIO(new_bytes))
     return current_audio + audio_segment
 
 def sync_export_audio(audio_obj, file_name):
     audio_obj.export(file_name, format="mp3")
 
+
 # ==============================================================================
 # 🟢 پارت 7: کیبوردها (دکمه���های شیشه‌ای / منوها)
 # ==============================================================================