Create app.py

app.py

@@ -0,0 +1,225 @@
+"""
+bot.py – Telegram bot for Frenix API key generation (free tier only).
+
+Setup:
+  1. pip install python-telegram-bot requests python-dotenv
+  2. Set BOT_TOKEN and API_BASE_URL in your .env or environment:
+       BOT_TOKEN=<your Telegram bot token from @BotFather>
+       API_BASE_URL=https://your-vercel-app.vercel.app   (or http://localhost:5000)
+  3. python bot.py
+
+The bot will:
+  - Ask the user for their first name, last name, and username
+  - Call POST /v1/api-keys on your Frenix API server
+  - Return the generated free-tier API key
+"""
+
+import os
+import logging
+import requests
+from dotenv import load_dotenv
+from telegram import Update, ReplyKeyboardRemove
+from telegram.ext import (
+    ApplicationBuilder,
+    CommandHandler,
+    MessageHandler,
+    ConversationHandler,
+    ContextTypes,
+    filters,
+)
+
+load_dotenv()
+
+# ── Config ────────────────────────────────────────────────────────────────────
+BOT_TOKEN    = os.environ.get("BOT_TOKEN", "8388247852:AAEAfqWAB0selpoWGt6WzzuhkpK-VCU4ES4")
+API_BASE_URL = os.environ.get("API_BASE_URL", "https://api.frenix.sh").rstrip("/")
+
+if not BOT_TOKEN:
+    raise RuntimeError("BOT_TOKEN is not set. Add it to your .env file.")
+
+logging.basicConfig(
+    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+    level=logging.INFO,
+)
+log = logging.getLogger(__name__)
+
+# ── Conversation states ───────────────────────────────────────────────────────
+FIRST_NAME, LAST_NAME, USERNAME_INPUT = range(3)
+
+
+# ── Helpers ───────────────────────────────────────────────────────────────────
+
+def _create_api_key(user_id: str, username: str, first_name: str, last_name: str) -> dict:
+    """Call POST /v1/api-keys and return the JSON response."""
+    url = f"{API_BASE_URL}/v1/api-keys"
+    payload = {
+        "user_id":    str(user_id),
+        "username":   username,
+        "first_name": first_name,
+        "last_name":  last_name,
+    }
+    try:
+        resp = requests.post(url, json=payload, timeout=15)
+        return resp.json()
+    except requests.exceptions.ConnectionError:
+        return {"error": f"Could not connect to the API server at {API_BASE_URL}"}
+    except Exception as e:
+        return {"error": str(e)}
+
+
+# ── Handlers ──────────────────────────────────────────────────────────────────
+
+async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
+    """Entry point – /start command."""
+    user = update.effective_user
+    await update.message.reply_text(
+        f"👋 Hello, {user.first_name}!\n\n"
+        "Welcome to *Frenix API* key generator.\n"
+        "I'll create a *free-tier* API key for you in just a few steps.\n\n"
+        "What is your *first name*? (or /cancel to abort)",
+        parse_mode="Markdown",
+        reply_markup=ReplyKeyboardRemove(),
+    )
+    return FIRST_NAME
+
+
+async def get_first_name(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
+    """Store first name and ask for last name."""
+    context.user_data["first_name"] = update.message.text.strip()
+    await update.message.reply_text(
+        "Great! Now enter your *last name*:",
+        parse_mode="Markdown",
+    )
+    return LAST_NAME
+
+
+async def get_last_name(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
+    """Store last name and ask for username."""
+    context.user_data["last_name"] = update.message.text.strip()
+    await update.message.reply_text(
+        "Almost done! Enter your preferred *username* (no spaces, e.g. john_doe):",
+        parse_mode="Markdown",
+    )
+    return USERNAME_INPUT
+
+
+async def get_username(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
+    """Store username, call the API, and return the key."""
+    tg_user    = update.effective_user
+    first_name = context.user_data.get("first_name", "")
+    last_name  = context.user_data.get("last_name", "")
+    username   = update.message.text.strip().replace(" ", "_")
+
+    await update.message.reply_text("⏳ Generating your API key, please wait…")
+
+    result = _create_api_key(
+        user_id    = tg_user.id,
+        username   = username,
+        first_name = first_name,
+        last_name  = last_name,
+    )
+
+    if "error" in result:
+        await update.message.reply_text(
+            f"❌ Error: {result['error']}\n\nPlease try again with /start.",
+        )
+        return ConversationHandler.END
+
+    api_key = result.get("api_key", "N/A")
+    plan    = result.get("plan", "free")
+    rpm     = result.get("limits", {}).get("rpm", "?")
+    message = result.get("message", "")
+
+    await update.message.reply_text(
+        f"✅ *Your Frenix API Key is ready!*\n\n"
+        f"```\n{api_key}\n```\n\n"
+        f"📋 *Details:*\n"
+        f"• Plan: `{plan}`\n"
+        f"• Rate limit: `{rpm}` requests/min\n"
+        f"• Name: `{first_name} {last_name}`\n"
+        f"• Username: `{username}`\n\n"
+        f"ℹ️ {message}\n\n"
+        f"🔗 *Base URL:* `{API_BASE_URL}/v1`\n\n"
+        f"Use this key as your `Authorization: Bearer <key>` header.\n"
+        f"Type /start to generate another key.",
+        parse_mode="Markdown",
+    )
+    return ConversationHandler.END
+
+
+async def cancel(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
+    """Cancel the conversation."""
+    await update.message.reply_text(
+        "❌ Cancelled. Type /start whenever you want a new API key.",
+        reply_markup=ReplyKeyboardRemove(),
+    )
+    return ConversationHandler.END
+
+
+async def mykey(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    """/mykey – retrieve existing key for this Telegram user."""
+    tg_user = update.effective_user
+    result  = _create_api_key(
+        user_id    = tg_user.id,
+        username   = tg_user.username or "",
+        first_name = tg_user.first_name or "",
+        last_name  = tg_user.last_name or "",
+    )
+    if "error" in result:
+        await update.message.reply_text(f"❌ Error: {result['error']}")
+        return
+
+    api_key = result.get("api_key", "N/A")
+    plan    = result.get("plan", "free")
+    rpm     = result.get("limits", {}).get("rpm", "?")
+    message = result.get("message", "")
+
+    await update.message.reply_text(
+        f"🔑 *Your Frenix API Key:*\n\n"
+        f"```\n{api_key}\n```\n\n"
+        f"• Plan: `{plan}`\n"
+        f"• Rate limit: `{rpm}` req/min\n\n"
+        f"ℹ️ {message}",
+        parse_mode="Markdown",
+    )
+
+
+async def help_cmd(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    """/help command."""
+    await update.message.reply_text(
+        "🤖 *Frenix API Bot*\n\n"
+        "Commands:\n"
+        "• /start – Generate a new free-tier API key\n"
+        "• /mykey – Retrieve your existing API key\n"
+        "• /help  – Show this help message\n"
+        "• /cancel – Cancel current operation\n\n"
+        f"🔗 API Base URL: `{API_BASE_URL}/v1`",
+        parse_mode="Markdown",
+    )
+
+
+# ── Main ──────────────────────────────────────────────────────────────────────
+
+def main():
+    app = ApplicationBuilder().token(BOT_TOKEN).build()
+
+    conv_handler = ConversationHandler(
+        entry_points=[CommandHandler("start", start)],
+        states={
+            FIRST_NAME:     [MessageHandler(filters.TEXT & ~filters.COMMAND, get_first_name)],
+            LAST_NAME:      [MessageHandler(filters.TEXT & ~filters.COMMAND, get_last_name)],
+            USERNAME_INPUT: [MessageHandler(filters.TEXT & ~filters.COMMAND, get_username)],
+        },
+        fallbacks=[CommandHandler("cancel", cancel)],
+    )
+
+    app.add_handler(conv_handler)
+    app.add_handler(CommandHandler("mykey", mykey))
+    app.add_handler(CommandHandler("help", help_cmd))
+
+    log.info(f"Bot started. API server: {API_BASE_URL}")
+    app.run_polling(allowed_updates=Update.ALL_TYPES)
+
+
+if __name__ == "__main__":
+    main()