Update chatbot/plugins/keyboards.py

chatbot/plugins/keyboards.py

@@ -1,28 +1,33 @@
 from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton
 from chatbot.plugins.languages import LANGUAGES
 
-def get_language_keyboard():
+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.
+    """
     keyboard = []
+    buttons = []
 
-    # 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("Cancel", callback_data="cancel")
-    ])
+    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")])
 
     return InlineKeyboardMarkup(keyboard)