v0.2.9 - Fix cost processing and enhance attachment download handling

router.py
- Fix type consistency in total_cost and operation_costs conversions
- Clarify bot instruction text formatting

utils/download_attachment.py
- Handle TelegramBadRequest for large files explicitly, removing generic exception handling

utils/download_attachment_pyrogram.py
- Update default target_dir path to "./downloads/"

Author: calmmage

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "new-whisper-bot"
-version = "0.2.8"
+version = "0.2.9"
 description = ""
 authors = ["Petr Lavrov <calmmage@gmail.com>"]
 readme = "README.md"

src/router.py

@@ -26,10 +26,7 @@ async def start_handler(message: Message, app: App):
         You can also <b>reply</b> to any text message from me to chat about its transcript or summary with LLM.
         """
     )
-    await send_safe(
-        message.chat.id,
-        welcome_message
-    )
+    await send_safe(message.chat.id, welcome_message)
 
 
 @commands_menu.botspot_command("help", "Show this help message")
@@ -63,14 +60,13 @@ async def help_handler(message: Message, app: App):
         So you might need to explore that as well to understand it. Basic template is available <a href="https://github.com/calmmage/botspot-template">here</a>
         """
     )
-    await send_safe(
-        message.chat.id,
-        help_message
-    )
+    await send_safe(message.chat.id, help_message)
+
 
 class Language(BaseModel):
     language_code: str
 
+
 @router.message(F.audio | F.voice | F.video | F.document | F.video_note)
 async def media_handler(message: Message, app: App, state: FSMContext):
     assert message.from_user is not None
@@ -99,7 +95,7 @@ async def media_handler(message: Message, app: App, state: FSMContext):
     notif = await reply_safe(
         message, "Processing your media file. Estimating transcription time..."
     )
-    
+
     # Transcribe the audio
     transcription = await app.process_message(
         message,
@@ -128,9 +124,9 @@ async def media_handler(message: Message, app: App, state: FSMContext):
 
     # Get and display cost information
     cost_info = await app.get_total_cost(username, message_id=message.message_id)
-    total_cost = cost_info["total_cost"]
+    total_cost = float(cost_info["total_cost"])
     if total_cost > 0.01:
-        operation_costs = float(cost_info["operation_costs"])
+        operation_costs = cost_info["operation_costs"]
 
         cost_breakdown = "\n".join(
             [
@@ -150,7 +146,7 @@ async def media_handler(message: Message, app: App, state: FSMContext):
 
 def create_notification_callback(notification_message: Message):
     text = notification_message.text
-    
+
     async def callback(update_text: str):
         nonlocal text
         assert text is not None
@@ -159,6 +155,7 @@ async def callback(update_text: str):
 
     return callback
 
+
 async def ask_user_language(message: Message, app: App, state: FSMContext):
     assert message.from_user is not None
     username = message.from_user.username
@@ -190,18 +187,21 @@ async def ask_user_language(message: Message, app: App, state: FSMContext):
             await message.reply("No language provided, please retry.")
             return
         parsed_language: Language = await aquery_llm_structured(
-            prompt = language_str,
+            prompt=language_str,
             output_schema=Language,
             model="gpt-4.1-nano",
             system_message="You are a language detection assistant. Your goal is to return the language code in ISO 639-1 format (e.g., 'en' for English, 'ru' for Russian).",
             user=username,
         )
         language_code = parsed_language.language_code
-        logger.info(f"User input: {language_str}, detected language code: {language_code}")
+        logger.info(
+            f"User input: {language_str}, detected language code: {language_code}"
+        )
     else:
         language_code = language
     return language_code
 
+
 @router.message()
 async def chat_handler(message: Message, app: App):
     if message.reply_to_message:
@@ -224,7 +224,9 @@ async def _reply_chat_handler(message: Message, app: App):
     prompt = await get_message_text(message, include_reply=True)
 
     # Process chat request
-    response = await app.chat_about_transcript(full_prompt=prompt, username=username, message_id=message.message_id)
+    response = await app.chat_about_transcript(
+        full_prompt=prompt, username=username, message_id=message.message_id
+    )
     response = markdown_to_html(response)
 
     cost = await app.get_total_cost(username, message_id=message.message_id)
@@ -246,6 +248,6 @@ async def _basic_chat_handler(message: Message, app: App):
     await answer_safe(
         message,
         "This is a Whisper bot. "
-        "\n\nSend an Audio, Video, Voice or Video Note message to transcribe and summarize it.",
+        "\n\nSend an Audio, Video, Voice or Video Note message to transcribe and summarize it."
         "\n\n<b>Reply</b> to a text message from bot to chat about its transcript or summary with LLM",
     )

src/utils/download_attachment.py

@@ -1,3 +1,4 @@
+from aiogram.exceptions import TelegramBadRequest
 from typing import BinaryIO, Optional, Union
 from pathlib import Path
 from loguru import logger
@@ -26,15 +27,11 @@ async def download_file(
             in_memory=True if in_memory is None else in_memory,
         )
 
-    except Exception as e:  # todo: specify api error
-        # use pyrogram downloader
-        logger.warning(
-            f"Generic error captured, update code to handle specific error, Type: {type(e)}, Error: {e}"
-        )
-        import traceback
-
-        traceback.print_exc()
-
+    except TelegramBadRequest as e:
+        if "file is too big" in str(e):
+            logger.info("File is too big, using pyrogram downloader")
+        else:
+            raise e
         from src.utils.download_attachment_pyrogram import (
             download_file_from_aiogram_message,
         )

src/utils/download_attachment_pyrogram.py

@@ -204,7 +204,7 @@ async def download_file_with_pyrogram(
     assert not isinstance(pyrogram_message, list)
 
     if target_dir is None:
-        target_dir = Path("../../utils")
+        target_dir = Path("./downloads/")
 
     if file_name is None:
         file_name = extract_file_name_from_pyrogram_message(