Update app.py

app.py

@@ -4,24 +4,40 @@ from telebot.types import InlineKeyboardMarkup, InlineKeyboardButton, WebAppInfo
 import os
 from flask import Flask, jsonify, make_response, request
 from supabase import create_client
-import traceback
+import threading
 
-# ডিটেইলস
+# টোকেন ও ডিটেইলস
 BOT_TOKEN = "8628213901:AAFvfHBpZ6tok40ZQuhIDLAVIMrHeiheMNY"
 SUPABASE_URL = "https://yctirvnryrzygoxbpvoy.supabase.co"
 SUPABASE_KEY = "sb_publishable_aBcD-atruskWwoCiLr0lWw_inT8GLoN"
 WEB_APP_URL = "https://rony90790.github.io/Forward-bot/index.html" 
 HF_SPACE_URL = "https://pmrony-forwardbot.hf.space" 
 
-# --- কানেকশন টাইমআউট বাড়ানো হলো (সমাধান) ---
-apihelper.CONNECT_TIMEOUT = 60
-apihelper.READ_TIMEOUT = 60
+# কানেকশন টাইমআউট সর্বোচ্চ করা হলো
+apihelper.CONNECT_TIMEOUT = 100
+apihelper.READ_TIMEOUT = 100
 
 bot = telebot.TeleBot(BOT_TOKEN, threaded=False)
 supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
 app = Flask(__name__)
 
-admin_states = {}
+# --- ডাটাবেজ কাজ আলাদা থ্রেডে করার ফাংশন ---
+def save_referral_bg(user_id, referrer_id):
+    try:
+        user_check = supabase.table('referrals').select('*').eq('user_id', user_id).execute()
+        if not user_check.data:
+            supabase.table('referrals').insert({
+                'user_id': user_id, 'referral_count': 0, 
+                'referrer_id': referrer_id if referrer_id != user_id else None
+            }).execute()
+            if referrer_id and referrer_id != user_id:
+                ref_data = supabase.table('referrals').select('referral_count').eq('user_id', referrer_id).execute()
+                if ref_data.data:
+                    new_count = ref_data.data[0]['referral_count'] + 1
+                    supabase.table('referrals').update({'referral_count': new_count}).eq('user_id', referrer_id).execute()
+                    bot.send_message(referrer_id, "🎉 কেউ আপনার রেফারে জয়েন করেছে!")
+    except:
+        pass
 
 @app.route(f'/{BOT_TOKEN}', methods=['POST'])
 def webhook():
@@ -35,81 +51,51 @@ def webhook():
 @app.route('/set_webhook')
 def set_webhook():
     bot.remove_webhook()
-    success = bot.set_webhook(url=f"{HF_SPACE_URL}/{BOT_TOKEN}")
+    # ওয়েব হুক সেট করার সময় ১০ সেকেন্ড ওয়েট করবে
+    success = bot.set_webhook(url=f"{HF_SPACE_URL}/{BOT_TOKEN}", timeout=60)
     if success:
-        return "<h1>✅ Webhook Set Successfully!</h1>"
+        return "<h1>✅ Webhook Set Success!</h1>"
     else:
         return "<h1>❌ Webhook Set Failed!</h1>"
 
 @app.route('/')
 def index():
-    return "Bot is Running smoothly!"
+    return "Bot is Running Perfectly!"
 
-# ================= COMMAND HANDLERS =================
+# ================= COMMANDS =================
 
 @bot.message_handler(commands=['start'])
 def start(message):
-    try:
-        user_id = message.from_user.id
-        args = message.text.split()
-        referrer_id = int(args[1]) if len(args) > 1 and args[1].isdigit() else None
-
-        # ডাটাবেজে ডাটা সেভ করার চেষ্টা (এটা ফেইল করলেও মেসেজ যাবে)
-        try:
-            user_check = supabase.table('referrals').select('*').eq('user_id', user_id).execute()
-            if not user_check.data:
-                supabase.table('referrals').insert({
-                    'user_id': user_id, 'referral_count': 0, 
-                    'referrer_id': referrer_id if referrer_id != user_id else None
-                }).execute()
-
-                if referrer_id and referrer_id != user_id:
-                    ref_data = supabase.table('referrals').select('referral_count').eq('user_id', referrer_id).execute()
-                    if ref_data.data:
-                        new_count = ref_data.data[0]['referral_count'] + 1
-                        supabase.table('referrals').update({'referral_count': new_count}).eq('user_id', referrer_id).execute()
-                        bot.send_message(referrer_id, "🎉 কেউ আপনার রেফারে জয়েন করেছে!")
-        except Exception as db_e:
-            print(f"Database issue: {db_e}")
-
-        # কী-বোর্ড এবং মেসেজ পাঠানো
-        markup = InlineKeyboardMarkup()
-        markup.add(InlineKeyboardButton("Play video 🔞", web_app=WebAppInfo(url=WEB_APP_URL)))
-        
-        # সরাসরি send_message ব্যবহার করা হয়েছে রিলাইবিলিটির জন্য
-        bot.send_message(message.chat.id, "ভাইরাল ভিডিও দেখতে নিচের বাটনে ক্লিক করো 👇", reply_markup=markup)
-        
-    except Exception as e:
-        print(f"Error in /start: {e}")
+    user_id = message.from_user.id
+    args = message.text.split()
+    referrer_id = int(args[1]) if len(args) > 1 and args[1].isdigit() else None
+    
+    # ডাটাবেজের কাজ হবে ব্যাকগ্রাউন্ডে, যাতে বট স্লো না হয়
+    threading.Thread(target=save_referral_bg, args=(user_id, referrer_id)).start()
+
+    # ইউজারকে সাথে সাথে রিপ্লাই দিবে
+    markup = InlineKeyboardMarkup()
+    markup.add(InlineKeyboardButton("Play video 🔞", web_app=WebAppInfo(url=WEB_APP_URL)))
+    bot.send_message(message.chat.id, "ভাইরাল ভিডিও দেখতে নিচের বাটনে ক্লিক করো 👇", reply_markup=markup)
 
 @bot.message_handler(commands=['png'])
 def add_png(message):
-    try:
-        parts = message.text.split(maxsplit=1)
-        if len(parts) > 1:
-            img_url = parts[1].strip()
-            admin_states[message.chat.id] = {"step": 1, "thumbnail_url": img_url}
-            bot.send_message(message.chat.id, "✅ ছবি সেট হয়েছে। এখন এই ছবির জন্য **Google Drive ভিডিও লিংক** দিন।")
-        else:
-            bot.send_message(message.chat.id, "সঠিক নিয়ম: `/png https://link-to-image.jpg`")
-    except Exception as e:
-        print(f"Error in /png: {e}")
+    parts = message.text.split(maxsplit=1)
+    if len(parts) > 1:
+        img_url = parts[1].strip()
+        bot.send_message(message.chat.id, f"✅ ছবি সেট হয়েছে। এখন এই ছবির জন্য ভিডিওর **Google Drive লিংক** পাঠান।")
+        # ভিডিও অ্যাড করার লজিক এখানে সিম্পল রাখা হয়েছে
+        bot.register_next_step_handler(message, save_video_final, img_url)
+    else:
+        bot.send_message(message.chat.id, "সঠিক নিয়ম: `/png https://link.jpg`")
 
-@bot.message_handler(func=lambda m: admin_states.get(m.chat.id, {}).get("step") == 1)
-def add_video_step(message):
+def save_video_final(message, img_url):
+    video_url = message.text.strip()
     try:
-        video_url = message.text.strip()
-        thumbnail_url = admin_states[message.chat.id]["thumbnail_url"]
-        
-        supabase.table('videos').insert({
-            "video_url": video_url,
-            "thumbnail_url": thumbnail_url
-        }).execute()
-        
-        bot.send_message(message.chat.id, "🎉 অভিনন্দন! ভিডিওটি সফলভাবে অ্যাপে অ্যাড হয়েছে।")
-        admin_states[message.chat.id] = {}
-    except Exception as e:
-        bot.send_message(message.chat.id, "❌ ডাটাবেজে সেভ করতে সমস্যা হয়েছে।")
+        supabase.table('videos').insert({"video_url": video_url, "thumbnail_url": img_url}).execute()
+        bot.send_message(message.chat.id, "🎉 সফলভাবে ভিডিও অ্যাড হয়েছে!")
+    except:
+        bot.send_message(message.chat.id, "❌ ডাটাবেজ এরর!")
 
 @app.route('/api/videos')
 def api_videos():