Update bot.py

bot.py

@@ -4,6 +4,7 @@ import math
 import json
 import asyncio
 import subprocess
+import uuid
 import yt_dlp
 from datetime import datetime, timedelta
 from threading import Thread
@@ -43,6 +44,7 @@ admin_state = None
 # --- मैसेज डिलीट करने के लिए नया डेटाबेस (टेम्परेरी) ---
 sent_message_map = {} 
 recent_admin_messages =[] 
+GLOBAL_SCHEDULE_TIMES = [] # एडमिन के सेट किये गए प्री-डिफाइंड शेड्यूल्स 
 
 # --- Data Persistence (बॉट रिस्टार्ट होने पर डेटा सेव रखने के लिए) ---
 DATA_FILE = "bot_config.json"
@@ -57,7 +59,8 @@ def save_data():
                 "PASSWORD_LIMIT": PASSWORD_LIMIT,
                 "GLOBAL_LIMIT": GLOBAL_LIMIT,
                 "PASSWORD_USED_BY": {str(k): v for k, v in PASSWORD_USED_BY.items()},
-                "authenticated_users": auth_users
+                "authenticated_users": auth_users,
+                "GLOBAL_SCHEDULE_TIMES": GLOBAL_SCHEDULE_TIMES
             }, f)
     except Exception as e: pass
 
@@ -71,6 +74,7 @@ def load_data():
                 PASSWORD_LIMIT = data.get("PASSWORD_LIMIT", 999999)
                 GLOBAL_LIMIT = data.get("GLOBAL_LIMIT", 1)
                 PASSWORD_USED_BY = {int(k): v for k, v in data.get("PASSWORD_USED_BY", {}).items()}
+                GLOBAL_SCHEDULE_TIMES = data.get("GLOBAL_SCHEDULE_TIMES", [])
                 
                 saved_users = data.get("authenticated_users", {})
                 for uid, name in saved_users.items():
@@ -140,6 +144,7 @@ def init_user(user_id, message=None):
             'video_url': None,         # डायरेक्ट लिंक
             'schedule_time': None,     # शेड्यूलिंग
             'processes':[], 
+            'scheduled_tasks': {},     # <--- नया: शेड्यूल्ड टास्क को ट्रैक करने के लिए
             'stream_limit': GLOBAL_LIMIT, 
             'file_path': None,
             'manual_stop': False,
@@ -178,7 +183,13 @@ def schedule_choice_keyboard():
     return InlineKeyboardMarkup([[InlineKeyboardButton("▶️ Start Now", callback_data="sch_now")],[InlineKeyboardButton("⏳ Schedule Time", callback_data="sch_later")]])
 
 def admin_keyboard():
-    return InlineKeyboardMarkup([[InlineKeyboardButton("👥 Users List", callback_data="admin_users"), InlineKeyboardButton("🔴 Live Status", callback_data="admin_live")],[InlineKeyboardButton("🔑 Change Password", callback_data="admin_pass"), InlineKeyboardButton("⚙️ Set Limit", callback_data="admin_limit")],[InlineKeyboardButton("📢 Broadcast / Send Msg", callback_data="admin_msg")],[InlineKeyboardButton("🗑 Delete Last 5 Broadcasts", callback_data="admin_del")]])
+    return InlineKeyboardMarkup([
+        [InlineKeyboardButton("👥 Users List", callback_data="admin_users"), InlineKeyboardButton("🔴 Live Status", callback_data="admin_live")],
+        [InlineKeyboardButton("🔑 Change Password", callback_data="admin_pass"), InlineKeyboardButton("⚙️ Set Limit", callback_data="admin_limit")],
+        [InlineKeyboardButton("📅 Manage Schedule Times", callback_data="admin_sch_times")],
+        [InlineKeyboardButton("📢 Broadcast / Send Msg", callback_data="admin_msg")],
+        [InlineKeyboardButton("🗑 Delete Last 5 Broadcasts", callback_data="admin_del")]
+    ])
 
 # --- Live Progress Bar ---
 async def progress_bar(current, total, msg, start_time, last_update):
@@ -342,6 +353,28 @@ async def admin_callbacks(client, callback_query: CallbackQuery):
                                            "👉 **किसी एक को भेजने के लिए:** ID स्पेस मेसेज लिखें।\n"
                                            "`उदाहरण: 8383838487 Hello kese ho`")
     
+    elif data == "admin_sch_times":
+        markup = []
+        for st in GLOBAL_SCHEDULE_TIMES:
+            markup.append([InlineKeyboardButton(f"❌ Delete {st}", callback_data=f"admindelscht_{st}")])
+        markup.append([InlineKeyboardButton("➕ Add New Time", callback_data="admin_sch_add")])
+        markup.append([InlineKeyboardButton("🔙 Back", callback_data="admin_back")])
+        await callback_query.message.edit_text("📅 **Manage Pre-defined Schedule Times:**\nये समय यूज़र्स को बटन के रूप में दिखेंगे।", reply_markup=InlineKeyboardMarkup(markup))
+
+    elif data == "admin_back":
+        await callback_query.message.edit_text("👑 **Welcome to Admin Panel**\nChoose an option below:", reply_markup=admin_keyboard())
+
+    elif data == "admin_sch_add":
+        admin_state = "AWAIT_SCH_TIME"
+        await callback_query.message.reply("📅 **कृपया नया शेड्यूल्ड समय भेजें:**\n\n**Format:** `DD/MM/YYYY hh:mm AM/PM`\n**Example:** `24/06/2026 05:50 AM`")
+
+    elif data.startswith("admindelscht_"):
+        time_to_del = data.replace("admindelscht_", "")
+        if time_to_del in GLOBAL_SCHEDULE_TIMES:
+            GLOBAL_SCHEDULE_TIMES.remove(time_to_del)
+            save_data()
+        await admin_callbacks(client, callback_query._client.callback_query(id=callback_query.id, from_user=callback_query.from_user, message=callback_query.message, data="admin_sch_times"))
+
     elif data == "admin_del":
         if not recent_admin_messages:
             return await callback_query.message.reply("❌ **No recent messages/broadcasts found to delete.**")
@@ -471,14 +504,74 @@ async def select_schedule_choice(client, callback_query: CallbackQuery):
     user_id = callback_query.message.chat.id
     if callback_query.data == "sch_now":
         user_data[user_id]['step'] = 'READY'
-        # 🚀 यह मैसेज डिलीट हो जाएगा ताकि स्ट्रीम खत्म होने के बाद स्क्रीनशॉट की तरह फालतू न दिखे
         await callback_query.message.delete() 
         await process_stream_start(client, callback_query.message)
     else:
         user_data[user_id]['step'] = 'AWAIT_DATETIME'
-        await callback_query.message.edit_text("📅 **Schedule Time भेजो!**\n\n**Format:** `DD/MM/YYYY hh:mm AM/PM`\n**Example:** `26/05/2026 12:40 PM`")
+        markup = []
+        for st in GLOBAL_SCHEDULE_TIMES:
+            # Inline button format for specific times
+            markup.append([InlineKeyboardButton(f"🗓 {st}", callback_data=f"set_schtime_{st}")])
+        
+        # बटन के बिना मैन्युअल एंट्री का निर्देश
+        text = "📅 **Schedule Time चुनें या लिखें!**\n\n"
+        if GLOBAL_SCHEDULE_TIMES:
+            text += "👇 **एडमिन द्वारा दिए गए समय में से चुनें:**\n"
+        text += "\n✍️ **या अपना खुद का समय लिखकर भेजें:**\n**Format:** `DD/MM/YYYY hh:mm AM/PM`\n**Example:** `24/06/2026 05:50 AM`"
+        
+        await callback_query.message.edit_text(text, reply_markup=InlineKeyboardMarkup(markup) if markup else None)
 
-@bot.on_message(~filters.command(["start", "stop", "new", "admin"]))
+@bot.on_callback_query(filters.regex("^set_schtime_"))
+async def select_predefined_time(client, callback_query: CallbackQuery):
+    user_id = callback_query.message.chat.id
+    time_str = callback_query.data.replace("set_schtime_", "")
+    
+    # Create fake message to pass to AWAIT_DATETIME logic
+    class FakeMessage:
+        def __init__(self, text, chat_id):
+            self.text = text
+            self.chat = type('Chat', (), {'id': chat_id})()
+            self.id = callback_query.message.id
+    
+    await callback_query.message.edit_text(f"⏳ Processing schedule for {time_str}...")
+    await handle_all_messages(client, FakeMessage(time_str, user_id))
+
+@bot.on_message(filters.command("edit"))
+async def edit_cmd(client, message):
+    user_id = message.chat.id
+    init_user(user_id, message)
+    
+    markup = []
+    text = "🛠 **Manage Your Streams:**\nयहाँ से आप अपनी ग़लत बनी हुई या चल रही स्ट्रीम्स को Delete/Stop कर सकते हैं।\n\n"
+    
+    # Running streams
+    active_procs = [p for p in user_data[user_id].get('processes', []) if p['process'].poll() is None]
+    if active_procs:
+        text += "🔴 **Running Streams:**\n"
+        for idx, p in enumerate(active_procs, 1):
+            p_id = p.get('id', 'unknown')
+            plat = p.get('platform', 'Unknown')
+            text += f"{idx}. 🟢 {plat} - Started: {p['start_time'].strftime('%I:%M %p')}\n"
+            markup.append([InlineKeyboardButton(f"🛑 Stop Running: {plat} Stream {idx}", callback_data=f"user_stop_{p_id}")])
+    
+    # Scheduled streams
+    scheduled = user_data[user_id].get('scheduled_tasks', {})
+    if scheduled:
+        text += "\n⏳ **Scheduled Streams:**\n"
+        idx = 1
+        for s_id, s_info in list(scheduled.items()):
+            time_str = s_info['time_str']
+            plat = s_info['config'].get('platform', 'Unknown')
+            text += f"{idx}. 🗓 {plat} - Scheduled for: {time_str}\n"
+            markup.append([InlineKeyboardButton(f"❌ Cancel Scheduled: {plat} {time_str}", callback_data=f"user_cancelsch_{s_id}")])
+            idx += 1
+
+    if not active_procs and not scheduled:
+        text = "⚠️ **आपकी कोई भी स्ट्रीम अभी चल नहीं रही है और ना ही शेड्यूल है।**"
+
+    await message.reply(text, reply_markup=InlineKeyboardMarkup(markup) if markup else None)
+
+@bot.on_message(~filters.command(["start", "stop", "new", "admin", "edit"]))
 async def handle_all_messages(client, message):
     user_id = message.chat.id
     init_user(user_id, message)
@@ -595,6 +688,18 @@ async def handle_all_messages(client, message):
             admin_state = None
             return
 
+        elif admin_state == "AWAIT_SCH_TIME" and message.text:
+            try:
+                # Validation
+                dt_obj = datetime.strptime(message.text.strip(), "%d/%m/%Y %I:%M %p")
+                GLOBAL_SCHEDULE_TIMES.append(message.text.strip())
+                save_data()
+                await message.reply(f"✅ **Time Added Successfully!**\n`{message.text.strip()}`\n\nअब यह समय सभी यूज़र्स को बटन में दिखाई देगा।")
+                admin_state = None
+            except Exception as e:
+                await message.reply("❌ **Invalid Format!**\nकृपया इस फॉर्मेट में भेजें: `24/06/2026 05:50 AM`")
+            return
+
     # === Normal User Handling ===
     if not user_data[user_id]['authenticated'] and step != 'PASSWORD':
         user_data[user_id]['step'] = 'PASSWORD'
@@ -723,24 +828,94 @@ async def handle_all_messages(client, message):
 
     elif step == 'AWAIT_DATETIME':
         try:
-            dt_obj = datetime.strptime(message.text.strip(), "%d/%m/%Y %I:%M %p")
+            time_str = message.text.strip()
+            dt_obj = datetime.strptime(time_str, "%d/%m/%Y %I:%M %p")
             now_ist = get_ist_time().replace(tzinfo=None)
             if dt_obj <= now_ist:
-                return await message.reply("⚠️ **समय (Time) भविष्य का होना चाहिए!** कृपया फिर से भेजें।\nExample: `26/05/2026 12:40 PM`")
+                # If pre-defined time is passed, allow it logically but warn if past
+                try: 
+                    await client.send_message(user_id, "⚠️ **समय (Time) भविष्य का होना चाहिए!** कृपया आगे का समय भेजें।")
+                except:
+                    await message.reply("⚠️ **समय (Time) भविष्य का होना चाहिए!** कृपया आगे का समय भेजें।")
+                return
             
-            user_data[user_id]['schedule_time'] = dt_obj
             user_data[user_id]['step'] = 'READY'
-            
             delay = (dt_obj - now_ist).total_seconds()
-            await message.reply(f"✅ **Stream Scheduled!**\nयह वीडियो आटोमेटिकली `{message.text.strip()}` पर चालू हो जाएगा।")
             
-            asyncio.create_task(scheduled_stream_starter(client, message, delay))
+            # --- Snapshot Configuration ---
+            # (अगर यूज़र दूसरी स्ट्रीम बनायेगा तो पहली स्ट्रीम का डेटा डिलीट न हो)
+            stream_config = {
+                'file_path': user_data[user_id].get('file_path'),
+                'video_url': user_data[user_id].get('video_url'),
+                'custom_audio_path': user_data[user_id].get('custom_audio_path'),
+                'platform': user_data[user_id].get('platform', 'Custom'),
+                'loop_hours': user_data[user_id].get('loop_hours', 0),
+                'rtmp_url': user_data[user_id].get('rtmp_url'),
+                'orientation': user_data[user_id].get('orientation', 'landscape'),
+                'custom_image_path': user_data[user_id].get('custom_image_path'),
+                'is_image_only': user_data[user_id].get('is_image_only', False),
+                'watermark_text': user_data[user_id].get('watermark_text')
+            }
+            
+            s_id = str(uuid.uuid4())[:8]
+            task = asyncio.create_task(scheduled_stream_starter(client, message, delay, stream_config, s_id))
+            
+            user_data[user_id]['scheduled_tasks'][s_id] = {
+                'task': task,
+                'time_str': time_str,
+                'config': stream_config
+            }
+            
+            try:
+                await client.send_message(user_id, f"✅ **Stream Scheduled!**\nयह वीडियो आटोमेटिकली `{time_str}` पर चालू हो जाएगा।\nअगर कोई गलती हुई हो तो `/edit` टाइप करें।")
+            except:
+                await message.reply(f"✅ **Stream Scheduled!**\nयह वीडियो आटोमेटिकली `{time_str}` पर चालू हो जाएगा।\nअगर कोई गलती हुई हो तो `/edit` टाइप करें।")
+            
         except Exception as e:
-            await message.reply("❌ **Invalid Format!**\nकृपया इस फॉर्मेट में भेजें: `26/05/2026 12:40 PM`")
+            try:
+                await client.send_message(user_id, "❌ **Invalid Format!**\nकृपया इस फॉर्मेट में भेजें: `24/06/2026 05:50 AM`")
+            except:
+                await message.reply("❌ **Invalid Format!**\nकृपया इस फॉर्मेट में भेजें: `24/06/2026 05:50 AM`")
+
+@bot.on_callback_query(filters.regex("^user_cancelsch_"))
+async def cancel_sch_callback(client, callback_query: CallbackQuery):
+    user_id = callback_query.message.chat.id
+    s_id = callback_query.data.replace("user_cancelsch_", "")
+    if s_id in user_data[user_id].get('scheduled_tasks', {}):
+        user_data[user_id]['scheduled_tasks'][s_id]['task'].cancel()
+        del user_data[user_id]['scheduled_tasks'][s_id]
+        await callback_query.message.edit_text("✅ **Scheduled Stream Cancelled / Delete हो गई!**\nअब आप `/new` से नई बना सकते हैं।")
+    else:
+        await callback_query.message.edit_text("❌ **यह शेड्यूल पहले ही कैंसिल हो चुका है या चल चुका है।**")
 
-async def scheduled_stream_starter(client, message, delay):
-    await asyncio.sleep(delay)
-    await process_stream_start(client, message)
+@bot.on_callback_query(filters.regex("^user_stop_"))
+async def stop_running_callback(client, callback_query: CallbackQuery):
+    user_id = callback_query.message.chat.id
+    p_id = callback_query.data.replace("user_stop_", "")
+    
+    stopped = False
+    for p in user_data[user_id].get('processes', []):
+        if p.get('id') == p_id and p['process'].poll() is None:
+            user_data[user_id]['manual_stop'] = True
+            p['process'].terminate()
+            stopped = True
+            break
+            
+    if stopped:
+        await callback_query.message.edit_text("✅ **Running Stream को सफलतापूर्वक बंद कर दिया गया है!**")
+    else:
+        await callback_query.message.edit_text("❌ **यह स्ट्रीम पहले ही बंद हो चुकी है।**")
+
+async def scheduled_stream_starter(client, message, delay, stream_config, s_id):
+    try:
+        await asyncio.sleep(delay)
+        user_id = message.chat.id
+        # Remove from pending queue as it starts
+        if s_id in user_data[user_id].get('scheduled_tasks', {}):
+            del user_data[user_id]['scheduled_tasks'][s_id]
+        await handle_video(client, message, stream_config)
+    except asyncio.CancelledError:
+        pass # Stream was cancelled by user
 
 async def process_stream_start(client, message):
     await handle_video(client, message)
@@ -754,15 +929,21 @@ def get_direct_url(url):
     except:
         return url
 
-async def handle_video(client, message):
+async def handle_video(client, message, stream_config=None):
     user_id = message.chat.id
     
+    # अगर कॉन्फ़िग स्नैपशॉट (शेड्यूल से) आया है, तो उसका इस्तेमाल करें। नहीं तो करंट डेटा का।
+    config = stream_config if stream_config else user_data[user_id]
+    
     user_data[user_id]['manual_stop'] = False
-    msg = await client.send_message(user_id, "⏳ **Preparing Stream Environment...**")
+    try:
+        msg = await client.send_message(user_id, "⏳ **Preparing Stream Environment...**")
+    except:
+        msg = await message.reply("⏳ **Preparing Stream Environment...**")
     
-    file_path = user_data[user_id].get('file_path')
-    video_url = user_data[user_id].get('video_url')
-    custom_audio = user_data[user_id].get('custom_audio_path')
+    file_path = config.get('file_path')
+    video_url = config.get('video_url')
+    custom_audio = config.get('custom_audio_path')
     
     input_source = file_path
     if video_url:
@@ -771,11 +952,11 @@ async def handle_video(client, message):
         if not input_source:
             return await msg.edit_text("❌ **Failed to extract link!**")
             
-    platform = user_data[user_id].get('platform', 'Custom')
-    loop_hours = user_data[user_id].get('loop_hours', 0)
-    rtmp_url = user_data[user_id]['rtmp_url']
-    orientation = user_data[user_id].get('orientation', 'landscape')
-    image_path = user_data[user_id].get('custom_image_path')
+    platform = config.get('platform', 'Custom')
+    loop_hours = config.get('loop_hours', 0)
+    rtmp_url = config.get('rtmp_url')
+    orientation = config.get('orientation', 'landscape')
+    image_path = config.get('custom_image_path')
     
     user_data[user_id]['total_streams'] += 1
     user_data[user_id]['last_stream'] = get_ist_time()
@@ -797,7 +978,7 @@ async def handle_video(client, message):
         total_stream_time = vid_duration
         actual_stream_seconds = vid_duration
 
-    is_image_only = user_data[user_id].get('is_image_only', False)
+    is_image_only = config.get('is_image_only', False)
 
     # --- FFMPEG Command Setup ---
     command = ["ffmpeg", "-re"]
@@ -825,7 +1006,7 @@ async def handle_video(client, message):
             vf_filter = "scale=1920:1080:force_original_aspect_ratio=decrease,pad=1920:1080:(ow-iw)/2:(oh-ih)/2:color=black"
         
         # --- NEW: SLOW SMOOTH MOVING WATERMARK ---
-        wm_text = user_data[user_id].get('watermark_text')
+        wm_text = config.get('watermark_text')
         if wm_text:
             safe_text = wm_text.replace("'", "\\'").replace(":", "\\:")
             # sin(t/10) और cos(t/15) का मतलब है कि यह बहुत धीरे-धीरे (Slowly) पूरी स्क्रीन पर गोल घूमेगा
@@ -872,7 +1053,7 @@ async def handle_video(client, message):
             command.extend(["-b:v", "6000k", "-maxrate", "6000k", "-minrate", "6000k", "-bufsize", "12000k", "-nal-hrd", "cbr"])
         else:
             # YouTube Strict Bitrate & Keyframe (GOP) Fix for Landscape Video
-            wm_text = user_data[user_id].get('watermark_text')
+            wm_text = config.get('watermark_text')
             
             if wm_text:
                 safe_text = wm_text.replace("'", "\\'").replace(":", "\\:")
@@ -910,12 +1091,15 @@ async def handle_video(client, message):
     start_t = get_ist_time()
     end_t = start_t + timedelta(seconds=actual_stream_seconds) if loop_hours > 0 else "Auto (Depends on Video)"
     
+    unique_p_id = str(uuid.uuid4())[:8]
     user_data[user_id]['processes'].append({
+        'id': unique_p_id,
         'process': process_obj,
         'start_time': start_t,
         'end_time': end_t,
         'loop_hours': loop_hours,
-        'is_image': is_image_only
+        'is_image': is_image_only,
+        'platform': platform
     })
     
     asyncio.create_task(monitor_stream(client, message.chat.id, process_obj, file_path, live_msg.id, loop_hours, platform, total_stream_time, image_path, custom_audio))