Update app.py

app.py

@@ -1,36 +1,67 @@
 import telebot
+from telebot.types import InlineKeyboardMarkup, InlineKeyboardButton, WebAppInfo
 import os
 import threading
 from flask import Flask
 from supabase import create_client
 
-# --- কনফিগারেশন (Secrets থেকে আসবে) ---
-BOT_TOKEN = os.environ.get('BOT_TOKEN')
+# --- কনফিগারেশন ---
+BOT_TOKEN = os.environ.get('BOT_TOKEN') # আপনার বট টোকেন দিন
 SUPABASE_URL = os.environ.get('SUPABASE_URL')
 SUPABASE_KEY = os.environ.get('SUPABASE_KEY')
+WEB_APP_URL = "https://rony90790.github.io/Forward-bot" # আপনার Web App এর লিংক
 
 bot = telebot.TeleBot(BOT_TOKEN)
 supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
-
-# ফ্লাস্ক অ্যাপ (HF-কে চালু রাখার জন্য)
 app = Flask(__name__)
 
 @app.route('/')
 def index():
     return "Bot is Running!"
 
-# --- আপনার রেফারেল লজিক এবং বটের কমান্ড এখানে দিন ---
+# --- স্মার্ট রেফারেল লজিক ---
 @bot.message_handler(commands=['start'])
 def start(message):
-    bot.reply_to(message, "বটটি সক্রিয় আছে!")
-    # আপনার আগের রেফারেল লজিক এখানে যুক্ত করুন...
+    user_id = message.from_user.id
+    text = message.text.split() # চেক করবে /start এর পর কোন ID আছে কিনা
+
+    referrer_id = None
+    if len(text) > 1:
+        referrer_id = text[1] # যে ইনভাইট করেছে তার ID
+
+    # চেক করবো ইউজার ডেটাবেসে আগে থেকে আছে কিনা
+    user_check = supabase.table('referrals').select('*').eq('user_id', user_id).execute()
+
+    if len(user_check.data) == 0:
+        # ইউজার নতুন, তাই ডেটাবেসে যুক্ত করবো
+        supabase.table('referrals').insert({
+            'user_id': user_id,
+            'referral_count': 0,
+            'referred_by': referrer_id if referrer_id != str(user_id) else None
+        }).execute()
+
+        # যদি সে কারো লিংক থেকে আসে, তাহলে ইনভাইটকারীর পয়েন্ট বাড়াবো
+        if referrer_id and referrer_id != str(user_id):
+            referrer_data = supabase.table('referrals').select('referral_count').eq('user_id', referrer_id).execute()
+            if len(referrer_data.data) > 0:
+                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:
+                    pass
+
+    # Web App বাটন তৈরি
+    markup = InlineKeyboardMarkup()
+    markup.add(InlineKeyboardButton("Play video 🔞", web_app=WebAppInfo(url=WEB_APP_URL)))
+
+    bot.reply_to(message, "ভাইরাল ভিডিও দেখতে নিচের বাটনে ক্লিক করো 👇", reply_markup=markup)
 
-# বট চালানোর ফাংশন
 def run_bot():
     bot.infinity_polling()
 
 if __name__ == "__main__":
-    # বটকে আলাদা থ্রেডে চালানো
     threading.Thread(target=run_bot).start()
-    # ফ্লাস্ক সার্ভার চালানো (Port 7860 HF এর জন্য ফিক্সড)
     app.run(host="0.0.0.0", port=7860)