| import os |
| import tempfile |
| import shutil |
| import threading |
| import time |
| import requests |
|
|
| from flask import Flask |
|
|
| from pyrogram import Client, filters |
| from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton |
|
|
| from huggingface_hub import HfApi, hf_hub_download |
|
|
| |
| |
| |
|
|
| API_ID = int(os.environ["API_ID"]) |
| API_HASH = os.environ["API_HASH"] |
| BOT_TOKEN = os.environ["BOT_TOKEN"] |
| HF_TOKEN = os.environ["HF_TOKEN"] |
|
|
| DATASET_REPO = "Cxops/Bzxv" |
|
|
| |
| KEEP_ALIVE_URL = "https://cxops-push.hf.space/" |
| PING_INTERVAL = 300 |
|
|
| |
| |
| |
|
|
| web = Flask(__name__) |
|
|
| @web.route("/") |
| def home(): |
| return "Bot Running..." |
|
|
| def run_web(): |
| web.run(host="0.0.0.0", port=7860) |
|
|
| |
| |
| |
|
|
| def keep_alive(): |
| headers = { |
| "User-Agent": "Mozilla/5.0 (KeepAliveBot)" |
| } |
|
|
| while True: |
| try: |
| r = requests.get(KEEP_ALIVE_URL, headers=headers, timeout=20) |
| print(f"[KEEPALIVE] Pinged {KEEP_ALIVE_URL} | Status: {r.status_code}") |
| except Exception as e: |
| print(f"[KEEPALIVE ERROR] {e}") |
|
|
| time.sleep(PING_INTERVAL) |
|
|
| |
| |
| |
|
|
| hf = HfApi(token=HF_TOKEN) |
|
|
| |
| |
| |
|
|
| app = Client( |
| "storage_bot", |
| api_id=API_ID, |
| api_hash=API_HASH, |
| bot_token=BOT_TOKEN |
| ) |
|
|
| |
| |
| |
|
|
| def cleanup(path): |
| try: |
| if os.path.isdir(path): |
| shutil.rmtree(path, ignore_errors=True) |
| elif os.path.isfile(path): |
| os.remove(path) |
| except Exception: |
| pass |
|
|
| |
| |
| |
|
|
| @app.on_message(filters.command("start")) |
| async def start(_, message): |
| await message.reply_text( |
| "Send me a file.\n\n" |
| "/files - List uploaded files" |
| ) |
|
|
| @app.on_message(filters.command("files")) |
| async def files(_, message): |
| user_id = str(message.from_user.id) |
|
|
| try: |
| repo_files = hf.list_repo_files( |
| repo_id=DATASET_REPO, |
| repo_type="dataset" |
| ) |
|
|
| user_files = [ |
| f for f in repo_files |
| if f.startswith(f"{user_id}/") |
| ] |
|
|
| if not user_files: |
| return await message.reply_text("You have no uploaded files.") |
|
|
| buttons = [ |
| [ |
| InlineKeyboardButton( |
| os.path.basename(path)[:60], |
| callback_data=f"get:{path}" |
| ) |
| ] |
| for path in user_files |
| ] |
|
|
| await message.reply_text( |
| "Your uploaded files:", |
| reply_markup=InlineKeyboardMarkup(buttons) |
| ) |
|
|
| except Exception as e: |
| await message.reply_text(f"Error:\n{e}") |
|
|
| |
| |
| |
|
|
| @app.on_callback_query(filters.regex("^get:")) |
| async def get_file(_, query): |
| path = query.data.replace("get:", "") |
|
|
| temp_dir = tempfile.mkdtemp() |
|
|
| try: |
| await query.message.edit_text("Downloading from dataset...") |
|
|
| local_file = hf_hub_download( |
| repo_id=DATASET_REPO, |
| repo_type="dataset", |
| filename=path, |
| token=HF_TOKEN, |
| local_dir=temp_dir |
| ) |
|
|
| await query.message.reply_document( |
| local_file, |
| caption=os.path.basename(path) |
| ) |
|
|
| try: |
| await query.message.delete() |
| except Exception: |
| pass |
|
|
| except Exception as e: |
| await query.message.reply_text(f"Error:\n{e}") |
|
|
| finally: |
| cleanup(temp_dir) |
|
|
| |
| |
| |
|
|
| @app.on_message( |
| filters.document | |
| filters.video | |
| filters.audio | |
| filters.photo |
| ) |
| async def upload_file(_, message): |
|
|
| temp_dir = tempfile.mkdtemp() |
|
|
| try: |
| status = await message.reply_text("Downloading file...") |
|
|
| if message.document: |
| media = message.document |
| filename = media.file_name |
|
|
| elif message.video: |
| media = message.video |
| filename = media.file_name or f"{media.file_unique_id}.mp4" |
|
|
| elif message.audio: |
| media = message.audio |
| filename = media.file_name or f"{media.file_unique_id}.mp3" |
|
|
| else: |
| media = message.photo[-1] |
| filename = f"{media.file_unique_id}.jpg" |
|
|
| local_path = await message.download( |
| file_name=os.path.join(temp_dir, filename) |
| ) |
|
|
| await status.edit_text("Uploading to Hugging Face...") |
|
|
| repo_path = f"{message.from_user.id}/{filename}" |
|
|
| hf.upload_file( |
| path_or_fileobj=local_path, |
| path_in_repo=repo_path, |
| repo_id=DATASET_REPO, |
| repo_type="dataset" |
| ) |
|
|
| await status.edit_text(f"Uploaded successfully.\n\n{filename}") |
|
|
| except Exception as e: |
| await message.reply_text(f"Upload failed:\n{e}") |
|
|
| finally: |
| cleanup(temp_dir) |
|
|
| |
| |
| |
|
|
| if __name__ == "__main__": |
| print("Starting Flask health server...") |
|
|
| threading.Thread(target=run_web, daemon=True).start() |
|
|
| print("Starting Keep-Alive Pinger...") |
|
|
| threading.Thread(target=keep_alive, daemon=True).start() |
|
|
| print("Starting Telegram bot...") |
|
|
| app.run() |
