Update bot.py

bot.py

@@ -1,46 +1,83 @@
 import requests
-from telegram import Update
-from telegram.ext import ApplicationBuilder, CommandHandler, ContextTypes
+from telegram import Update, ReplyKeyboardMarkup, ReplyKeyboardRemove, InputFile
+from telegram.ext import ApplicationBuilder, CommandHandler, MessageHandler, filters, ContextTypes, ConversationHandler
 import os
 
 # Replace with your actual Hugging Face Space backend URL
-BACKEND_URL = "https://dragxd-host.hf.space"
+BACKEND_URL = "https://<your-hf-space-username>-<space-name>.hf.space"
+
+SELECT_REPO, UPLOAD_ENV = range(2)
 
 async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    await update.message.reply_text("Send /deploy <github_repo_url> to deploy your project!")
+    await update.message.reply_text("Send /connect to link your GitHub account.")
+
+async def connect(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    user_id = str(update.effective_user.id)
+    resp = requests.get(f"{BACKEND_URL}/github/login", params={"user_id": user_id})
+    if resp.status_code == 200:
+        auth_url = resp.json()["auth_url"]
+        await update.message.reply_text(f"Connect your GitHub: {auth_url}\nAfter connecting, send /repos to choose a repo.")
+    else:
+        await update.message.reply_text("Failed to get GitHub auth URL.")
 
-async def deploy(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    if len(context.args) != 1:
-        await update.message.reply_text("Usage: /deploy <github_repo_url>")
-        return
-    github_url = context.args[0]
+async def repos(update: Update, context: ContextTypes.DEFAULT_TYPE):
     user_id = str(update.effective_user.id)
-    resp = requests.post(f"{BACKEND_URL}/deploy", json={"github_url": github_url, "user_id": user_id})
+    resp = requests.get(f"{BACKEND_URL}/github/repos", params={"user_id": user_id})
     if resp.status_code == 200:
-        deploy_id = resp.json()["deploy_id"]
-        await update.message.reply_text(f"Deployment started! ID: {deploy_id}\nCheck status with /status {deploy_id}")
+        repos = resp.json()
+        if not repos:
+            await update.message.reply_text("No repos found.")
+            return ConversationHandler.END
+        repo_names = [r["full_name"] for r in repos]
+        markup = ReplyKeyboardMarkup([[name] for name in repo_names], one_time_keyboard=True)
+        await update.message.reply_text("Select a repo:", reply_markup=markup)
+        return SELECT_REPO
     else:
-        await update.message.reply_text("Failed to start deployment.")
-
-async def status(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    if len(context.args) != 1:
-        await update.message.reply_text("Usage: /status <deploy_id>")
-        return
-    deploy_id = context.args[0]
-    resp = requests.get(f"{BACKEND_URL}/status/{deploy_id}")
+        await update.message.reply_text("Failed to fetch repos. Make sure you are connected with GitHub.")
+        return ConversationHandler.END
+
+async def select_repo(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    user_id = str(update.effective_user.id)
+    repo_full_name = update.message.text
+    resp = requests.post(f"{BACKEND_URL}/repo/select", json={"user_id": user_id, "repo_full_name": repo_full_name})
     if resp.status_code == 200:
-        data = resp.json()
-        await update.message.reply_text(f"Status: {data['status']}\nURL: {data.get('url', 'N/A')}")
+        await update.message.reply_text(f"Repo selected: {repo_full_name}\nNow upload your .env file.", reply_markup=ReplyKeyboardRemove())
+        return UPLOAD_ENV
+    else:
+        await update.message.reply_text("Failed to select repo.", reply_markup=ReplyKeyboardRemove())
+        return ConversationHandler.END
+
+async def upload_env(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    user_id = str(update.effective_user.id)
+    if update.message.document:
+        file = await update.message.document.get_file()
+        file_bytes = await file.download_as_bytearray()
+        files = {"file": (update.message.document.file_name, file_bytes)}
+        data = {"user_id": user_id}
+        resp = requests.post(f"{BACKEND_URL}/env/upload", data=data, files=files)
+        if resp.status_code == 200:
+            await update.message.reply_text(".env uploaded! Now you can /deploy your project.")
+        else:
+            await update.message.reply_text("Failed to upload .env.")
     else:
-        await update.message.reply_text("Deployment not found.")
+        await update.message.reply_text("Please upload your .env file as a document.")
+    return ConversationHandler.END
 
 def main():
     TOKEN = os.environ.get("TELEGRAM_BOT_TOKEN")  # Set your bot token as env variable
     app = ApplicationBuilder().token(TOKEN).build()
+    conv_handler = ConversationHandler(
+        entry_points=[CommandHandler("repos", repos)],
+        states={
+            SELECT_REPO: [MessageHandler(filters.TEXT & ~filters.COMMAND, select_repo)],
+            UPLOAD_ENV: [MessageHandler(filters.Document.ALL, upload_env)]
+        },
+        fallbacks=[]
+    )
     app.add_handler(CommandHandler("start", start))
-    app.add_handler(CommandHandler("deploy", deploy))
-    app.add_handler(CommandHandler("status", status))
+    app.add_handler(CommandHandler("connect", connect))
+    app.add_handler(conv_handler)
     app.run_polling()
 
 if __name__ == "__main__":
-    main()
+    main()