Update bot/ui/keyboards.py

bot/ui/keyboards.py

@@ -38,15 +38,6 @@ def auth_menu_keyboard() -> InlineKeyboardMarkup:
 
 
 def profiles_keyboard(profiles: list[dict], default_profile_id: str = "") -> InlineKeyboardMarkup:
-    """
-    Must match handlers.py:
-      profiles_keyboard(profiles, default_id)
-
-    handlers.py expects callback actions:
-      pdef:<profile_id>  set default
-      plog:<profile_id>  get login url
-      pdel:<profile_id>  delete profile (confirm by YES message)
-    """
     rows: list[list[InlineKeyboardButton]] = []
 
     default_profile_id = str(default_profile_id or "")
@@ -65,10 +56,8 @@ def profiles_keyboard(profiles: list[dict], default_profile_id: str = "") -> Inl
         status = "✅" if is_connected else ("🔑" if not has_refresh else "⚠️")
         title = f"{prefix}{status} [{i}] {label}"[:50]
 
-        # Title/info row
         rows.append([InlineKeyboardButton(title, callback_data=make("noop", pid[:8]))])
 
-        # Action row
         btns: list[InlineKeyboardButton] = []
         if pid:
             btns.append(InlineKeyboardButton("✅ Set default", callback_data=make("pdef", pid)))
@@ -78,7 +67,6 @@ def profiles_keyboard(profiles: list[dict], default_profile_id: str = "") -> Inl
         if btns:
             rows.append(btns)
 
-        # Optional small error row (if present)
         if last_error:
             rows.append([InlineKeyboardButton(f"⚠️ {last_error[:48]}", callback_data=make("noop", "err"))])
 
@@ -86,12 +74,23 @@ def profiles_keyboard(profiles: list[dict], default_profile_id: str = "") -> Inl
     return InlineKeyboardMarkup(rows)
 
 
-def upload_confirm_keyboard(privacy: str) -> InlineKeyboardMarkup:
+def upload_confirm_keyboard(privacy: str, title_mode: str = "caption") -> InlineKeyboardMarkup:
     priv_label = {"private": "🔒 Private", "unlisted": "🔗 Unlisted", "public": "🌍 Public"}.get(
         (privacy or "private").lower(), "🔒 Private"
     )
+
+    tm = (title_mode or "").lower().strip()
+    b1 = "✅ Filename" if tm == "filename" else "Filename"
+    b2 = "✅ Caption" if tm == "caption" else "Caption"
+    b3 = "✅ Custom" if tm == "custom" else "Custom"
+
     return InlineKeyboardMarkup(
         [
+            [
+                InlineKeyboardButton(b1, callback_data=make(NAME_ORIGINAL)),
+                InlineKeyboardButton(b2, callback_data=make(NAME_CAPTION)),
+                InlineKeyboardButton(b3, callback_data=make(NAME_CUSTOM)),
+            ],
             [InlineKeyboardButton("✅ Upload", callback_data=make(UP_GO))],
             [
                 InlineKeyboardButton("✏️ Edit title/desc", callback_data=make(UP_EDIT)),
@@ -102,15 +101,4 @@ def upload_confirm_keyboard(privacy: str) -> InlineKeyboardMarkup:
                 InlineKeyboardButton("⬅️ Back", callback_data=make(BACK)),
             ],
         ]
-    )
-
-
-def filename_keyboard() -> InlineKeyboardMarkup:
-    return InlineKeyboardMarkup(
-        [
-            [InlineKeyboardButton("📄 Original filename", callback_data=make(NAME_ORIGINAL))],
-            [InlineKeyboardButton("📝 From caption", callback_data=make(NAME_CAPTION))],
-            [InlineKeyboardButton("✍️ Custom name", callback_data=make(NAME_CUSTOM))],
-            [InlineKeyboardButton("❌ Cancel", callback_data=make(CANCEL))],
-        ]
     )