Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -51,10 +51,15 @@ from database import db
 from database import users_collection
 from logger import LOGS
 
+from akenoai import *
+from akenoai.types import DifferentAPIDefault
+
 from google import genai
 from google.genai import types as ty
 
 trans = SyncTranslator()
+js = AkenoXJs(DifferentAPIDefault()).connect()
+
 message_memory_state = {}
 
 async def geni_files_delete(name: str):
@@ -256,6 +261,67 @@ async def deletemydata(client, callback):
     else:
         await callback.answer(delm, True)
 
+@Client.on_callback_query(filters.regex("^fluxprompt_(\d+)$"))
+async def flux_prompt(client, callback):
+    user_id = int(callback.matches[0].group(1))
+    file_path = None
+    file_photo = None
+    try:
+        backup_chat = await db._get_chatbot_chat_from_db(user_id)
+        data = await db.backup_chatbot.find_one({"user_id": int(user_id)})
+        if not data:
+            return await callback.answer("Can't found user", True)
+        get_response = data.get("prompt_flux", None)
+        if not get_response:
+            return await callback.answer("Server busy try again later", True)
+        await callback.message.delete()
+        backup_chat.append({"role": "user", "parts": [{"text": get_response}]})
+        response_js = await js.image.create(
+            "black-forest-labs/flux-1-schnell",
+            image_read=True,
+            params_data={"query": get_response},
+        )
+        file_path = "randydev.jpg"
+        with open(file_path, "wb") as f:
+            f.write(response_js)
+
+        file_photo = await gen.aio.files.upload(file=file_path)
+        while file_photo.state.name == "PROCESSING":
+            await client.send_chat_action(message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
+            await asyncio.sleep(10)
+            file_photo = await gen.aio.files.get(name=file_photo.name)
+        if file_photo.state.name == "FAILED":
+            return await callback.message.reply_text(f"Error: {file_photo.state.name}")
+        response = await gen.aio.models.generate_content(
+            model="gemini-2.0-flash",
+            contents=[
+                "You are a picture making a story use small don't be too many",
+                file_photo
+            ]
+        )
+        keyboard = create_keyboard(user_id=user_id)
+        await callback.message.reply_photo(
+            file_path,
+            caption=response.text,
+            reply_markup=keyboard
+        )
+        await db.backup_chatbot.update_one(
+            {"user_id": user_id},
+            {"$set": {"translate_text": response.text}},
+            upsert=True
+        )
+        backup_chat.append({"role": "model", "parts": [{"text": response.text}]})
+    except Exception as e:
+        LOGS.error(f"flux_prompt failed: {str(e)}")
+        await callback.message.reply_text("Server error")
+    finally:
+        if file_path and file_photo:
+            try:
+                os.remove(file_path)
+                await gen.aio.files.delete(name=file_photo.name)
+            except:
+                pass
+
 @Client.on_callback_query(filters.regex("^removerbg_(\d+)$"))
 async def remover_bg(client, callback):
     user_id = int(callback.matches[0].group(1))