Spaces:
Running on Zero
Running on Zero
| import asyncio | |
| import os | |
| import sys | |
| import psutil | |
| import shutil | |
| import threading | |
| import gradio as gr | |
| from bot import Bot | |
| from config import * | |
| from pyrogram import compose | |
| from pyrogram.errors import FloodWait | |
| try: | |
| import spaces | |
| except ImportError: | |
| spaces = None | |
| if spaces is not None: | |
| def dummy_gpu_loader(): | |
| return "ZeroGPU loaded" | |
| # Import the keep_alive_loop | |
| try: | |
| from plugins.anti_sleep import keep_alive_loop | |
| except ImportError: | |
| keep_alive_loop = None | |
| def get_system_stats(): | |
| try: | |
| cpu = psutil.cpu_percent() | |
| ram = psutil.virtual_memory().percent | |
| total, used, free = shutil.disk_usage("/") | |
| disk = round((used / total) * 100, 1) | |
| status_text = ( | |
| f"⚡ System Status: ACTIVE / ONLINE 🟢\n" | |
| f"🖥️ CPU Usage: {cpu}%\n" | |
| f"💾 RAM Usage: {ram}%\n" | |
| f"📁 Disk Space: {disk}%\n" | |
| f"🌐 Node/Space: {os.environ.get('SPACE_ID', 'Local Server')}" | |
| ) | |
| return status_text | |
| except Exception as e: | |
| return f"Status: Error retrieving system metrics ({str(e)})" | |
| async def start_bot(): | |
| bot = Bot( | |
| SESSION, | |
| WORKERS, | |
| DB_CHANNEL, | |
| FSUBS, | |
| TOKEN, | |
| ADMINS, | |
| MESSAGES, | |
| AUTO_DEL, | |
| DB_URI, | |
| DB_NAME, | |
| API_ID, | |
| API_HASH, | |
| PROTECT, | |
| DISABLE_BTN | |
| ) | |
| # Handle FloodWait on startup | |
| while True: | |
| try: | |
| await bot.start() | |
| break | |
| except FloodWait as e: | |
| wait = e.value if hasattr(e, 'value') else e.x | |
| print(f"FloodWait on startup! Waiting {wait} seconds before retrying...") | |
| await asyncio.sleep(wait + 10) | |
| except Exception as e: | |
| print(f"Startup error: {e}") | |
| return | |
| # Keep background thread event loop active without registering signal handlers | |
| while True: | |
| await asyncio.sleep(3600) | |
| def run_bot_in_thread(): | |
| loop = asyncio.new_event_loop() | |
| asyncio.set_event_loop(loop) | |
| # Start Keep Alive loop if found | |
| if keep_alive_loop: | |
| loop.create_task(keep_alive_loop()) | |
| loop.run_until_complete(start_bot()) | |
| # Start the Telegram Bot in a separate daemon thread | |
| threading.Thread(target=run_bot_in_thread, daemon=True).start() | |
| # Build Gradient theme Gradio Blocks UI | |
| theme = gr.themes.Default( | |
| primary_hue="indigo", | |
| secondary_hue="violet", | |
| neutral_hue="slate" | |
| ) | |
| with gr.Blocks(theme=theme, title="Yugen File Store Bot Dashboard") as demo: | |
| gr.Markdown( | |
| """ | |
| # 📂 File Store Bot - Active Dashboard | |
| This Hugging Face Space runs the File Store Bot 24/7 under the Gradio Container. | |
| --- | |
| ### 🟢 Live Metrics & Status | |
| """ | |
| ) | |
| status_display = gr.Textbox( | |
| label="Environment Details & Health Metric", | |
| value=get_system_stats(), | |
| interactive=False, | |
| lines=5 | |
| ) | |
| # Refresh metrics button | |
| refresh_btn = gr.Button("🔄 Refresh Stats", variant="primary") | |
| refresh_btn.click(fn=get_system_stats, outputs=status_display) | |
| gr.Markdown( | |
| """ | |
| --- | |
| ### 🔗 Resources & Navigation | |
| - 📢 **Channel:** [Anime Pirates Tamil](https://t.me/AnimePiratesTamil) | |
| - 🛠️ **Developer:** [@ProYato](https://t.me/ProYato) | |
| """ | |
| ) | |
| if __name__ == "__main__": | |
| # Launch Gradio server on HF default port 7860 | |
| demo.launch(server_name="0.0.0.0", server_port=7860, show_api=False) | |