66 lines
2.8 KiB
Python
66 lines
2.8 KiB
Python
import os
|
|
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
|
|
from telegram.ext import ContextTypes
|
|
|
|
browse_command_bot = None
|
|
|
|
async def browse_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
|
prompts_dir = "prompts"
|
|
|
|
await navigate_to(prompts_dir, update.message.reply_text)
|
|
|
|
async def navigate_to(directory: str, updateMethod) -> None:
|
|
try:
|
|
subdirs, files = await get_files_and_directories(directory)
|
|
|
|
existing_buttons = []
|
|
|
|
if (directory != "prompts"):
|
|
parent_directory = os.path.split(directory)[0]
|
|
existing_buttons.append([InlineKeyboardButton("↩️ Back", callback_data=f"browse:{parent_directory}")])
|
|
|
|
reply_markup = await create_keyboard(subdirs, files, existing_buttons)
|
|
|
|
await updateMethod("Browse files and directories:", reply_markup=reply_markup)
|
|
|
|
except FileNotFoundError:
|
|
await updateMethod(f"The {directory} directory does not exist.")
|
|
except Exception as e:
|
|
await updateMethod(f"An error occurred: {str(e)}")
|
|
|
|
async def create_keyboard(subdirs, files, existing_buttons) -> InlineKeyboardMarkup:
|
|
keyboard = existing_buttons
|
|
for subdir in subdirs:
|
|
keyboard.append([InlineKeyboardButton(subdir, callback_data=f"browse:{subdir}")])
|
|
for file in files:
|
|
keyboard.append([InlineKeyboardButton(file, callback_data=f"file:{file}")])
|
|
return InlineKeyboardMarkup(keyboard)
|
|
|
|
async def get_files_and_directories(directory: str) -> list:
|
|
# Get all items in the prompts directory
|
|
items = os.listdir(directory)
|
|
|
|
# Separate and sort subdirectories and files
|
|
subdirs = sorted([os.path.join(directory, item) for item in items if os.path.isdir(os.path.join(directory, item))])
|
|
files = sorted([os.path.join(directory, item) for item in items if os.path.isfile(os.path.join(directory, item))])
|
|
return subdirs, files
|
|
|
|
# This function will need to be called when a button is pressed
|
|
async def button_callback(update: Update, context: ContextTypes.DEFAULT_TYPE, browse_command_bot) -> None:
|
|
query = update.callback_query
|
|
await query.answer()
|
|
|
|
data = query.data
|
|
if data.startswith("browse:"):
|
|
# Handle subdirectory navigation (for future implementation)
|
|
subdir = data.split(":")[1]
|
|
await navigate_to(subdir, query.edit_message_text)
|
|
elif data.startswith("file:"):
|
|
# Handle file selection
|
|
file_path = data.split(":")[1]
|
|
os.environ['SYSTEM_PROMPT_PATH'] = file_path
|
|
if hasattr(browse_command_bot, 'system_prompt_path'):
|
|
browse_command_bot.system_prompt_path = file_path
|
|
browse_command_bot.system_prompt = browse_command_bot.load_system_prompt()
|
|
await query.edit_message_text(f"Selected: {file_path}. System prompt updated.")
|