Create app.py

app.py

@@ -0,0 +1,163 @@
+import os
+import json
+import string
+import random
+import subprocess
+import threading
+import time
+import requests
+from flask import Flask
+import telebot
+from telebot import types
+
+# === CONFIG ===
+BOT_TOKEN = "8018538301:AAGRSngj4Bj5nqizPsU3Zux48Yr6fpaZ7To"  # 🔁 Replace this with your Telegram Bot Token
+SCRIPT_DIR = "scripts"
+DB_FILE = "scripts.json"
+PORT = int(os.environ.get('PORT', 7860))
+PING_URL = "https://dexsecon-xoomhost.hf.space/ping"  # ✅ Auto-ping URL (your space)
+
+# === INIT ===
+os.makedirs(SCRIPT_DIR, exist_ok=True)
+app = Flask(__name__)
+bot = telebot.TeleBot(BOT_TOKEN)
+file_upload_mode = {}
+status_map = {}
+
+# === UTILITIES ===
+def generate_filename():
+    return ''.join(random.choices(string.ascii_letters + string.digits, k=6)) + ".py"
+
+def load_db():
+    if not os.path.exists(DB_FILE):
+        with open(DB_FILE, "w") as f:
+            json.dump({}, f)
+    with open(DB_FILE, "r") as f:
+        return json.load(f)
+
+def save_db(data):
+    with open(DB_FILE, "w") as f:
+        json.dump(data, f, indent=4)
+
+def update_status(filename, proc):
+    stdout, stderr = proc.communicate()
+    if proc.returncode == 0:
+        status_map[filename] = 'sleep'
+    else:
+        status_map[filename] = 'error'
+
+def auto_ping():
+    while True:
+        try:
+            requests.get(PING_URL)
+            print("✅ Ping sent to HF")
+        except Exception as e:
+            print(f"❌ Ping failed: {e}")
+        time.sleep(300)  # 5 minutes
+
+# === TELEGRAM COMMANDS ===
+@bot.message_handler(commands=['start'])
+def start_message(message):
+    markup = types.ReplyKeyboardMarkup(resize_keyboard=True)
+    markup.add('📥 Upload Script', '📜 My Scripts')
+    markup.add('📊 Status', '🔄 Refresh')
+    bot.send_message(message.chat.id, "👋 Welcome to XOOM Bot!", reply_markup=markup)
+
+@bot.message_handler(commands=['help'])
+def help_message(message):
+    bot.send_message(message.chat.id, "Available:\n/start – Menu\n/help – Help info\n/status – Status of your scripts\n/myscripts – Show your scripts")
+
+@bot.message_handler(commands=['status'])
+def status_all(message):
+    chat_id = str(message.chat.id)
+    db = load_db()
+    scripts = db.get(chat_id, [])
+    counts = {"running": 0, "error": 0, "sleep": 0}
+    for s in scripts:
+        stat = status_map.get(s, 'unknown')
+        if stat in counts:
+            counts[stat] += 1
+    msg = f"📊 Status:\nRunning: {counts['running']} 🔄\nDone: {counts['sleep']} ✅\nError: {counts['error']} ❌"
+    bot.send_message(message.chat.id, msg)
+
+@bot.message_handler(commands=['myscripts'])
+def my_scripts(message):
+    chat_id = str(message.chat.id)
+    db = load_db()
+    scripts = db.get(chat_id, [])
+    if not scripts:
+        bot.send_message(message.chat.id, "📂 No scripts found.")
+        return
+    msg = "📜 Your Scripts:\n"
+    for s in scripts:
+        stat = status_map.get(s, 'unknown')
+        icon = {'running': '🔄', 'sleep': '✅', 'error': '❌'}.get(stat, '❓')
+        msg += f"{icon} {s} - {stat}\n"
+    bot.send_message(message.chat.id, msg)
+
+# === BUTTON HANDLERS ===
+@bot.message_handler(func=lambda m: m.text == '📥 Upload Script')
+def upload_script_btn(message):
+    bot.send_message(message.chat.id, "📂 Send your .py file now.")
+    file_upload_mode[message.chat.id] = True
+
+@bot.message_handler(func=lambda m: m.text == '📜 My Scripts')
+def my_scripts_btn(message):
+    my_scripts(message)
+
+@bot.message_handler(func=lambda m: m.text == '📊 Status')
+def status_btn(message):
+    status_all(message)
+
+@bot.message_handler(func=lambda m: m.text == '🔄 Refresh')
+def refresh_btn(message):
+    start_message(message)
+
+# === FILE HANDLER ===
+@bot.message_handler(content_types=['document'])
+def handle_file(message):
+    if message.chat.id not in file_upload_mode:
+        return
+    file_info = bot.get_file(message.document.file_id)
+    if not file_info.file_path.endswith(".py"):
+        bot.send_message(message.chat.id, "❌ Only .py files allowed.")
+        return
+
+    file_data = bot.download_file(file_info.file_path)
+    filename = generate_filename()
+    filepath = os.path.join(SCRIPT_DIR, filename)
+
+    with open(filepath, "wb") as f:
+        f.write(file_data)
+
+    chat_id = str(message.chat.id)
+    db = load_db()
+    db.setdefault(chat_id, []).append(filename)
+    save_db(db)
+
+    try:
+        proc = subprocess.Popen(["python3", filepath], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        status_map[filename] = 'running'
+        bot.send_message(message.chat.id, f"✅ Script `{filename}` is running.")
+        threading.Thread(target=lambda: update_status(filename, proc)).start()
+    except Exception as e:
+        status_map[filename] = 'error'
+        bot.send_message(message.chat.id, f"❌ Failed: {e}")
+
+    file_upload_mode.pop(message.chat.id, None)
+
+# === FLASK ROUTES ===
+@app.route('/')
+def home():
+    return "✅ XOOM Bot Running - dexsecon/xoomhost"
+
+@app.route('/ping')
+def ping():
+    return "✅ Ping OK"
+
+# === START EVERYTHING ===
+threading.Thread(target=lambda: bot.infinity_polling(), daemon=True).start()
+threading.Thread(target=auto_ping, daemon=True).start()
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=PORT)