Implement get_history method and update message handling in predict function

app.py

@@ -213,9 +213,9 @@ def predict(message,
                 index, first_user_message = state.get_user_message(source=state.USER, position='first')
                 if first_user_message is not None and \
                     DEFAULT_IMAGE_TOKEN not in first_user_message:
-                    state.messages[index]['content'] = DEFAULT_IMAGE_TOKEN + "\n" + first_user_message
+                    state.update_message(state.USER, DEFAULT_IMAGE_TOKEN + "\n" + first_user_message, None, index)
 
-        history = state.get_prompt()[:-1]
+        history = state.get_history()
         logger.info(f"====  History ====\n{history}")
 
         logger.info(f"==== Lenght Pixel values ====\n{len(pixel_values)}")

conversation.py

@@ -134,6 +134,33 @@ class Conversation:
             else:
                 raise ValueError(f"Invalid role: {message['role']}")
         return send_messages
+    
+    def get_history(self):
+        results = []
+        system_message = self.system_message
+        messages = self.messages[:-1]
+
+        for i in range(len(messages)):
+            if messages[i]['role'] == 'user':
+                # Create the question by combining system and user messages
+                user_content = messages[i]['content']
+
+                # 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}"
+                else:
+                    question = f"{system_message}\n{user_content}"
+
+                # Check for the corresponding assistant response
+                answer = ""
+                if i + 1 < len(messages) and messages[i + 1]['role'] == 'assistant':
+                    answer = messages[i + 1]['content']
+
+                # Add the question-answer pair to results
+                results.append((question, answer))
+
+        return results
+        
 
     def append_message(
         self,