From 7cba39d1f9cc127aee8656d5ecf769b0b9e4638b Mon Sep 17 00:00:00 2001 From: ned Date: Sat, 1 Apr 2023 10:45:52 +0200 Subject: [PATCH] allow disabling image generation and transcriptions (#129) --- README.md | 2 ++ bot/main.py | 2 ++ bot/telegram_bot.py | 4 ++-- 3 files changed, 6 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 9134d6cd..3e646ac8 100644 --- a/README.md +++ b/README.md @@ -52,6 +52,8 @@ Customize the configuration by copying `.env.example` and renaming it to `.env`, ### Optional configuration | Parameter | Description | Default value | |------------------------------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------------| +| `ENABLE_IMAGE_GENERATION` | Whether to enable image generation via the `/image` command | true | +| `ENABLE_TRANSCRIPTION` | Whether to enable transcriptions of audio and video messages | true | | `MONTHLY_USER_BUDGETS` | A comma-separated list of $-amounts per user from list `ALLOWED_TELEGRAM_USER_IDS` to set custom usage limit of OpenAI API costs for each. **Note**: by default, *no limits* for anyone (`*`) | `*` | | `MONTHLY_GUEST_BUDGET` | $-amount as usage limit for all guest users. Guest users are users in group chats that are not in the `ALLOWED_TELEGRAM_USER_IDS` list. Value is ignored if no usage limits are set in user budgets (`MONTHLY_USER_BUDGETS`="*") | `100.0` | | `PROXY` | Proxy to be used for OpenAI and Telegram bot (e.g. `http://localhost:8080`) | - | diff --git a/bot/main.py b/bot/main.py index 95bb01f7..2165f0c6 100644 --- a/bot/main.py +++ b/bot/main.py @@ -48,6 +48,8 @@ def main(): 'token': os.environ['TELEGRAM_BOT_TOKEN'], 'admin_user_ids': os.environ.get('ADMIN_USER_IDS', '-'), 'allowed_user_ids': os.environ.get('ALLOWED_TELEGRAM_USER_IDS', '*'), + 'enable_image_generation': os.environ.get('ENABLE_IMAGE_GENERATION', 'true').lower() == 'true', + 'enable_transcription': os.environ.get('ENABLE_TRANSCRIPTION', 'true').lower() == 'true', 'monthly_user_budgets': os.environ.get('MONTHLY_USER_BUDGETS', '*'), 'monthly_guest_budget': float(os.environ.get('MONTHLY_GUEST_BUDGET', '100.0')), 'stream': os.environ.get('STREAM', 'true').lower() == 'true', diff --git a/bot/telegram_bot.py b/bot/telegram_bot.py index 9a932c81..459a13df 100644 --- a/bot/telegram_bot.py +++ b/bot/telegram_bot.py @@ -173,7 +173,7 @@ async def image(self, update: Update, context: ContextTypes.DEFAULT_TYPE): """ Generates an image for the given prompt using DALL·E APIs """ - if not await self.check_allowed_and_within_budget(update, context): + if not self.config['enable_image_generation'] or not await self.check_allowed_and_within_budget(update, context): return chat_id = update.effective_chat.id @@ -215,7 +215,7 @@ async def transcribe(self, update: Update, context: ContextTypes.DEFAULT_TYPE): """ Transcribe audio messages. """ - if not await self.check_allowed_and_within_budget(update, context): + if not self.config['enable_transcription'] or not await self.check_allowed_and_within_budget(update, context): return if self.is_group_chat(update) and self.config['ignore_group_transcriptions']: