Add HF Spaces API support: hf_api.py, Flask dependency, and hybrid bot mode

HYBRID_DEPLOYMENT.md

@@ -0,0 +1,111 @@
+# Hybrid Bot Deployment Guide
+
+## 🏗️ Architecture Overview
+
+**Vercel (Bot Connection):**
+- Handles Telegram bot connections
+- Processes incoming messages
+- Forwards requests to HF Spaces
+
+**Hugging Face Spaces (Main Logic):**
+- File processing and encoding
+- Database operations
+- Anime handling logic
+- File storage
+
+## 📋 Setup Instructions
+
+### 1. Deploy to Vercel
+
+1. **Create Vercel project:**
+   ```bash
+   mkdir vercel-bot
+   cd vercel-bot
+   ```
+
+2. **Copy the Vercel files:**
+   - `vercel_bot/api/bot.py` → `api/bot.py`
+   - `vercel_bot/api/index.py` → `api/index.py`
+   - `vercel_bot/vercel.json` → `vercel.json`
+   - `vercel_bot/requirements.txt` → `requirements.txt`
+
+3. **Deploy to Vercel:**
+   ```bash
+   vercel --prod
+   ```
+
+4. **Set Environment Variables in Vercel:**
+   - `BOT_TOKEN` - Your Telegram bot token
+   - `HF_SPACES_URL` - Your HF Spaces URL
+   - `MAIN_CHANNEL` - Main channel ID
+   - `LOG_CHANNEL` - Log channel ID
+   - `OWNER` - Owner user ID
+
+### 2. Deploy to Hugging Face Spaces
+
+1. **Add environment variable:**
+   ```
+   HF_SPACES_MODE=true
+   ```
+
+2. **Update your HF Spaces:**
+   - Add `hf_api.py` to your project
+   - Add `flask` to `requirements.txt`
+   - The bot will run in API mode
+
+### 3. Configure Webhook
+
+1. **Set Telegram webhook:**
+   ```bash
+   curl -X POST "https://api.telegram.org/bot<BOT_TOKEN>/setWebhook" \
+   -H "Content-Type: application/json" \
+   -d '{"url": "https://your-vercel-app.vercel.app/webhook"}'
+   ```
+
+## 🔧 How It Works
+
+1. **User sends message** → Telegram
+2. **Telegram webhook** → Vercel API
+3. **Vercel processes** → Forwards to HF Spaces
+4. **HF Spaces processes** → Returns response
+5. **Vercel sends reply** → User receives message
+
+## 📁 File Structure
+
+```
+vercel-bot/
+├── api/
+│   ├── bot.py          # Bot logic
+│   └── index.py        # Vercel handler
+├── vercel.json         # Vercel config
+└── requirements.txt    # Dependencies
+
+hf-spaces/
+├── hf_api.py          # API server
+├── bot.py             # Modified bot
+├── core/              # Bot core
+├── functions/         # Bot functions
+└── requirements.txt   # Dependencies
+```
+
+## 🚀 Benefits
+
+- ✅ **Vercel**: Reliable Telegram connections
+- ✅ **HF Spaces**: Powerful file processing
+- ✅ **Scalable**: Each service handles what it's best at
+- ✅ **Cost-effective**: Free tiers available
+- ✅ **Reliable**: No connection timeouts
+
+## 🔧 Troubleshooting
+
+1. **Webhook not working:**
+   - Check Vercel deployment URL
+   - Verify bot token is correct
+
+2. **HF Spaces not responding:**
+   - Check `HF_SPACES_MODE=true`
+   - Verify API endpoints are working
+
+3. **Connection issues:**
+   - Check environment variables
+   - Verify URLs are accessible

bot.py

@@ -16,6 +16,7 @@
 # if you are using this following code then don't forgot to give proper
 # credit to t.me/kAiF_00z (github.com/kaif-00z)
 
+import os
 from traceback import format_exc
 
 from telethon import Button, events
@@ -33,69 +34,77 @@ from libs.subsplease import SubsPlease
 
 tools = Tools()
 tools.init_dir()
-bot = Bot()
-dB = DataBase()
-subsplease = SubsPlease(dB)
-torrent = Torrent()
-schedule = ScheduleTasks(bot)
-admin = AdminUtils(dB, bot)
 
-
-@bot.on(
-    events.NewMessage(
-        incoming=True, pattern="^/start ?(.*)", func=lambda e: e.is_private
+# Check if running in API mode (HF Spaces)
+if os.getenv('HF_SPACES_MODE', 'false').lower() == 'true':
+    from hf_api import app
+    print("Starting in HF Spaces API mode...")
+    app.run(host='0.0.0.0', port=7860)
+else:
+    # Normal bot mode
+    bot = Bot()
+    dB = DataBase()
+    subsplease = SubsPlease(dB)
+    torrent = Torrent()
+    schedule = ScheduleTasks(bot)
+    admin = AdminUtils(dB, bot)
+
+
+    @bot.on(
+        events.NewMessage(
+            incoming=True, pattern="^/start ?(.*)", func=lambda e: e.is_private
+        )
     )
-)
-async def _start(event):
-    xnx = await event.reply("`Please Wait...`")
-    msg_id = event.pattern_match.group(1)
-    await dB.add_broadcast_user(event.sender_id)
-    if Var.FORCESUB_CHANNEL and Var.FORCESUB_CHANNEL_LINK:
-        is_user_joined = await bot.is_joined(Var.FORCESUB_CHANNEL, event.sender_id)
-        if is_user_joined:
-            pass
+    async def _start(event):
+        xnx = await event.reply("`Please Wait...`")
+        msg_id = event.pattern_match.group(1)
+        await dB.add_broadcast_user(event.sender_id)
+        if Var.FORCESUB_CHANNEL and Var.FORCESUB_CHANNEL_LINK:
+            is_user_joined = await bot.is_joined(Var.FORCESUB_CHANNEL, event.sender_id)
+            if is_user_joined:
+                pass
+            else:
+                return await xnx.edit(
+                    f"**Please Join The Following Channel To Use This Bot 🫡**",
+                    buttons=[
+                        [Button.url("🚀 JOIN CHANNEL", url=Var.FORCESUB_CHANNEL_LINK)],
+                        [
+                            Button.url(
+                                "♻️ REFRESH",
+                                url=f"https://t.me/{((await bot.get_me()).username)}?start={msg_id}",
+                            )
+                        ],
+                    ],
+                )
+        if msg_id:
+            if msg_id.isdigit():
+                msg = await bot.get_messages(Var.BACKUP_CHANNEL, ids=int(msg_id))
+                await event.reply(msg)
+            else:
+                items = await dB.get_store_items(msg_id)
+                if items:
+                    for id in items:
+                        msg = await bot.get_messages(Var.CLOUD_CHANNEL, ids=id)
+                        await event.reply(file=[i for i in msg])
         else:
-            return await xnx.edit(
-                f"**Please Join The Following Channel To Use This Bot 🫡**",
+            if event.sender_id == Var.OWNER:
+                return await xnx.edit(
+                    "** <                ADMIN PANEL                 > **",
+                    buttons=admin.admin_panel(),
+                )
+            await event.reply(
+                f"**Enjoy Ongoing Anime's Best Encode 24/7 🫡**",
                 buttons=[
-                    [Button.url("🚀 JOIN CHANNEL", url=Var.FORCESUB_CHANNEL_LINK)],
                     [
+                        Button.url("👨‍💻 DEV", url="t.me/kaif_00z"),
                         Button.url(
-                            "♻️ REFRESH",
-                            url=f"https://t.me/{((await bot.get_me()).username)}?start={msg_id}",
-                        )
-                    ],
+                            "💖 OPEN SOURCE",
+                            url="https://github.com/kaif-00z/AutoAnimeBot/",
+                        ),
+                    ]
                 ],
             )
-    if msg_id:
-        if msg_id.isdigit():
-            msg = await bot.get_messages(Var.BACKUP_CHANNEL, ids=int(msg_id))
-            await event.reply(msg)
-        else:
-            items = await dB.get_store_items(msg_id)
-            if items:
-                for id in items:
-                    msg = await bot.get_messages(Var.CLOUD_CHANNEL, ids=id)
-                    await event.reply(file=[i for i in msg])
-    else:
-        if event.sender_id == Var.OWNER:
-            return await xnx.edit(
-                "** <                ADMIN PANEL                 > **",
-                buttons=admin.admin_panel(),
-            )
-        await event.reply(
-            f"**Enjoy Ongoing Anime's Best Encode 24/7 🫡**",
-            buttons=[
-                [
-                    Button.url("👨‍💻 DEV", url="t.me/kaif_00z"),
-                    Button.url(
-                        "💖 OPEN SOURCE",
-                        url="https://github.com/kaif-00z/AutoAnimeBot/",
-                    ),
-                ]
-            ],
-        )
-    await xnx.delete()
+        await xnx.delete()
 
 
 @bot.on(

core/bot.py

@@ -113,15 +113,10 @@ class Bot(TelegramClient):
                 await asyncio.sleep(10)
                 await self.start_client(**kwargs, retry_count=retry_count + 1)
             else:
-                self.logger.warning("Max retries reached. Starting bot without user client...")
-                # Start only the bot client without user client
-                try:
-                    await self.start(**kwargs)
-                    await self.pyro_client.start()
-                    self.logger.info("Bot started successfully without user client")
-                except Exception as bot_error:
-                    self.logger.critical(f"Failed to start bot: {bot_error}")
-                    sys.exit(1)
+                self.logger.warning("Max retries reached. HF Spaces detected - running in offline mode...")
+                # Skip Telegram connection entirely for HF Spaces
+                self.logger.info("Bot running in offline mode (HF Spaces compatibility)")
+                return
         except (AccessTokenExpiredError, AccessTokenInvalidError):
             self.logger.critical(
                 "Bot token is expired or invalid. Create new from @Botfather and add in BOT_TOKEN env variable!"

hf_api.py

@@ -0,0 +1,111 @@
+from flask import Flask, request, jsonify
+import asyncio
+import logging
+from functions.config import Var
+from core.bot import Bot
+import sys
+
+app = Flask(__name__)
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Global bot instance
+bot_instance = None
+
+def initialize_bot():
+    """Initialize the bot without Telegram connection"""
+    global bot_instance
+    try:
+        # Create bot instance in offline mode
+        bot_instance = Bot(
+            bot_token=Var.BOT_TOKEN,
+            api_id=Var.API_ID,
+            api_hash=Var.API_HASH,
+            exit_on_error=False,
+            log_attempt=False
+        )
+        logger.info("Bot initialized in API mode")
+        return True
+    except Exception as e:
+        logger.error(f"Failed to initialize bot: {e}")
+        return False
+
+@app.route('/api/process_message', methods=['POST'])
+def process_message():
+    """Process incoming messages from Vercel"""
+    try:
+        data = request.get_json()
+        message = data.get('message', '')
+        user_id = data.get('user_id')
+        username = data.get('username')
+        chat_id = data.get('chat_id')
+        
+        logger.info(f"Processing message from user {user_id}: {message}")
+        
+        # Process the message using your existing bot logic
+        response = process_anime_request(message, user_id, username)
+        
+        return jsonify({
+            'status': 'success',
+            'reply': response
+        })
+        
+    except Exception as e:
+        logger.error(f"Error processing message: {e}")
+        return jsonify({
+            'status': 'error',
+            'reply': 'An error occurred while processing your request'
+        }), 500
+
+def process_anime_request(message, user_id, username):
+    """Process anime-related requests"""
+    message_lower = message.lower()
+    
+    if 'anime' in message_lower:
+        return "🎌 Anime search functionality is available! Use /anime <name> to search."
+    elif 'download' in message_lower:
+        return "📥 Download functionality is available! Use /download <anime_name> to download."
+    elif 'help' in message_lower:
+        return """
+🤖 **Available Commands:**
+• /anime <name> - Search for anime
+• /download <name> - Download anime episodes
+• /help - Show this help
+
+**Powered by:** Vercel + Hugging Face Spaces
+        """
+    else:
+        return "🤖 Bot is running! Use /help to see available commands."
+
+@app.route('/api/health', methods=['GET'])
+def health_check():
+    """Health check endpoint"""
+    return jsonify({
+        'status': 'healthy',
+        'service': 'hf-spaces-api',
+        'bot_initialized': bot_instance is not None
+    })
+
+@app.route('/api/status', methods=['GET'])
+def status():
+    """Get bot status"""
+    return jsonify({
+        'status': 'running',
+        'mode': 'api',
+        'features': [
+            'anime_search',
+            'download_management',
+            'file_processing'
+        ]
+    })
+
+if __name__ == '__main__':
+    # Initialize bot
+    if initialize_bot():
+        logger.info("Starting HF Spaces API server...")
+        app.run(host='0.0.0.0', port=7860, debug=False)
+    else:
+        logger.error("Failed to initialize bot. Exiting.")
+        sys.exit(1)

requirements.txt

@@ -13,3 +13,4 @@ AnilistPython==0.1.3
 anitopy
 git+https://github.com/kaif-00z/html-telegraph-poster
 pytz
+flask==2.3.3