Update app.py

app.py

@@ -1,69 +1,167 @@
 import os
-import gradio as gr
-import logging
-from multiprocessing import Process
+import asyncio
+import threading
 from pyrogram import Client, filters
-from io import StringIO
-from time import sleep
+import gradio as gr
 
-# ---------- FIXED CONFIG ----------
 API_ID = int(os.environ.get("API_ID", "6"))
 API_HASH = os.environ.get("API_HASH", "eb06d4abfb49dc3eeb1aeb98ae0f581e")
 BOT_TOKEN = os.environ.get("BOT_TOKEN", "7510817339:AAHFZoFzPkUO_-nAwfh9bjY_qHsWMprM2PI")
 
-# ---------- LOGGING ----------
-log_stream = StringIO()
-logging.basicConfig(stream=log_stream, level=logging.INFO, format="%(asctime)s - %(message)s")
-logger = logging.getLogger()
 
-# ---------- BOT FUNCTION ----------
-def run_bot():
-    app = Client("my_bot", api_id=API_ID, api_hash=API_HASH, bot_token=BOT_TOKEN)
+app = Client("my_bot", api_id=API_ID, api_hash=API_HASH, bot_token=BOT_TOKEN)
+
+bot_running = False
+bot_lock = threading.Lock()
 
-    @app.on_message(filters.command("start") & filters.private)
-    async def start_cmd(client, message):
-        await message.reply("hi")
-        logger.info(f"Replied to /start from user ID: {message.from_user.id}")
+command_history = []
+history_pos = -1
+output_logs = ""
 
-    logger.info("Bot started.")
-    app.run()
-    logger.info("Bot stopped.")
+def append_log(text):
+    global output_logs
+    output_logs += text + "\n"
+    # Limit to last 20k chars to keep it manageable
+    if len(output_logs) > 20000:
+        output_logs = output_logs[-20000:]
 
-# ---------- PROCESS CONTROL ----------
-bot_process = None
+# Telegram /start command handler
+@app.on_message(filters.command("start"))
+async def start_handler(client, message):
+    await message.reply_text("Hi 👋")
+    append_log(f"[BOT] Replied 'Hi' to user {message.from_user.id}")
 
 def start_bot():
-    global bot_process
-    if bot_process and bot_process.is_alive():
-        return "Bot is already running."
-    bot_process = Process(target=run_bot)
-    bot_process.start()
-    sleep(2)
-    return "✅ Bot started."
+    global bot_running
+    with bot_lock:
+        if not bot_running:
+            asyncio.run(app.start())
+            bot_running = True
+            append_log("[SYSTEM] Bot started")
+        else:
+            append_log("[SYSTEM] Bot already running")
 
 def stop_bot():
-    global bot_process
-    if bot_process and bot_process.is_alive():
-        bot_process.terminate()
-        bot_process.join()
-        logger.info("Bot manually stopped.")
-        return "🛑 Bot stopped."
-    return "Bot is not running."
-
-def get_logs():
-    return log_stream.getvalue()[-20000:]
-
-# ---------- GRADIO UI ----------
-with gr.Blocks() as ui:
-    gr.Markdown("## 🤖 Telegram Bot Control Panel")
-    start_btn = gr.Button("🚀 Start Bot")
-    stop_btn = gr.Button("🛑 Stop Bot")
-    refresh_btn = gr.Button("🔄 Refresh Logs")
-    logs_box = gr.Textbox(label="📜 Bot Logs", lines=25)
-
-    start_btn.click(fn=start_bot, outputs=logs_box)
-    stop_btn.click(fn=stop_bot, outputs=logs_box)
-    refresh_btn.click(fn=get_logs, outputs=logs_box)
-
-if __name__ == "__main__":
-    ui.launch()
+    global bot_running
+    with bot_lock:
+        if bot_running:
+            asyncio.run(app.stop())
+            bot_running = False
+            append_log("[SYSTEM] Bot stopped")
+        else:
+            append_log("[SYSTEM] Bot already stopped")
+
+def run_command(cmd):
+    """
+    This simulates running a command in the 'terminal'.
+    We support:
+    - telegram commands: /start etc will be sent to the bot (simulate)
+    - shell commands: run on local shell and return output
+    """
+    append_log(f"> {cmd}")
+    # For example purposes, just handle /start command specially
+    if cmd.strip() == "/start":
+        # simulate bot reply (because real bot interaction needs telegram user)
+        append_log("[BOT] Hi 👋")
+    else:
+        # Run shell command and capture output
+        import subprocess
+        try:
+            output = subprocess.check_output(cmd, shell=True, stderr=subprocess.STDOUT, timeout=10, text=True)
+            append_log(output.strip())
+        except subprocess.CalledProcessError as e:
+            append_log(f"Error:\n{e.output}")
+        except Exception as e:
+            append_log(f"Exception: {str(e)}")
+    return output_logs
+
+# --- Gradio interface callbacks ---
+
+def on_send_command(cmd, hist_pos):
+    global command_history, history_pos
+    if cmd.strip():
+        command_history.append(cmd.strip())
+        history_pos = len(command_history)  # reset position after new command
+        logs = run_command(cmd)
+        return "", logs, history_pos
+    else:
+        return "", output_logs, hist_pos
+
+def on_up_arrow(hist_pos):
+    global command_history
+    if command_history and hist_pos > 0:
+        hist_pos -= 1
+        return command_history[hist_pos], output_logs, hist_pos
+    return gr.Textbox.update(), output_logs, hist_pos
+
+def on_down_arrow(hist_pos):
+    global command_history
+    if command_history and hist_pos < len(command_history) - 1:
+        hist_pos += 1
+        return command_history[hist_pos], output_logs, hist_pos
+    elif hist_pos == len(command_history) - 1:
+        hist_pos += 1
+        return "", output_logs, hist_pos
+    return gr.Textbox.update(), output_logs, hist_pos
+
+def start_bot_ui():
+    threading.Thread(target=start_bot).start()
+    return output_logs
+
+def stop_bot_ui():
+    threading.Thread(target=stop_bot).start()
+    return output_logs
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Telegram Bot Terminal with Start/Stop and Command History")
+
+    with gr.Row():
+        start_btn = gr.Button("🚀 Start Bot")
+        stop_btn = gr.Button("🛑 Stop Bot")
+    logs = gr.Textbox(label="Logs / Terminal Output", lines=20, interactive=False)
+    cmd_input = gr.Textbox(label="Enter command", lines=1, interactive=True)
+
+    # Hidden state for history position
+    hist_pos_state = gr.State(value=len(command_history))
+
+    # Button clicks
+    start_btn.click(start_bot_ui, outputs=logs)
+    stop_btn.click(stop_bot_ui, outputs=logs)
+
+    # Send command on Enter
+    cmd_input.submit(on_send_command, inputs=[cmd_input, hist_pos_state], outputs=[cmd_input, logs, hist_pos_state])
+
+    # JavaScript events to detect arrow keys
+    # We use custom JS to send event to Gradio for arrow keys pressed in the textbox
+    # Note: Gradio does not natively support up/down key callbacks,
+    # so we inject this JS trick below.
+
+    cmd_input.js(
+        """
+        () => {
+            const input = document.querySelector('textarea[aria-label="Enter command"]');
+            input.addEventListener('keydown', e => {
+                if (e.key === "ArrowUp") {
+                    e.preventDefault();
+                    gradioApp().getElementById('up-arrow-btn').click();
+                }
+                if (e.key === "ArrowDown") {
+                    e.preventDefault();
+                    gradioApp().getElementById('down-arrow-btn').click();
+                }
+            });
+        }
+        """
+    )
+
+    # Hidden buttons for up/down arrow to call Python
+    up_btn = gr.Button(visible=False, elem_id="up-arrow-btn")
+    down_btn = gr.Button(visible=False, elem_id="down-arrow-btn")
+
+    up_btn.click(on_up_arrow, inputs=hist_pos_state, outputs=[cmd_input, logs, hist_pos_state])
+    down_btn.click(on_down_arrow, inputs=hist_pos_state, outputs=[cmd_input, logs, hist_pos_state])
+
+    # Initialize logs textbox content
+    logs.value = output_logs
+
+demo.launch()