file modified

app.py

@@ -8,7 +8,7 @@ import os
 import logging
 
 # Set up logger
-logger = logging.getLogger(__name__)
+logger = logging.getLogger(_name_)
 logger.setLevel(logging.DEBUG)
 console_handler = logging.StreamHandler()
 file_handler = logging.FileHandler('chatbot_log.log')
@@ -80,17 +80,21 @@ def customLLMBot(user_input, uploaded_image, chat_history):
             logger.debug(f"Image received, size: {len(base64_image)} bytes")
 
             # Create a message for the image prompt
-            messages.append({
-                "role": "user",
-                "content": "What's in this image?",
-                "type": "image_url",  # If this is supported in Groq API
-                "image_url": {"url": f"data:image/png;base64,{base64_image}"}
-            })
+            # Create a message specifically for image prompts
+            messages_image=[
+                {
+                    "role": "user",
+                    "content": [
+                        {"type": "text", "text": "What's in this image?"},
+                        {"type": "image_url", "image_url": {"url": f"data:image/png;base64,{base64_image}"}}
+                    ]
+                }
+            ]
 
             logger.info("Sending image to Groq API for processing...")
             response = client.chat.completions.create(
                 model="llama-3.2-11b-vision-preview",
-                messages=messages,
+                messages=messages_image,
             )
             logger.info("Image processed successfully.")
         else:
@@ -112,6 +116,7 @@ def customLLMBot(user_input, uploaded_image, chat_history):
 
         # Append the bot's response to the chat history
         chat_history.append(("bot", LLM_reply))
+        messages.append({"role":"assistant","content":LLM_reply})
 
         # Generate audio for response
         audio_file = f"response_{uuid.uuid4().hex}.mp3"
@@ -181,9 +186,7 @@ def chatbot_ui():
 
     return demo
 
-
 # Launch the interface
 chatbot_ui().launch(server_name="0.0.0.0", server_port=7860)
 
-
 #chatbot_ui().launch(server_name="localhost", server_port=7860)