Add logging for message history and improve question formatting in Conversation class

app.py

@@ -214,6 +214,7 @@ def predict(message,
                 if first_user_message is not None and \
                     DEFAULT_IMAGE_TOKEN not in first_user_message:
                     state.update_message(state.USER, DEFAULT_IMAGE_TOKEN + "\n" + first_user_message, None, index)
+        
 
         history = state.get_history()
         logger.info(f"====  History ====\n{history}")

conversation.py

@@ -10,7 +10,8 @@ from typing import Any, List, Dict, Union
 from dataclasses import field
 
 from utils import LOGDIR
-
+import logging
+logger = logging.getLogger(__name__)
 
 def pil2base64(img: Image.Image) -> str:
     buffered = BytesIO()
@@ -140,6 +141,8 @@ class Conversation:
         system_message = self.system_message
         messages = self.messages[:-1]
 
+        logger.info(f"=== Raw messages ===\n{messages}")
+
         for i in range(len(messages)):
             if messages[i]['role'] == 'user':
                 # Create the question by combining system and user messages
@@ -147,8 +150,8 @@ class Conversation:
 
                 # Check if it's the first user message and contains <image>
                 if i == 0 and '<image>' in user_content:
-                    question = f"<image> {system_message}\n{user_content}"
-                elif i == 0 and '<imgage>' not in user_content:
+                    question = f"<image>\n{system_message}\n{user_content}"
+                elif i == 0 and '<image>' not in user_content:
                     question = f"{system_message}\n{user_content}"
                 else:
                     question = f"{user_content}"