Upload 6 files

README.md

@@ -1,6 +1,51 @@
+# Hugging Face Space Pinger Bot 🚀
+
+A Telegram bot built with **Pyrogram** and **FastAPI** to keep Hugging Face spaces alive by pinging them sequentially every 5 minutes. The database containing authorized users, admins, and space URLs is automatically created and synced to a Hugging Face dataset for persistent storage.
+
+## 🛠️ Required Secret Environment Variables
+
+When deploying this bot on Hugging Face Spaces (or any Docker platform), you **MUST** configure the following variables in the Space settings (under **Variables and Secrets**):
+
+1. **`API_ID`**: The API ID for your Telegram application (get it from [my.telegram.org](https://my.telegram.org)).
+2. **`API_HASH`**: The API Hash for your Telegram application (get it from [my.telegram.org](https://my.telegram.org)).
+3. **`BOT_TOKEN`**: The Telegram Bot Token generated by [@BotFather](https://t.me/BotFather).
+4. **`ADMIN_ID`**: The numeric Telegram ID of the **Super Admin** (e.g., `123456789`). The Super Admin has highest authority (e.g., can remove admins).
+5. **`HF_TOKEN`**: A Hugging Face User Access Token with **WRITE** permissions (generate it under your Hugging Face account settings at [huggingface.co/settings/tokens](https://huggingface.co/settings/tokens)).
+6. **`HF_DATASET_URL`**: The full URL of your Hugging Face dataset repository (e.g. `https://huggingface.co/datasets/your_username/your_dataset_name` or simply the repository ID like `your_username/your_dataset_name`). The bot will automatically create this dataset if it doesn't already exist and maintain `database.json` within it.
+
 ---
-license: apache-2.0
-title: C1
-sdk: docker
-emoji: 🏆
----
+
+## 🤖 Available Commands
+
+### 💬 Public Commands (Anyone)
+- `/id` - View your Telegram User ID (used to request authorization).
+- `/start` - Displays a simple welcome message.
+- `/help` - Explains what the bot does.
+- `/commands` - List commands matching the caller's authorization level.
+
+### 👤 Authorized User Commands
+- `/add <hf_url>` - Register a Hugging Face Space URL. (You can also just send a raw Hugging Face URL directly, and the bot will normalize and register it automatically).
+- `/remove <hf_url>` - Remove a Space URL you registered.
+- `/list_url` - List the Space URLs that *you* registered.
+
+### 👮 Admin Commands
+- `/add_user <user_id>` - Authorize a user to register and manage their own URLs.
+- `/remove_user <user_id>` - Deauthorize a user.
+- `/add_admin <user_id>` - Promote a user to Admin.
+- `/list_url` - List **ALL** registered Space URLs in the system along with who registered them.
+- `/list` - List all Admins and Users (excluding the Super Admin).
+
+### 👑 Super Admin Commands
+- `/remove_admin <user_id>` - Demote an Admin back to a regular user.
+- `/list` - List the Super Admin, all Admins, and all Users.
+
+---
+
+## 🐳 Deployment to Hugging Face Spaces
+
+1. Create a new Space on Hugging Face: [huggingface.co/new-space](https://huggingface.co/new-space).
+2. Choose **Docker** as the SDK.
+3. Select **Blank** (or any Docker template).
+4. Go to the Space **Settings** page and add the **Secrets** listed above.
+5. Push the repository files (`Dockerfile`, `requirements.txt`, `config.py`, `database.py`, `main.py`) to the Space.
+6. The space will automatically build the Docker image, run the FastAPI web server on port `7860`, launch the Telegram bot, and begin the sequential 5-minute pinger loop.

database.py

@@ -83,7 +83,11 @@ def init_db():
                     
         print("Successfully synced database from Hugging Face.")
     except Exception as e:
-        print(f"Database not found on HF or error loading: {e}. Initializing a new one.")
+        err_msg = str(e)
+        if "404" in err_msg or "Entry Not Found" in err_msg:
+            print("Database file not found in HF dataset. Initializing a new one.")
+        else:
+            print(f"Error loading database from HF: {err_msg}. Initializing a new one.")
         if os.path.exists(LOCAL_PATH):
             try:
                 with open(LOCAL_PATH, "r", encoding="utf-8") as f:

main.py

@@ -2,7 +2,6 @@ import asyncio
 import os
 import httpx
 from fastapi import FastAPI
-from fastapi.responses import HTMLResponse
 from contextlib import asynccontextmanager
 from pyrogram import Client, filters
 import config
@@ -126,238 +125,10 @@ async def lifespan(app: FastAPI):
 
 app = FastAPI(lifespan=lifespan)
 
-# Web Dashboard
-@app.get("/", response_class=HTMLResponse)
-async def dashboard():
-    urls = database.get_all_urls()
-    own_url = get_own_space_url() or "Not detected (running locally?)"
-    
-    urls_list_html = "".join([f"<li><code>{url}</code></li>" for url in urls])
-    if not urls:
-        urls_list_html = "<li>No external spaces added yet.</li>"
-        
-    config_status = {
-        "Telegram Bot Token": config.BOT_TOKEN is not None,
-        "Super Admin ID": config.SUPER_ADMIN_ID is not None,
-        "HF Access Token": config.HF_TOKEN is not None,
-        "HF Dataset URL": config.HF_DATASET_URL is not None,
-    }
-    
-    config_rows = ""
-    for name, ok in config_status.items():
-        status_badge = '<span class="badge ok">Configured</span>' if ok else '<span class="badge missing">Missing</span>'
-        config_rows += f"<tr><td>{name}</td><td>{status_badge}</td></tr>"
-        
-    html_content = f"""
-    <!DOCTYPE html>
-    <html lang="en">
-    <head>
-        <meta charset="UTF-8">
-        <meta name="viewport" content="width=device-width, initial-scale=1.0">
-        <title>Hugging Face Space Pinger Dashboard</title>
-        <link rel="preconnect" href="https://fonts.googleapis.com">
-        <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
-        <link href="https://fonts.googleapis.com/css2?family=Outfit:wght@300;400;600;800&family=Space+Grotesk:wght@400;600&display=swap" rel="stylesheet">
-        <style>
-            :root {{
-                --bg: #0b0f19;
-                --card-bg: rgba(255, 255, 255, 0.03);
-                --card-border: rgba(255, 255, 255, 0.08);
-                --primary: #4f46e5;
-                --primary-glow: rgba(79, 70, 229, 0.4);
-                --success: #10b981;
-                --success-glow: rgba(16, 185, 129, 0.4);
-                --text-main: #f3f4f6;
-                --text-muted: #9ca3af;
-            }}
-            * {{
-                box-sizing: border-box;
-                margin: 0;
-                padding: 0;
-            }}
-            body {{
-                font-family: 'Outfit', sans-serif;
-                background-color: var(--bg);
-                color: var(--text-main);
-                min-height: 100vh;
-                display: flex;
-                flex-direction: column;
-                justify-content: center;
-                align-items: center;
-                padding: 2rem;
-                background-image: 
-                    radial-gradient(circle at 10% 20%, rgba(79, 70, 229, 0.08) 0%, transparent 40%),
-                    radial-gradient(circle at 90% 80%, rgba(16, 185, 129, 0.06) 0%, transparent 40%);
-            }}
-            .container {{
-                max-width: 800px;
-                width: 100%;
-                display: flex;
-                flex-direction: column;
-                gap: 2rem;
-            }}
-            header {{
-                text-align: center;
-                margin-bottom: 1rem;
-            }}
-            h1 {{
-                font-family: 'Space Grotesk', sans-serif;
-                font-weight: 800;
-                font-size: 2.5rem;
-                background: linear-gradient(135deg, #a5b4fc, #818cf8, #6366f1);
-                -webkit-background-clip: text;
-                -webkit-text-fill-color: transparent;
-                margin-bottom: 0.5rem;
-            }}
-            .subtitle {{
-                color: var(--text-muted);
-                font-size: 1.1rem;
-            }}
-            .grid {{
-                display: grid;
-                grid-template-columns: 1fr 1fr;
-                gap: 1.5rem;
-            }}
-            @media (max-width: 600px) {{
-                .grid {{
-                    grid-template-columns: 1fr;
-                }}
-            }}
-            .card {{
-                background: var(--card-bg);
-                border: 1px solid var(--card-border);
-                border-radius: 16px;
-                padding: 1.5rem;
-                backdrop-filter: blur(12px);
-                box-shadow: 0 8px 32px 0 rgba(0, 0, 0, 0.37);
-                transition: transform 0.3s ease, border-color 0.3s ease;
-            }}
-            .card:hover {{
-                transform: translateY(-5px);
-                border-color: rgba(99, 102, 241, 0.2);
-            }}
-            .card-title {{
-                font-family: 'Space Grotesk', sans-serif;
-                font-size: 1.25rem;
-                font-weight: 600;
-                margin-bottom: 1rem;
-                display: flex;
-                align-items: center;
-                gap: 0.5rem;
-                color: #a5b4fc;
-            }}
-            .status-badge {{
-                display: inline-flex;
-                align-items: center;
-                gap: 0.5rem;
-                padding: 0.5rem 1rem;
-                background: rgba(16, 185, 129, 0.1);
-                border: 1px solid var(--success);
-                color: var(--success);
-                border-radius: 9999px;
-                font-weight: 600;
-                box-shadow: 0 0 15px var(--success-glow);
-                animation: pulse 2s infinite;
-            }}
-            @keyframes pulse {{
-                0% {{ box-shadow: 0 0 5px var(--success-glow); }}
-                50% {{ box-shadow: 0 0 20px var(--success-glow); }}
-                100% {{ box-shadow: 0 0 5px var(--success-glow); }}
-            }}
-            ul {{
-                list-style: none;
-                display: flex;
-                flex-direction: column;
-                gap: 0.75rem;
-            }}
-            li {{
-                background: rgba(255, 255, 255, 0.02);
-                border: 1px solid rgba(255, 255, 255, 0.05);
-                border-radius: 8px;
-                padding: 0.75rem;
-                font-size: 0.9rem;
-                overflow-x: auto;
-            }}
-            code {{
-                font-family: 'Courier New', Courier, monospace;
-                color: #818cf8;
-            }}
-            table {{
-                width: 100%;
-                border-collapse: collapse;
-            }}
-            td {{
-                padding: 0.75rem 0;
-                border-bottom: 1px solid rgba(255, 255, 255, 0.05);
-            }}
-            tr:last-child td {{
-                border-bottom: none;
-            }}
-            .badge {{
-                display: inline-block;
-                padding: 0.25rem 0.5rem;
-                border-radius: 4px;
-                font-size: 0.8rem;
-                font-weight: 600;
-            }}
-            .badge.ok {{
-                background: rgba(16, 185, 129, 0.15);
-                color: var(--success);
-            }}
-            .badge.missing {{
-                background: rgba(239, 68, 68, 0.15);
-                color: #ef4444;
-            }}
-            footer {{
-                text-align: center;
-                color: var(--text-muted);
-                font-size: 0.85rem;
-                margin-top: 2rem;
-            }}
-        </style>
-    </head>
-    <body>
-        <div class="container">
-            <header>
-                <h1>HF Space Pinger</h1>
-                <p class="subtitle">Reliable Hugging Face Space keep-alive bot</p>
-            </header>
-            
-            <div class="card" style="text-align: center; display: flex; flex-direction: column; align-items: center; gap: 1rem;">
-                <div class="card-title">SYSTEM STATUS</div>
-                <div>
-                    <span class="status-badge">
-                        <span style="display:inline-block; width:8px; height:8px; background:var(--success); border-radius:50%;"></span>
-                        BOT IS ACTIVE
-                    </span>
-                </div>
-                <p style="color: var(--text-muted); margin-top: 0.5rem;">Own Space URL: <code>{own_url}</code></p>
-            </div>
-            
-            <div class="grid">
-                <div class="card">
-                    <div class="card-title">📋 Config Settings</div>
-                    <table>
-                        {config_rows}
-                    </table>
-                </div>
-                
-                <div class="card">
-                    <div class="card-title">🔗 Monitored Spaces ({len(urls)})</div>
-                    <ul>
-                        {urls_list_html}
-                    </ul>
-                </div>
-            </div>
-            
-            <footer>
-                Powered by FastAPI, Pyrogram & Hugging Face • Kept awake automatically
-            </footer>
-        </div>
-    </body>
-    </html>
-    """
-    return html_content
+# Health Check Endpoint
+@app.get("/")
+async def health_check():
+    return {"status": "ok", "message": "HF Space Pinger Bot is running successfully."}
 
 # ================== PYROGRAM BOT HANDLERS ==================
 
@@ -366,6 +137,17 @@ async def start_command(client, message):
     if not message.from_user:
         return
     log_and_update(message)
+    
+    user_id = message.from_user.id
+    if not database.is_authorized(user_id):
+        await message.reply_text(
+            "❌ **Unauthorized User**\n\n"
+            f"You are unauthorized to use the bot.\n"
+            f"Your ID: `{user_id}`\n\n"
+            "Tell an admin to add you."
+        )
+        return
+
     # Welcomes the user
     await message.reply_text(
         "👋 **Hello! Welcome to the Hugging Face Space Pinger Bot.** 🚀\n\n"