Update bot/ui/keyboards.py

bot/ui/keyboards.py

@@ -2,39 +2,60 @@
 from hydrogram.types import InlineKeyboardMarkup, InlineKeyboardButton
 from bot.ui.callbacks import (
     make,
-    AUTH_JSON, AUTH_CI, CANCEL,
-    MENU_HELP, MENU_AUTH, MENU_PROFILES, MENU_SPEEDTEST
+    AUTH_JSON, AUTH_CI, CANCEL, BACK,
+    MENU_HELP, MENU_AUTH, MENU_PROFILES, MENU_SPEEDTEST,
+    UP_GO, UP_EDIT, UP_PRIV, UP_CANCEL
 )
 
-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 current: {label}", callback_data=f"setdef:{pid}")])
-    if not rows:
-        rows = [[InlineKeyboardButton("No profiles", callback_data="noop")]]
-    return InlineKeyboardMarkup(rows)
+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)),
+            ],
+        ]
+    )
 
 def auth_menu_keyboard() -> InlineKeyboardMarkup:
     return InlineKeyboardMarkup(
         [
-            [InlineKeyboardButton("📄 Paste client JSON", callback_data=make(AUTH_JSON))],
-            [InlineKeyboardButton("🔑 Enter Client ID & Secret", callback_data=make(AUTH_CI))],
-            [InlineKeyboardButton("❌ Cancel", callback_data=make(CANCEL))],
+            [InlineKeyboardButton("📄 Send .json file / Paste JSON", callback_data=make(AUTH_JSON))],
+            [InlineKeyboardButton("🔑 Send Client ID + Secret", callback_data=make(AUTH_CI))],
+            [InlineKeyboardButton("⬅️ Back", callback_data=make(BACK))],
         ]
     )
 
-def main_menu_keyboard() -> InlineKeyboardMarkup:
+def profiles_keyboard(profiles: list[dict]) -> InlineKeyboardMarkup:
+    rows = []
+    for i, p in enumerate(profiles, start=1):
+        pid = p.get("profile_id") or ""
+        ch = (p.get("channel_title") or p.get("label") or "Profile").strip()
+        btn = InlineKeyboardButton(f"🟢 Set current: [{i}] {ch}", callback_data=f"setdef:{pid}")
+        rows.append([btn])
+
+    rows.append([InlineKeyboardButton("⬅️ Back", callback_data=make(BACK))])
+    return InlineKeyboardMarkup(rows)
+
+def upload_confirm_keyboard(privacy: str) -> InlineKeyboardMarkup:
+    priv_label = {"private": "🔒 Private", "unlisted": "🔗 Unlisted", "public": "🌍 Public"}.get(
+        (privacy or "private").lower(),
+        "🔒 Private"
+    )
     return InlineKeyboardMarkup(
         [
+            [InlineKeyboardButton("✅ Upload", callback_data=make(UP_GO))],
             [
-                InlineKeyboardButton("🔐 Add Profile", callback_data=make(MENU_AUTH)),
-                InlineKeyboardButton("👤 Profiles", callback_data=make(MENU_PROFILES)),
+                InlineKeyboardButton("✏️ Edit title/desc", callback_data=make(UP_EDIT)),
+                InlineKeyboardButton(priv_label, callback_data=make(UP_PRIV)),
             ],
             [
-                InlineKeyboardButton("⚡ Speedtest", callback_data=make(MENU_SPEEDTEST)),
-                InlineKeyboardButton("❓ Help", callback_data=make(MENU_HELP)),
+                InlineKeyboardButton("❌ Cancel", callback_data=make(UP_CANCEL)),
+                InlineKeyboardButton("⬅️ Back", callback_data=make(BACK)),
             ],
         ]
     )