Update app.py

app.py

@@ -4,19 +4,19 @@ import asyncio
 import threading
 from queue import Queue, Empty
 import os
+import time
 from telegram import Update
 from telegram.ext import ApplicationBuilder, CommandHandler, ContextTypes
 
-# Env vars or hardcode your Telegram bot token here
 BOT_TOKEN = os.environ.get("BOT_TOKEN", "7510817339:AAHFZoFzPkUO_-nAwfh9bjY_qHsWMprM2PI")
-# Globals
-bot_app = None  # Telegram Application instance
+
+bot_app = None
 bot_thread = None
 bot_running = False
 log_queue = Queue()
 telegram_cmd_logs = []
 
-# -- Telegram Bot Handlers --
+# --- Telegram Bot Handlers ---
 
 async def bash_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
     cmd = ' '.join(context.args)
@@ -26,7 +26,6 @@ async def bash_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
 
     log_queue.put(f"[Telegram] Received command: {cmd}")
 
-    # Run command and collect output (limit max 4000 chars for Telegram)
     try:
         proc = await asyncio.create_subprocess_shell(
             cmd,
@@ -41,14 +40,12 @@ async def bash_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
         output = f"Error running command: {e}"
 
     await update.message.reply_text(f"$ {cmd}\n{output}")
-
     telegram_cmd_logs.append(f"$ {cmd}\n{output}")
-    # Limit logs to last 50 commands
     if len(telegram_cmd_logs) > 50:
         telegram_cmd_logs.pop(0)
 
 def start_bot():
-    global bot_app, bot_running
+    global bot_app, bot_running, bot_thread
 
     if bot_running:
         log_queue.put("[Bot] Already running.")
@@ -68,12 +65,12 @@ def start_bot():
         asyncio.set_event_loop(loop)
         loop.run_until_complete(run_bot())
 
-    global bot_thread
     bot_thread = threading.Thread(target=runner, daemon=True)
     bot_thread.start()
 
 def stop_bot():
     global bot_app, bot_running
+
     if not bot_running:
         log_queue.put("[Bot] Bot is not running.")
         return
@@ -86,7 +83,7 @@ def stop_bot():
     asyncio.run(shutdown())
     bot_running = False
 
-# -- Gradio Web Terminal --
+# --- Terminal Execution ---
 
 def live_terminal(cmd):
     if not cmd.strip():
@@ -105,8 +102,9 @@ def live_terminal(cmd):
 def run_terminal(cmd):
     return live_terminal(cmd)
 
+# --- Log Update Function ---
+
 def update_log_box():
-    # Show last 20 logs
     logs = []
     try:
         while True:
@@ -118,25 +116,23 @@ def update_log_box():
     logs.extend(telegram_cmd_logs[-20:])
     return "\n".join(logs[-20:])
 
-# -- Gradio UI --
+# --- Gradio UI ---
 
 with gr.Blocks() as demo:
-    gr.Markdown("### Live Terminal with Telegram Bot Control")
+    gr.Markdown("## 🖥️ Live Terminal + Telegram Bot Interface")
 
     with gr.Row():
         terminal_output = gr.Textbox(value="$ ", label="Live Terminal", lines=20, interactive=False)
-        cmd_input = gr.Textbox(placeholder="Type command here and press Enter", label="Command Input")
+        cmd_input = gr.Textbox(placeholder="Type command here", label="Command Input")
 
     with gr.Row():
-        start_btn = gr.Button("Start Telegram Bot")
-        stop_btn = gr.Button("Stop Telegram Bot")
+        start_btn = gr.Button("▶️ Start Telegram Bot")
+        stop_btn = gr.Button("⏹️ Stop Telegram Bot")
 
-    log_output = gr.Textbox(value="", label="Bot Logs & Telegram Bash Command Logs", lines=15, interactive=False)
+    log_output = gr.Textbox(value="", label="📜 Bot & Telegram Logs", lines=15, interactive=False)
 
-    # Command input triggers terminal
     cmd_input.submit(run_terminal, inputs=cmd_input, outputs=terminal_output)
 
-    # Start/stop buttons update logs
     def start_click():
         start_bot()
         return update_log_box()
@@ -148,10 +144,13 @@ with gr.Blocks() as demo:
     start_btn.click(start_click, outputs=log_output)
     stop_btn.click(stop_click, outputs=log_output)
 
-    # Auto-update logs every 3 seconds
-    def auto_update():
-        return update_log_box()
+    def background_log_updater():
+        while True:
+            time.sleep(3)
+            yield update_log_box()
 
-    demo.load(auto_update, None, log_output, every=3)
+    demo.load(None, outputs=log_output, every=None).then(
+        background_log_updater, outputs=log_output
+    )
 
 demo.launch()