Update app.py

app.py

@@ -1,26 +1,30 @@
 import telebot
 from telebot.types import InlineKeyboardMarkup, InlineKeyboardButton, WebAppInfo
 import os
-from flask import Flask, jsonify, make_response, request as flask_request
+from flask import Flask, jsonify, make_response, request
 from supabase import create_client
 
-BOT_TOKEN = os.environ.get('BOT_TOKEN') 
+# টোকেন চেক
+BOT_TOKEN = os.environ.get('BOT_TOKEN')
 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" 
 
+if not BOT_TOKEN:
+    print("Error: BOT_TOKEN not found in environment variables!")
+
 bot = telebot.TeleBot(BOT_TOKEN)
 supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
 app = Flask(__name__)
 
-# অ্যাডমিন কোন ধাপে আছে তা মনে রাখার জন্য ডিকশনারি
+# অ্যাডমিন স্টেট
 admin_states = {}
 
 @app.route(f'/{BOT_TOKEN}', methods=['POST'])
 def webhook():
-    if 'application/json' in flask_request.headers.get('content-type', ''):
-        json_string = flask_request.get_data().decode('utf-8')
+    if request.headers.get('content-type') == 'application/json':
+        json_string = request.get_data().decode('utf-8')
         update = telebot.types.Update.de_json(json_string)
         bot.process_new_updates([update])
         return "OK", 200
@@ -28,114 +32,85 @@ def webhook():
 
 @app.route('/set_webhook')
 def set_webhook():
-    bot.remove_webhook()
-    success = bot.set_webhook(url=f"{HF_SPACE_URL}/{BOT_TOKEN}")
-    if success:
-        return f"<h1>✅ Webhook Successfully Set!</h1>"
-    else:
-        return "<h1>❌ Failed to set webhook!</h1>"
+    try:
+        bot.remove_webhook()
+        webhook_url = f"{HF_SPACE_URL}/{BOT_TOKEN}"
+        success = bot.set_webhook(url=webhook_url)
+        if success:
+            return f"<h1>✅ Webhook Set Successfully!</h1><p>URL: {webhook_url}</p>"
+        else:
+            return "<h1>❌ Webhook Set Failed!</h1>"
+    except Exception as e:
+        return f"<h1>❌ Error:</h1><p>{str(e)}</p>"
 
 @app.route('/')
 def index():
-    return "Bot is Running smoothly!"
+    return "Bot is Running!"
 
-# ==========================================
-#          ADMIN COMMAND TO ADD VIDEO
-# ==========================================
+# ================= ADMIN LOGIC =================
 
 @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.reply_to(message, "ঠিক আছে, এখন Google Drive এর ভিডিও লিংক দেন।")
-        else:
-            bot.reply_to(message, "⚠️ সঠিক নিয়ম: `/png <ছবির লিংক>` এভাবে দিন।", parse_mode="Markdown")
-    except Exception as e:
-        bot.reply_to(message, f"Error: {e}")
+    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.reply_to(message, "✅ ছবি সেট হয়েছে। এখন ভিডিওর Google Drive লিংক দিন।")
+    else:
+        bot.reply_to(message, "সঠিক নিয়ম: `/png https://link-to-image.jpg`")
 
-# যখন ইউজারের স্টেট 1 থাকবে (অর্থাৎ সে ছবির লিংক দিয়েছে, এখন ভিডিওর লিংক দিবে)
-@bot.message_handler(func=lambda message: message.chat.id in admin_states and admin_states[message.chat.id]["step"] == 1)
-def add_video_link(message):
+@bot.message_handler(func=lambda m: admin_states.get(m.chat.id, {}).get("step") == 1)
+def add_video(message):
     video_url = message.text.strip()
     thumbnail_url = admin_states[message.chat.id]["thumbnail_url"]
     
     try:
-        # Supabase ডাটাবেজে সেভ করা
         supabase.table('videos').insert({
             "video_url": video_url,
             "thumbnail_url": thumbnail_url
         }).execute()
-        bot.reply_to(message, "✅ সফলভাবে ভিডিও এবং ছবি ডাটাবেজে সেভ হয়েছে! অ্যাপে গিয়ে চেক করুন।")
+        bot.reply_to(message, "🎉 সফলভাবে ভিডিও অ্যাড হয়েছে!")
     except Exception as e:
-        bot.reply_to(message, f"❌ সেভ করতে সমস্যা হয়েছে! নিশ্চিত করুন আপনি Supabase এ 'videos' টেবিল বানিয়েছেন কিনা।")
+        bot.reply_to(message, f"❌ ভুল ���য়েছে: {e}")
     
-    # কাজ শেষ, তাই স্টেট মুছে দিলাম
-    del admin_states[message.chat.id]
+    admin_states[message.chat.id] = {}
 
-# ==========================================
-#          API FOR MINI APP
-# ==========================================
+# ================= API =================
 
 @app.route('/api/videos')
 def api_videos():
     try:
-        # ডাটাবেজ থেকে সর্বশেষ অ্যাড করা ভিডিওগুলো আনবে
         res = supabase.table('videos').select('*').order('id', desc=True).execute()
-        videos_data = res.data
-    except Exception as e:
-        print(f"Error fetching videos: {e}")
-        videos_data = []
-        
-    response = make_response(jsonify(videos_data))
-    response.headers['Access-Control-Allow-Origin'] = '*' 
-    return response
+        return jsonify(res.data)
+    except:
+        return jsonify([])
 
-# ==========================================
-#          BOT START / REFERRAL LOGIC
-# ==========================================
+# ================= START / REFERRAL =================
 
 @bot.message_handler(commands=['start'])
 def start(message):
     user_id = message.from_user.id
-    text = message.text.split() 
-
-    referrer_id = None
-    if len(text) > 1:
-        try:
-            referrer_id = int(text[1])
-        except ValueError:
-            referrer_id = None 
+    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,
+                '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:
-                referrer_data = supabase.table('referrals').select('referral_count').eq('user_id', referrer_id).execute()
-                if referrer_data.data:
-                    new_count = referrer_data.data[0]['referral_count'] + 1
-                    supabase.table('referrals').update({'referral_count': new_count}).eq('user_id', referrer_id).execute()
-                    
-                    try:
-                        bot.send_message(referrer_id, "🎉 একজন আপনার লিংকে জয়েন করেছে! আপনার রেফারেল কাউন্ট বাড়লো।")
-                    except Exception:
-                        pass
-    except Exception as e:
-        print("[DATABASE ERROR]", e)
+                ref_data = supabase.table('referrals').select('referral_count').eq('user_id', referrer_id).execute()
+                if ref_data.data:
+                    supabase.table('referrals').update({'referral_count': ref_data.data[0]['referral_count'] + 1}).eq('user_id', referrer_id).execute()
+                    bot.send_message(referrer_id, "🎁 কেউ আপনার লিঙ্কে জয়েন করেছে!")
+    except: pass
 
     markup = InlineKeyboardMarkup()
     markup.add(InlineKeyboardButton("Play video 🔞", web_app=WebAppInfo(url=WEB_APP_URL)))
-    bot.reply_to(message, "ভাইরাল ভিডিও দেখতে নিচের বাটনে ক্লিক করো 👇", reply_markup=markup)
+    bot.reply_to(message, "ভিডিও দেখতে নিচের বাটনে ক্লিক করুন 👇", reply_markup=markup)
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, threaded=True)
+    app.run(host="0.0.0.0", port=7860)