Update app.py

app.py

@@ -1,8 +1,10 @@
+import telebot
+from telebot import apihelper
+from telebot.types import InlineKeyboardMarkup, InlineKeyboardButton, WebAppInfo
 import os
-import requests
 from flask import Flask, jsonify, make_response, request
 from supabase import create_client
-import threading
+import traceback
 
 # ডিটেইলস
 BOT_TOKEN = "8628213901:AAFvfHBpZ6tok40ZQuhIDLAVIMrHeiheMNY"
@@ -11,81 +13,132 @@ 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" 
 
-# কানেকশন এরর কমানোর জন্য বিশেষ সেটিংস
-requests.packages.urllib3.util.ssl_.DEFAULT_CIPHERS += 'HIGH:!DH:!aNULL'
+# 🔴 আপনার দেওয়া অ্যাডমিন আইডি এখানে সেট করা হয়েছে
+ADMIN_IDS = [7307789267]
 
+# --- কানেকশন টাইমআউট বাড়ানো হলো (সমাধান) ---
+apihelper.CONNECT_TIMEOUT = 60
+apihelper.READ_TIMEOUT = 60
+
+bot = telebot.TeleBot(BOT_TOKEN, threaded=False)
 supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
 app = Flask(__name__)
 
-# --- মিরর লিঙ্ক ব্যবহার করা হচ্ছে (SSL সমস্যা সমাধানের জন্য) ---
-def send_telegram_message(chat_id, text, reply_markup=None):
-    # যদি t.me কাজ না করে তবে মিরর URL ট্রাই করে
-    url = f"https://api.telegram.org/bot{BOT_TOKEN}/sendMessage"
-    payload = {
-        "chat_id": chat_id,
-        "text": text,
-        "parse_mode": "HTML"
-    }
-    if reply_markup:
-        payload["reply_markup"] = reply_markup
-    
-    try:
-        # ৫ বার ট্রাই করবে যদি কানেকশন এরর হয়
-        r = requests.post(url, json=payload, timeout=20)
-        return r.json()
-    except Exception as e:
-        print(f"Error: {e}")
-        return None
+admin_states = {}
 
 @app.route(f'/{BOT_TOKEN}', methods=['POST'])
 def webhook():
-    data = request.get_json()
-    if not data or "message" not in data:
+    if request.is_json:
+        json_string = request.get_data(as_text=True)
+        update = telebot.types.Update.de_json(json_string)
+        bot.process_new_updates([update])
         return "OK", 200
-    
-    msg = data["message"]
-    chat_id = msg["chat"]["id"]
-    text = msg.get("text", "")
-
-    if text.startswith("/start"):
-        # রেফারেল লজিক
-        args = text.split()
-        ref_id = int(args[1]) if len(args) > 1 and args[1].isdigit() else None
-        
-        # কিবোর্ড
-        markup = {
-            "inline_keyboard": [[
-                {"text": "Play video 🔞", "web_app": {"url": WEB_APP_URL}}
-            ]]
-        }
-        send_telegram_message(chat_id, "ভাইরাল ভিডিও দেখতে নিচের বাটনে ক্লিক করো 👇", markup)
-
-    return "OK", 200
+    return "Forbidden", 403
 
 @app.route('/set_webhook')
 def set_webhook():
-    webhook_url = f"{HF_SPACE_URL}/{BOT_TOKEN}"
-    # সরাসরি ব্রাউজার দিয়ে করার জন্য লিঙ্কটি রিটার্ন করছে
-    api_url = f"https://api.telegram.org/bot{BOT_TOKEN}/setWebhook?url={webhook_url}"
+    bot.remove_webhook()
+    success = bot.set_webhook(url=f"{HF_SPACE_URL}/{BOT_TOKEN}")
+    if success:
+        return "<h1>✅ Webhook Set Successfully!</h1>"
+    else:
+        return "<h1>❌ Webhook Set Failed!</h1>"
+
+@app.route('/')
+def index():
+    return "Bot is Running smoothly!"
+
+# ================= COMMAND HANDLERS =================
+
+@bot.message_handler(commands=['start'])
+def start(message):
     try:
-        r = requests.get(api_url, timeout=20)
-        return jsonify(r.json())
-    except:
-        return f"বট থেকে হচ্ছে না, আপনি নিজে এই লিঙ্কে ঢুকুন: {api_url}"
+        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)))
+        
+        bot.send_message(message.chat.id, "ভাইরাল ভিডিও দেখতে নিচের বাটনে ক্লিক করো 👇", reply_markup=markup)
+        
+    except Exception as e:
+        print(f"Error in /start: {e}")
+
+@bot.message_handler(commands=['png'])
+def add_png(message):
+    try:
+        # সিকিউরিটি চেক: শুধু অ্যাডমিনরা এই কমান্ড ব্যবহার করতে পারবে
+        if message.from_user.id not in ADMIN_IDS:
+            bot.send_message(message.chat.id, "❌ আপনার এই কমান্ড ব্যবহার করার অনুমতি নেই।")
+            return
+
+        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}")
+
+@bot.message_handler(func=lambda m: admin_states.get(m.chat.id, {}).get("step") == 1)
+def add_video_step(message):
+    try:
+        # যদি অ্যাডমিন ভিডিও লিংকের বদলে ভুল করে অন্য কোনো কমান্ড দেয়, তবে প্রসেস বাতিল হবে
+        if message.text.startswith('/'):
+            bot.send_message(message.chat.id, "❌ ভিডিও অ্যাড প্রসেস বাতিল করা হয়েছে।")
+            admin_states.pop(message.chat.id, None)
+            return
+
+        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, "🎉 অভিনন্দন! ভিডিওটি সফলভাবে অ্যাপে অ্যাড হয়েছে।")
+    except Exception as e:
+        bot.send_message(message.chat.id, "❌ ডাটাবেজে সেভ করতে সমস্যা হয়েছে।")
+    finally:
+        # সফল হোক বা ব্যর্থ, স্টেট ক্লিয়ার করা হলো যাতে মেমরি লিক না হয়
+        admin_states.pop(message.chat.id, None)
 
 @app.route('/api/videos')
 def api_videos():
     try:
         res = supabase.table('videos').select('*').order('id', desc=True).execute()
-        resp = make_response(jsonify(res.data))
-        resp.headers['Access-Control-Allow-Origin'] = '*'
-        return resp
-    except:
-        return jsonify([])
-
-@app.route('/')
-def index():
-    return "Running..."
+        response = make_response(jsonify(res.data))
+    except Exception as e:
+        print(f"API Error: {e}")
+        # error হলেও empty array পাঠাতে হবে
+        response = make_response(jsonify([]))
+    
+    # যেকোনো অবস্থাতেই CORS হেডার দিতে হবে, না হলে ওয়েব অ্যাপ ক্র্যাশ করবে
+    response.headers['Access-Control-Allow-Origin'] = '*'
+    return response
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)