Update bot.py

bot.py

@@ -2,6 +2,8 @@ import requests
 from telegram import Update, ReplyKeyboardMarkup, ReplyKeyboardRemove, InputFile
 from telegram.ext import ApplicationBuilder, CommandHandler, MessageHandler, filters, ContextTypes, ConversationHandler
 import os
+import time
+import threading
 
 # Replace with your actual Hugging Face Space backend URL
 BACKEND_URL = "https://dragxd-host.hf.space"
@@ -63,6 +65,40 @@ async def upload_env(update: Update, context: ContextTypes.DEFAULT_TYPE):
         await update.message.reply_text("Please upload your .env file as a document.")
     return ConversationHandler.END
 
+async def deploy(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    user_id = str(update.effective_user.id)
+    # Trigger deployment
+    resp = requests.post(f"{BACKEND_URL}/deploy", json={"user_id": user_id, "github_url": ""})
+    if resp.status_code != 200 or "deploy_id" not in resp.json():
+        await update.message.reply_text("Failed to start deployment. Make sure you have selected a repo and uploaded your .env file.")
+        return
+    deploy_id = resp.json()["deploy_id"]
+    await update.message.reply_text(f"Deployment started! ID: {deploy_id}\nStreaming logs:")
+
+    def poll_logs():
+        last_line = 0
+        finished = False
+        while not finished:
+            try:
+                logs_resp = requests.get(f"{BACKEND_URL}/logs/{deploy_id}", params={"last_line": last_line})
+                if logs_resp.status_code == 200:
+                    data = logs_resp.json()
+                    logs = data.get("logs", [])
+                    status = data.get("status", "unknown")
+                    if logs:
+                        for log in logs:
+                            context.application.create_task(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}"))
+                        break
+                time.sleep(2)
+            except Exception as e:
+                context.application.create_task(update.message.reply_text(f"Error streaming logs: {e}"))
+                break
+    threading.Thread(target=poll_logs, daemon=True).start()
+
 def main():
     TOKEN = os.environ.get("TELEGRAM_BOT_TOKEN")  # Set your bot token as env variable
     app = ApplicationBuilder().token(TOKEN).build()
@@ -76,6 +112,7 @@ def main():
     )
     app.add_handler(CommandHandler("start", start))
     app.add_handler(CommandHandler("connect", connect))
+    app.add_handler(CommandHandler("deploy", deploy))
     app.add_handler(conv_handler)
     app.run_polling()