diff --git a/handlers/__init__.py b/handlers/__init__.py index d8def7b..b2f9da6 100644 --- a/handlers/__init__.py +++ b/handlers/__init__.py @@ -220,9 +220,10 @@ def enrich_text_with_urls(text: str) -> str: def image_to_data_uri(file_path): + content_type = guess_type(file_path)[0] with open(file_path, "rb") as image_file: encoded_image = base64.b64encode(image_file.read()).decode("utf-8") - return f"data:image/png;base64,{encoded_image}" + return f"data:{content_type};base64,{encoded_image}" import json diff --git a/handlers/useful.py b/handlers/useful.py index 17d9228..239dd2e 100644 --- a/handlers/useful.py +++ b/handlers/useful.py @@ -389,7 +389,7 @@ def answer_it_handler(message: Message, bot: TeleBot) -> None: #### Complete Messages #### if CHATGPT_COMPLETE and CHATGPT_API_KEY: full_answer += complete_chatgpt_future.result() - if CLADUE_COMPLETE and ANTHROPIC_API_KEY and not local_image_path: + if CLADUE_COMPLETE and ANTHROPIC_API_KEY: full_answer += complete_claude_future.result() if COHERE_COMPLETE and COHERE_API_KEY and not local_image_path: full_answer += complete_cohere_future.result()