Upload chat.py

chatbot/plugins/chat.py

@@ -282,9 +282,9 @@ async def deletemydata(client, callback):
     else:
         await callback.answer(delm, True)
 
-@Client.on_callback_query(filters.regex("^genprompt$"))
+@Client.on_callback_query(filters.regex("^genprompt_(\d+)$"))
 async def geminigen_prompt(client, callback):
-    user_id = callback.from_user.id
+    user_id = int(callback.matches[0].group(1))
     user_bl = await db.user_blacklists.find_one({"user_id": user_id})
     if user_bl and user_bl.get("is_frozen", False) and user_id not in [6477856957]:
         await callback.answer("Sorry your account is frozen", True)
@@ -314,7 +314,6 @@ async def geminigen_prompt(client, callback):
                 response_modalities=['TEXT', 'IMAGE']
             )
         )
-        uuid_code = str(uuid.uuid4())[:8]
         for part in response.candidates[0].content.parts:
             if part.text is not None:
                 captions += part.text
@@ -322,6 +321,7 @@ async def geminigen_prompt(client, callback):
                 image = Image.open(BytesIO((part.inline_data.data)))
                 image.save(file_path)
 
+        uuid_code = str(uuid.uuid4())[:8]
         keyboard = create_keyboard(
             user_id=user_id,
             uuid_str=uuid_code
@@ -364,9 +364,9 @@ async def geminigen_prompt(client, callback):
             except:
                 pass
 
-@Client.on_callback_query(filters.regex("^fluxprompt$"))
+@Client.on_callback_query(filters.regex("^fluxprompt_(\d+)$"))
 async def flux_prompt(client, callback):
-    user_id = callback.from_user.id
+    user_id = int(callback.matches[0].group(1))
     user_bl = await db.user_blacklists.find_one({"user_id": user_id})
     if user_bl and user_bl.get("is_frozen", False) and user_id not in [6477856957]:
         await callback.answer("Sorry your account is frozen", True)
@@ -405,7 +405,6 @@ async def flux_prompt(client, callback):
             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}")
-        uuid_code = str(uuid.uuid4())[:8]
         response = await gen.aio.models.generate_content(
             model="gemini-2.0-flash",
             contents=[
@@ -413,6 +412,7 @@ async def flux_prompt(client, callback):
                 file_photo
             ]
         )
+        uuid_code = str(uuid.uuid4())[:8]
         keyboard = create_keyboard(
             user_id=user_id,
             uuid_str=uuid_code
@@ -639,7 +639,7 @@ async def multiple_langagues(client, callback):
         InlineKeyboardButton("🇷🇺", callback_data=f"tr_ru_{uuidstr}"),
     ])
     keyboard.append([
-        InlineKeyboardButton("💾", callback_data=f"memory_{uuidstr}")
+        InlineKeyboardButton("Back To Menu", callback_data=f"menu_back:{uuidstr}")
     ])
     try:
         await callback.edit_message_reply_markup(
@@ -1258,13 +1258,13 @@ async def chatbot_talk(client: Client, message: Message):
                         [
                             InlineKeyboardButton(
                                 text="🎨 Gemini AI Generate Image",
-                                callback_data="genprompt"
+                                callback_data=f"genprompt_{message.from_user.id}"
                             )
                         ],
                         [
                             InlineKeyboardButton(
                                 text="🖌️ FLUX AI Generate Image",
-                                callback_data="fluxprompt"
+                                callback_data=f"fluxprompt_{message.from_user.id}"
                             )
                         ],
                         [