Update bot.py

bot.py

@@ -4,6 +4,7 @@ from telegram.ext import ApplicationBuilder, CommandHandler, MessageHandler, fil
 import os
 import time
 import threading
+import asyncio
 
 # Replace with your actual Hugging Face Space backend URL
 BACKEND_URL = "https://dragxd-host.hf.space"
@@ -75,7 +76,7 @@ async def deploy(update: Update, context: ContextTypes.DEFAULT_TYPE):
     deploy_id = resp.json()["deploy_id"]
     await update.message.reply_text(f"Deployment started! ID: {deploy_id}\nStreaming logs:")
 
-    def poll_logs():
+    async def poll_logs():
         last_line = 0
         finished = False
         while not finished:
@@ -87,17 +88,17 @@ async def deploy(update: Update, context: ContextTypes.DEFAULT_TYPE):
                     status = data.get("status", "unknown")
                     if logs:
                         for log in logs:
-                            context.application.create_task(update.message.reply_text(log))
+                            await update.message.reply_text(log)
                     last_line = data.get("next_line", last_line)
                     if status != "deploying":
                         finished = True
-                        context.application.create_task(update.message.reply_text(f"Deployment finished with status: {status}"))
+                        await update.message.reply_text(f"Deployment finished with status: {status}")
                         break
-                time.sleep(2)
+                await asyncio.sleep(2)
             except Exception as e:
-                context.application.create_task(update.message.reply_text(f"Error streaming logs: {e}"))
+                await update.message.reply_text(f"Error streaming logs: {e}")
                 break
-    threading.Thread(target=poll_logs, daemon=True).start()
+    context.application.create_task(poll_logs())
 
 def main():
     TOKEN = os.environ.get("TELEGRAM_BOT_TOKEN")  # Set your bot token as env variable