Add application file

Dockerfile

@@ -30,4 +30,4 @@ COPY . .
 EXPOSE 7860
 
 # Command to run the application
-CMD ["python", "telegram_bot1.py"] 
+CMD ["python", "api.py"] 

api.py

@@ -0,0 +1,101 @@
+from fastapi import FastAPI, BackgroundTasks
+from fastapi.responses import JSONResponse
+import uvicorn
+from typing import Optional
+from telegram_bot import WeBookBot
+import os
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
+
+# Create FastAPI app
+app = FastAPI(title="WeBook Bot API", description="API for controlling the WeBook Telegram Bot")
+
+# Global bot instance
+bot: Optional[WeBookBot] = None
+
+@app.on_event("startup")
+async def startup_event():
+    """Initialize bot on startup"""
+    global bot
+    token = os.getenv('TELEGRAM_BOT_TOKEN')
+    if not token:
+        raise ValueError("TELEGRAM_BOT_TOKEN not found in environment variables")
+    bot = WeBookBot(token)
+
+@app.on_event("shutdown")
+async def shutdown_event():
+    """Cleanup on shutdown"""
+    global bot
+    if bot and bot.is_running:
+        await bot.stop_bot()
+
+@app.post("/startbot")
+async def start_bot(background_tasks: BackgroundTasks):
+    """Start the Telegram bot"""
+    global bot
+    if not bot:
+        return JSONResponse(
+            status_code=500,
+            content={"status": "error", "message": "Bot not initialized"}
+        )
+
+    if bot.is_running:
+        return JSONResponse(
+            status_code=200,
+            content={"status": "success", "message": "Bot is already running"}
+        )
+
+    # Start bot in background
+    background_tasks.add_task(bot.start_bot)
+    
+    return JSONResponse(
+        status_code=200,
+        content={"status": "success", "message": "Bot starting in background"}
+    )
+
+@app.post("/stopbot")
+async def stop_bot():
+    """Stop the Telegram bot"""
+    global bot
+    if not bot:
+        return JSONResponse(
+            status_code=500,
+            content={"status": "error", "message": "Bot not initialized"}
+        )
+
+    success, message = await bot.stop_bot()
+    
+    return JSONResponse(
+        status_code=200 if success else 500,
+        content={"status": "success" if success else "error", "message": message}
+    )
+
+@app.get("/botstatus")
+async def get_bot_status():
+    """Get the current status of the bot"""
+    global bot
+    if not bot:
+        return JSONResponse(
+            status_code=500,
+            content={"status": "error", "message": "Bot not initialized"}
+        )
+
+    return JSONResponse(
+        status_code=200,
+        content={
+            "status": "success",
+            "is_running": bot.is_running,
+            "message": "Bot is running" if bot.is_running else "Bot is stopped"
+        }
+    )
+
+if __name__ == "__main__":
+    # Run FastAPI app
+    uvicorn.run(
+        "api:app",
+        host="0.0.0.0",
+        port=7860,
+        reload=True
+    ) 

requirements.txt

@@ -1,4 +1,6 @@
 python-telegram-bot>=20.0
 requests>=2.31.0
 playwright>=1.40.0
-python-dotenv>=1.0.0 
+python-dotenv>=1.0.0
+fastapi>=0.68.0
+uvicorn>=0.15.0 

telegram_bot.py

@@ -7,6 +7,7 @@ from datetime import datetime, timedelta
 from monitor import EventMonitor
 from dotenv import load_dotenv
 import os
+import asyncio
 
 # Load environment variables
 load_dotenv()
@@ -36,8 +37,9 @@ class WeBookBot:
             "User-Agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
         }
         self.monitor = EventMonitor(self)
-        # Add storage for user credentials
         self.user_credentials = {}
+        self.application = None
+        self.is_running = False
 
     def _get_events_payload(self, category=None, date_filter=None, price_min=None, price_max=None, zones=None, country_code="SA", limit=20, skip=0):
         """Generate the payload for the events API request with flexible filtering using the getShows query."""
@@ -849,27 +851,51 @@ class WeBookBot:
         response.raise_for_status()
         return response.json()
 
-    def run(self):
-        """Run the bot"""
+    async def start_bot(self):
+        """Start the bot asynchronously"""
+        if self.is_running:
+            return False, "Bot is already running"
+
         try:
-            application = Application.builder().token(self.token).build()
+            self.application = Application.builder().token(self.token).build()
             
             # Add handlers
-            application.add_handler(CommandHandler("start", self.start_command))
-            application.add_handler(CommandHandler("setinfoaccountwebook", self.set_account_info_command))
-            application.add_handler(CommandHandler("stateloginwebook", self.account_status_command))
-            application.add_handler(CommandHandler("help", self.help_command))
-            application.add_handler(CommandHandler("info", self.info_command))
-            application.add_handler(CallbackQueryHandler(self.button_callback))
-            application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, self.handle_credentials_input))
+            self.application.add_handler(CommandHandler("start", self.start_command))
+            self.application.add_handler(CommandHandler("setinfoaccountwebook", self.set_account_info_command))
+            self.application.add_handler(CommandHandler("stateloginwebook", self.account_status_command))
+            self.application.add_handler(CommandHandler("help", self.help_command))
+            self.application.add_handler(CommandHandler("info", self.info_command))
+            self.application.add_handler(CallbackQueryHandler(self.button_callback))
+            self.application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, self.handle_credentials_input))
             
             # Start the bot
             logger.info("Starting bot...")
-            application.run_polling()
+            await self.application.initialize()
+            await self.application.start()
+            await self.application.run_polling()
+            self.is_running = True
+            return True, "Bot started successfully"
             
         except Exception as e:
             logger.error(f"Error running bot: {str(e)}")
-            raise
+            self.is_running = False
+            return False, f"Error starting bot: {str(e)}"
+
+    async def stop_bot(self):
+        """Stop the bot"""
+        if not self.is_running:
+            return False, "Bot is not running"
+
+        try:
+            if self.application:
+                await self.application.stop()
+                await self.application.shutdown()
+                self.is_running = False
+                return True, "Bot stopped successfully"
+            return False, "No application instance found"
+        except Exception as e:
+            logger.error(f"Error stopping bot: {str(e)}")
+            return False, f"Error stopping bot: {str(e)}"
 
 if __name__ == "__main__":
     # Get bot token from environment variable
@@ -878,4 +904,4 @@ if __name__ == "__main__":
         raise ValueError("TELEGRAM_BOT_TOKEN not found in environment variables")
     
     bot = WeBookBot(BOT_TOKEN)
-    bot.run() 
+    asyncio.run(bot.start_bot())