Update admin.py

admin.py

@@ -1557,6 +1557,38 @@ def format_message_to_html(text):
         
     return text
 
+# Define Bot Menus for post-broadcast restoration
+BOT_MENUS = {
+    "main": {
+        "text": "Welcome to the @iceboxai_bot.\n\n✨ <b>What you can create:</b>\n• Realistic photos\n• Anime & illustration\n• Cinematic portraits\n• Fantasy & concept art\n• Logos & product visuals\n\nChoose an option below to get started 👇",
+        "keyboard": [
+            [{"text": "Generate Image", "callback_data": "generate_mode"}],
+            [{"text": "Generate Voice", "url": "https://t.me/iceboxai_voice_bot"}],
+            [{"text": "GPT-5.1", "url": "https://t.me/chaticeboxai_bot"}],
+            [{"text": "My Profile", "callback_data": "profile"}],
+            [{"text": "Help & Support", "callback_data": "help"}]
+        ]
+    },
+    "voice": {
+        "text": "<b>Welcome to IceboxAI Voice</b>\n\nGenerate natural text-to-speech in seconds.\n\nSelect an option below to begin.\nEnter your text and choose your voice.",
+        "keyboard": [
+            [{"text": "Generate Voice", "callback_data": "generate_voice"}],
+            [{"text": "Generate Image", "url": "https://t.me/iceboxai_bot"}],
+            [{"text": "GPT-5.1", "url": "https://t.me/chaticeboxai_bot"}],
+            [{"text": "My Profile", "callback_data": "profile"}],
+            [{"text": "Help & Support", "callback_data": "help"}]
+        ]
+    },
+    "chat": {
+        "text": "<b>Welcome to Icebox AI Chat!</b>\n\nI'm an AI assistant that can help you:\n• Answer questions\n• Write and summarize text\n• Search the web\n\n<pre>\n📟 How to start:\nJust send any message.\n\nExamples:\n> Give me business ideas\n> Summarize this article\n> Find the latest AI news\n</pre>\n\n<b>⚙️ Commands:</b>\n• /help — show help\n• /model — model info\n• /clear — delete chat history",
+        "keyboard": [
+            [{"text": "⚙️ Model Settings", "callback_data": "select_model"}],
+            [{"text": "🗑️ Clear Chat History", "callback_data": "clear_history"}],
+            [{"text": "🏞️ Create Image", "url": "https://t.me/iceboxai_bot"}]
+        ]
+    }
+}
+
 @app.route('/api/broadcast', methods=['POST'])
 @login_required
 def api_broadcast():
@@ -1624,6 +1656,17 @@ def api_broadcast():
             
             if resp.status_code == 200:
                 sent_count += 1
+                # Restore Home Menu so user has keyboard back
+                menu = BOT_MENUS.get(bot_target)
+                if menu:
+                    import json
+                    menu_payload = {
+                        "chat_id": chat_id,
+                        "text": menu["text"],
+                        "parse_mode": "HTML",
+                        "reply_markup": json.dumps({"inline_keyboard": menu["keyboard"]})
+                    }
+                    requests.post(url, json=menu_payload, timeout=5)
             else:
                 error_detail = resp.json().get('description', 'Unknown error')
                 print(f"Broadcast Failed for {chat_id}: {error_detail}")
@@ -1699,12 +1742,25 @@ def cron_reset_coins():
                 if "/bot" in base_url.lower(): url = f"{base_url}{token}/sendMessage"
                 else: url = f"{base_url}/bot{token}/sendMessage"
                 
-                requests.post(url, json={
+                res_notify = requests.post(url, json={
                     "chat_id": chat_id,
                     "text": formatted_message,
                     "parse_mode": "HTML"
                 }, timeout=5)
-                sent_count += 1
+                
+                if res_notify.status_code == 200:
+                    sent_count += 1
+                    # Restore Home Menu
+                    menu = BOT_MENUS.get('main')
+                    if menu:
+                        import json
+                        menu_payload = {
+                            "chat_id": chat_id,
+                            "text": menu["text"],
+                            "parse_mode": "HTML",
+                            "reply_markup": json.dumps({"inline_keyboard": menu["keyboard"]})
+                        }
+                        requests.post(url, json=menu_payload, timeout=5)
             except:
                 continue