Telegram Bot Integration

import os
import requests
from dotenv import load_dotenv
from telegram import Update
from telegram.ext import Application, CommandHandler, ContextTypes

load_dotenv()

TELEGRAM_TOKEN = os.getenv("TELEGRAM_TOKEN")
ANIMEAPI_KEY = os.getenv("ANIMEAPI_KEY")

async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """Send welcome message when /start is issued."""
    await update.message.reply_text(
        "Welcome to AnimeAPI Bot! \n\n"
        "Use /anime <prompt> to generate anime images.\n"
        "Example: /anime cute anime girl with blue hair"
    )

async def anime(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """Generate an anime image from the prompt."""
    if not context.args:
        await update.message.reply_text(
            "Please provide a prompt!\n"
            "Example: /anime cute anime girl"
        )
        return

    prompt = " ".join(context.args)

    # Send "generating" message
    status_msg = await update.message.reply_text("Generating your anime image...")

    try:
        # Call AnimeAPI
        response = requests.post(
            "https://api.animeapi.com/api/generate",
            headers={
                "Authorization": f"Bearer {ANIMEAPI_KEY}",
                "Content-Type": "application/json",
            },
            json={
                "prompt": prompt,
                "orientation": "portrait",  # portrait, square, or landscape
                "allowNSFW": True,
            },
        )

        data = response.json()

        if not response.ok:
            raise Exception(data.get("error", "Generation failed"))

        # Send the image
        await update.message.reply_photo(
            photo=data["image_url"],
            caption=f"Balance: ${data['balance_usd']:.2f}"
        )

        # Delete status message
        await status_msg.delete()

    except Exception as e:
        await status_msg.edit_text(f"Failed to generate image: {str(e)}")

async def credits(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """Check remaining credits (by making a test request)."""
    await update.message.reply_text(
        "To check your credits, visit your AnimeAPI dashboard:\n"
        "https://animeapi.ai/dashboard"
    )

def main():
    """Start the bot."""
    application = Application.builder().token(TELEGRAM_TOKEN).build()

    # Add handlers
    application.add_handler(CommandHandler("start", start))
    application.add_handler(CommandHandler("anime", anime))
    application.add_handler(CommandHandler("credits", credits))

    # Run the bot
    print("Bot is running...")
    application.run_polling(allowed_updates=Update.ALL_TYPES)

if __name__ == "__main__":
    main()

require('dotenv').config();
const { Telegraf } = require('telegraf');

const bot = new Telegraf(process.env.TELEGRAM_TOKEN);
const ANIMEAPI_KEY = process.env.ANIMEAPI_KEY;

bot.start((ctx) => ctx.reply('Use /anime <prompt> to generate anime images!'));

bot.command('anime', async (ctx) => {
  const prompt = ctx.message.text.split(' ').slice(1).join(' ');

  if (!prompt) {
    return ctx.reply('Please provide a prompt! Example: /anime cute anime girl');
  }

  const statusMsg = await ctx.reply('Generating...');

  try {
    const response = await fetch('https://api.animeapi.com/api/generate', {
      method: 'POST',
      headers: {
        'Authorization': `Bearer ${ANIMEAPI_KEY}`,
        'Content-Type': 'application/json',
      },
      body: JSON.stringify({
        prompt,
        orientation: 'portrait', // portrait, square, or landscape
        allowNSFW: true,
      }),
    });

    const data = await response.json();

    if (!response.ok) throw new Error(data.error);

    await ctx.replyWithPhoto(data.image_url, {
      caption: `Balance: $${data.balance_usd.toFixed(2)}`
    });
    await ctx.deleteMessage(statusMsg.message_id);
  } catch (error) {
    await ctx.telegram.editMessageText(
      ctx.chat.id,
      statusMsg.message_id,
      null,
      `Error: ${error.message}`
    );
  }
});

bot.launch();
console.log('Bot is running...');