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.

main.py

@@ -141,10 +141,10 @@ async def start_command(client, 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."
+            "👋 **Hello! Welcome to the Hugging Face Space Pinger Bot.** 🚀\n\n"
+            "You are currently unauthorized to use this bot.\n"
+            f"🔑 Your ID: `{user_id}`\n\n"
+            "Please contact an admin to allow you access."
         )
         return
 
@@ -232,20 +232,25 @@ async def add_url_command(client, message):
         return
         
     if len(message.command) < 2:
-        await message.reply_text("Usage: `/add <hf_space_url>`\n\nExample:\n`/add https://huggingface.co/spaces/iozxv/C1`")
+        await message.reply_text("Usage: `/add <hf_space_url1> <hf_space_url2> ...`\n\nExample:\n`/add https://huggingface.co/spaces/iozxv/C1`")
         return
         
-    url_input = message.command[1]
-    try:
-        success, normalized = database.add_url(url_input, user_id)
-        if success:
-            await message.reply_text(f"✅ **Success!** URL normalized and added to ping list:\n`{normalized}`")
-        else:
-            await message.reply_text(f"⚠️ This space is already in the ping list:\n`{normalized}`")
-    except ValueError as e:
-        await message.reply_text(f"❌ **Invalid URL:** {str(e)}")
-    except Exception as e:
-        await message.reply_text(f"❌ **Error:** {str(e)}")
+    urls_input = message.command[1:]
+    results = []
+    
+    for url_input in urls_input:
+        try:
+            success, normalized = database.add_url(url_input, user_id)
+            if success:
+                results.append(f"✅ Added: `{normalized}`")
+            else:
+                results.append(f"⚠️ Already exists: `{normalized}`")
+        except ValueError as e:
+            results.append(f"❌ Invalid: {url_input} - {str(e)}")
+        except Exception as e:
+            results.append(f"❌ Error: {url_input} - {str(e)}")
+            
+    await message.reply_text("\n".join(results))
 
 @bot.on_message(filters.command("remove"))
 async def remove_url_command(client, message):
@@ -294,9 +299,21 @@ async def list_url_command(client, message):
     # If admin or super admin, show who added it. Otherwise, just list URLs
     if database.is_admin(user_id):
         response = f"🔗 **All Monitored Space URLs ({len(urls_data)}):**\n\n"
+        
+        # Group by user
+        grouped = {}
         for item in urls_data:
-            added_by_display = database.get_user_display(item["added_by"])
-            response += f"- `{item['url']}`\n  Added by: {added_by_display}\n\n"
+            added_by = item["added_by"]
+            if added_by not in grouped:
+                grouped[added_by] = []
+            grouped[added_by].append(item["url"])
+            
+        for user, urls in grouped.items():
+            user_display = database.get_user_display(user)
+            response += f"👤 **Added by: {user_display}**\n"
+            for u in urls:
+                response += f"- `{u}`\n"
+            response += "\n"
     else:
         response = f"🔗 **Your Monitored Space URLs ({len(urls_data)}):**\n\n"
         for item in urls_data:
@@ -482,16 +499,25 @@ async def handle_text_message(client, message):
             await message.reply_text("❌ You are not authorized to add URLs. Ask an Admin to authorize your ID.")
             return
             
-        try:
-            success, normalized = database.add_url(text, user_id)
-            if success:
-                await message.reply_text(f"✅ **URL Detected & Added to Ping List!**\nNormalized: `{normalized}`\nInterval: 5 minutes")
-            else:
-                await message.reply_text(f"⚠️ This space is already in the ping list:\n`{normalized}`")
-        except ValueError as e:
-            await message.reply_text(f"❌ **Invalid URL:** {str(e)}")
-        except Exception as e:
-            await message.reply_text(f"❌ **Error:** {str(e)}")
+        # Split text into possible multiple URLs
+        words = text.split()
+        potential_urls = [w for w in words if "huggingface.co/spaces" in w or "hf.space" in w]
+        
+        results = []
+        for url_input in potential_urls:
+            try:
+                success, normalized = database.add_url(url_input, user_id)
+                if success:
+                    results.append(f"✅ Added: `{normalized}`")
+                else:
+                    results.append(f"⚠️ Already exists: `{normalized}`")
+            except ValueError as e:
+                results.append(f"❌ Invalid: {url_input} - {str(e)}")
+            except Exception as e:
+                results.append(f"❌ Error: {url_input} - {str(e)}")
+                
+        if results:
+            await message.reply_text("\n".join(results))
 
 # ==================== MAIN EXECUTION ====================