Update bot/ui/keyboards.py

bot/ui/keyboards.py

@@ -1,14 +1,20 @@
 # PATH: bot/ui/keyboards.py
 from hydrogram.types import InlineKeyboardMarkup, InlineKeyboardButton
-from bot.ui.callbacks import make, AUTH_JSON, AUTH_CI, CANCEL
+from bot.ui.callbacks import (
+    make,
+    AUTH_JSON, AUTH_CI, CANCEL,
+    MENU_HELP, MENU_AUTH, MENU_PROFILES, MENU_SPEEDTEST
+)
 
 def profiles_keyboard(profiles: list[dict]) -> InlineKeyboardMarkup:
     rows = []
     for p in profiles:
         pid = p["profile_id"]
         label = p.get("label") or pid[:6]
-        rows.append([InlineKeyboardButton(f"Set default: {label}", callback_data=f"setdef:{pid}")])
-    return InlineKeyboardMarkup(rows) if rows else InlineKeyboardMarkup([[InlineKeyboardButton("No profiles", callback_data="noop")]])
+        rows.append([InlineKeyboardButton(f"🟢 Set current: {label}", callback_data=f"setdef:{pid}")])
+    if not rows:
+        rows = [[InlineKeyboardButton("No profiles", callback_data="noop")]]
+    return InlineKeyboardMarkup(rows)
 
 def auth_menu_keyboard() -> InlineKeyboardMarkup:
     return InlineKeyboardMarkup(
@@ -17,4 +23,18 @@ def auth_menu_keyboard() -> InlineKeyboardMarkup:
             [InlineKeyboardButton("🔑 Enter Client ID & Secret", callback_data=make(AUTH_CI))],
             [InlineKeyboardButton("❌ Cancel", callback_data=make(CANCEL))],
         ]
+    )
+
+def main_menu_keyboard() -> InlineKeyboardMarkup:
+    return InlineKeyboardMarkup(
+        [
+            [
+                InlineKeyboardButton("🔐 Add Profile", callback_data=make(MENU_AUTH)),
+                InlineKeyboardButton("👤 Profiles", callback_data=make(MENU_PROFILES)),
+            ],
+            [
+                InlineKeyboardButton("⚡ Speedtest", callback_data=make(MENU_SPEEDTEST)),
+                InlineKeyboardButton("❓ Help", callback_data=make(MENU_HELP)),
+            ],
+        ]
     )