_

akn/Gemini/gemini.py

@@ -41,6 +41,12 @@ from akn.utils.logger import LOGS
 import google.generativeai as genai
 from google.api_core.exceptions import InvalidArgument
 
+from akn.utils.scripts import progress
+from akenoai import *
+from akenoai.types import DifferentAPIDefault
+
+js = AkenoXJs(DifferentAPIDefault()).connect()
+
 async def geni_files_delete(name: str):
     url = f"https://generativelanguage.googleapis.com/v1beta/{name}"
     params = {"key": GOOGLE_API_KEY}
@@ -308,7 +314,31 @@ async def chatbot_talk(client: Client, message: Message):
         parts = query.split(maxsplit=1)
         command = parts[0].lower()
         pic_query = parts[1].strip() if len(parts) > 1 else ""
+        new_check_flux_matches = re.findall(r"\bimage\b", query_base)
         try:
+            if new_check_flux_matches:
+                new_js = await message.reply_text("Please wait, it's still being processed")
+                response_js = await js.image.create(
+                    "black-forest-labs/flux-1-schnell",
+                    image_read=True,
+                    params_data={"query": query_base},
+                )
+                file_path = "randydev.jpg"
+                with open(file_path, "wb") as f:
+                    f.write(response_js)
+                await new_js.edit_text("Uploading image...")
+                await message.reply_photo(
+                    file_path,
+                    progress=progress,
+                    progress_args=(
+                        pro,
+                        time.time(),
+                        "Uploading image..."
+                    )
+                )
+                await new_js.delete()
+                return
+    
             model_flash = genai.GenerativeModel(
                 model_name="gemini-1.5-flash"
             )