everything working again after cleanup

vibe_bot/main.py

@@ -20,6 +20,10 @@ from vibe_bot.config import (
     DISCORD_TOKEN,
     IMAGE_EDIT_ENDPOINT,
     IMAGE_EDIT_ENDPOINT_KEY,
+    IMAGE_EDIT_MODEL,
+    IMAGE_GEN_ENDPOINT,
+    IMAGE_GEN_ENDPOINT_KEY,
+    IMAGE_GEN_MODEL,
     MAX_COMPLETION_TOKENS,
     TTS_MODEL_PATH,
     TTS_SPEED,
@@ -415,7 +419,7 @@ async def _speak_with_bot(
                 message_id=f"{ctx.message.id}_response",
                 user_id=str(ctx.bot.user.id),
                 username=ctx.bot.user.name,
-                content=f"Bot: {bot_response}",
+                content=bot_response,
                 channel_id=str(ctx.channel.id),
                 guild_id=str(ctx.guild.id) if ctx.guild else None,
             )
@@ -497,14 +501,23 @@ async def doodlebob(ctx: CommandsContext[Bot], *, message: str) -> None:
 
     image_b64 = llama_wrapper.image_generation(
         prompt=image_prompt,
-        openai_url=IMAGE_EDIT_ENDPOINT,
-        openai_api_key=IMAGE_EDIT_ENDPOINT_KEY,
+        openai_url=IMAGE_GEN_ENDPOINT,
+        openai_api_key=IMAGE_GEN_ENDPOINT_KEY,
+        model=IMAGE_GEN_MODEL,
     )
 
-    # Save the image to a file
-    edited_image_data = BytesIO(base64.b64decode(image_b64))
-    send_img = discord.File(edited_image_data, filename="image.png")
-    await ctx.send(file=send_img)
+    if not image_b64:
+        logger.warning("Image generation returned empty response.")
+        await ctx.send("Failed to generate image. The server may be busy.")
+        return
+
+    try:
+        edited_image_data = BytesIO(base64.b64decode(image_b64))
+        send_img = discord.File(edited_image_data, filename="image.png")
+        await ctx.send(file=send_img)
+    except Exception:
+        logger.exception("Failed to decode image data")
+        await ctx.send("Failed to process the generated image.")
 
 
 @bot.command(name="retcon")
@@ -529,6 +542,7 @@ async def retcon(ctx: CommandsContext[Bot], *, message: str) -> None:
         prompt=message,
         openai_url=IMAGE_EDIT_ENDPOINT,
         openai_api_key=IMAGE_EDIT_ENDPOINT_KEY,
+        model=IMAGE_EDIT_MODEL,
     )
 
     # Save the image to a file
@@ -621,7 +635,7 @@ async def talkforme(ctx: CommandsContext[Bot], *, message: str) -> None:
         bot_response = llama_wrapper.chat_completion_with_history(
             system_prompt=(
                 current_bot[1] + f"\nKeep your responses under 2-3 sentences. "
-                f"{current_bot[flip_counter(bot_counter)]}"
+                f"You are talking to {current_bot[flip_counter(bot_counter)][0]}"
             ),
             prompts=prompt_histories[bot_counter],
             openai_url=CHAT_ENDPOINT,
@@ -709,7 +723,7 @@ async def handle_chat(
                 message_id=f"{ctx.message.id}_response",
                 user_id=str(ctx.bot.user.id),
                 username=ctx.bot.user.name,
-                content=f"Bot: {bot_response}",
+                content=bot_response,
                 channel_id=str(ctx.channel.id),
                 guild_id=str(ctx.guild.id) if ctx.guild else None,
             )