Update chatbot/plugins/keyboards.py

chatbot/plugins/keyboards.py

@@ -1,33 +1,34 @@
 from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton
 from chatbot.plugins.languages import LANGUAGES
 
-def get_language_keyboard(buttons_per_row=2):
-    """
-    Generates an InlineKeyboardMarkup for language selection.
-    
-    :param buttons_per_row: Number of language buttons per row.
-    :return: InlineKeyboardMarkup object.
-    """
+def get_language_keyboard():
     keyboard = []
-    buttons = []
 
-    for index, lang in enumerate(LANGUAGES, start=1):
-        buttons.append(
-            InlineKeyboardButton(
-                text=lang['name'], 
-                callback_data=f"lang_{lang['code']}"
-            )
-        )
-        
-        if index % buttons_per_row == 0:
-            keyboard.append(buttons)
-            buttons = []
-    
-    # Append any remaining buttons
-    if buttons:
-        keyboard.append(buttons)
-    
-    # Add the Cancel button on a new row
-    keyboard.append([InlineKeyboardButton("Cancel", callback_data="cancel")])
+    # Row 1: English, Indonesian
+    keyboard.append([
+        InlineKeyboardButton("English", callback_data="lang_en"),
+        InlineKeyboardButton("Indonesian", callback_data="lang_id")
+    ])
+
+    # Row 2: Spanish
+    keyboard.append([
+        InlineKeyboardButton("Spanish", callback_data="lang_es")
+    ])
+
+    # Row 3: French, German, Japanese
+    keyboard.append([
+        InlineKeyboardButton("French", callback_data="lang_fr"),
+        InlineKeyboardButton("German", callback_data="lang_de"),
+        InlineKeyboardButton("Japanese", callback_data="lang_jp")
+    ])
+
+    keyboard.append([
+        InlineKeyboardButton("Malagasy", callback_data="lang_mg"),
+        InlineKeyboardButton("Russian", callback_data="lang_ru"),
+        InlineKeyboardButton("Hindi", callback_data="lang_hi"),
+    ])
+    keyboard.append([
+        InlineKeyboardButton("Cancel", callback_data="cancel")
+    ])
 
     return InlineKeyboardMarkup(keyboard)