Merge pull request #1756 from THEGAMECHANGER416/imagify_bot

Added image generation telegram bot

Author: avinashkranjan

AI_Image_Generator_Telegram_Bot/README.md

@@ -0,0 +1,44 @@
+# Telegram AI Image Generator Bot 
+This is an AI Image generator bot. It can create original images from a single text prompt!
+
+
+# Screenshot
+![Bot Screenshot](telegramBotScreenShot.png)
+
+
+# Getting Started
+This package is simple to use and allows to build an image generation telegram bot
+
+## Prerequisites
+
+ - Python
+ - Telegram Bot API Key
+ - Replicate Stable Diffusion API Key
+
+## Installation
+
+ 
+
+ 1. Get a free telegram bot API key from @BotFather telegram bot. 
+ 2. Also get a free stable diffusion API key from https://replicate.com/account/api-tokens
+![Replicate Screenshot](replicatess.png)
+ 3. Clone the repo
+`git clone https://github.com/<your_username>/Amazing-Python-Scripts`
+`cd ai_image_generation_telegram_bot`
+ 3. Install the requirements from the requirements.txt file
+ ` pip install -r requirements.txt`
+ 4. Enter the bot token from BotFather in the **TELEGRAM_BOT_TOKEN** in the .env file.
+ 5. Enter the bot token from Replicate in the **REPLICATE_API_TOKEN** in the .env file.
+ 6. Now run the ai_bot.py file and start chatting.
+
+# Bot Commands
+You can see a list of commands by typing */help*
+
+| Commands          | Description                                   |
+|-------------------|-----------------------------------------------|
+| /start            | Start a conversation                          |
+| /help             | Get a list of all the commands                |
+| /imagine `prompt` | Create an image based on the given command    |
+
+#
+*Created by [Arnav Kohli](https://github.com/THEGAMECHANGER416)*

AI_Image_Generator_Telegram_Bot/ai_bot.py

@@ -0,0 +1,50 @@
+from dotenv import load_dotenv
+import os
+from telegram import Update
+import replicate
+from telegram.ext import ApplicationBuilder, CommandHandler, ContextTypes, MessageHandler, filters
+
+load_dotenv()
+
+
+def getUrl(prompt):
+    a = replicate.run(
+        "stability-ai/stable-diffusion:27b93a2413e7f36cd83da926f3656280b2931564ff050bf9575f1fdf9bcd7478",
+        input={"prompt": prompt}
+    )
+    print(a)
+    return a[0]
+
+
+async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    await update.message.reply_text(
+        f"Hello {update.effective_user.first_name} I am your personal AI Image Generator \n Use /help to know all commands.")
+
+
+async def help(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    await update.message.reply_text(
+        '''Here are a list of all commands:-
+        /start - Start a conversation
+        /help - Get a list of commands
+        /imagine <prompt> - Get an AI generated image'''
+    )
+
+
+async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    await update.message.reply_text("""
+        Sorry, I did not understand that command.
+        Type \" /help \" to see all possible commands""")
+
+
+async def imagine(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    arg = str(" ".join(context.args))
+    url = getUrl(arg)
+    await update.message.reply_photo(url)
+
+
+app = ApplicationBuilder().token(os.environ.get('TELEGRAM_BOT_TOKEN')).build()
+app.add_handler(CommandHandler("start", start))
+app.add_handler(CommandHandler("help", help))
+app.add_handler(CommandHandler("imagine", imagine))
+app.add_handler(MessageHandler(filters.COMMAND, handle_message))
+app.run_polling()

AI_Image_Generator_Telegram_Bot/replicatess.png

@@ -0,0 +1,4 @@
+requests
+python-telegram-bot
+python-dotenv
+replicate