Merge commit '3eaef5dc90b716cc0fa391cbe2394be56d5f6041' as 'userbot'

userbot/modules/flux.py

@@ -0,0 +1,51 @@
+import os
+import time
+
+import requests
+from pyrogram import Client, filters
+from pyrogram.types import Message
+from utils.misc import modules_help, prefix
+from utils.scripts import format_exc, progress
+
+
+def schellwithflux(args):
+    API_URL = 'https://randydev-ryuzaki-api.hf.space/api/v1/akeno/fluxai'
+    payload = {'user_id': 1191668125, 'args': args}  # Please don't edit here
+    response = requests.post(API_URL, json=payload, timeout=10)
+    if response.status_code != 200:
+        print(f'Error status {response.status_code}')
+        return None
+    return response.content
+
+
+@Client.on_message(filters.command('fluxai', prefix) & filters.me)
+async def imgfluxai_(client: Client, message: Message):
+    question = message.text.split(' ', 1)[1] if len(message.command) > 1 else None
+    if not question:
+        return await message.reply_text('Please provide a question for Flux.')
+    try:
+        image_bytes = schellwithflux(question)
+        if image_bytes is None:
+            return await message.reply_text('Failed to generate an image.')
+        pro = await message.reply_text('Generating image, please wait...')
+
+        # Write the image bytes directly to a file
+        with open('flux_gen.jpg', 'wb') as f:
+            f.write(image_bytes)
+
+        ok = await pro.edit_text('Uploading image...')
+        await message.reply_photo(
+            'flux_gen.jpg',
+            progress=progress,
+            progress_args=(ok, time.time(), 'Uploading image...'),
+        )
+        await ok.delete()
+        if os.path.exists('flux_gen.jpg'):
+            os.remove('flux_gen.jpg')
+    except Exception as e:
+        await message.edit_text(format_exc(e))
+
+
+modules_help['fluxai'] = {
+    'fluxai [prompt]*': 'text to image fluxai',
+}