Add logging for modified conversation history in prediction function

app.py

@@ -210,6 +210,9 @@ def predict(message,
                     DEFAULT_IMAGE_TOKEN not in first_user_message:
                     state.messages[index]['content'] = DEFAULT_IMAGE_TOKEN + "\n" + first_user_message
 
+                    history = state.get_prompt()
+                    logger.info(f"==== Modified History ====\n{history}")
+
         response, conv_history = model.chat(tokenizer, pixel_values, question, generation_config, history=history, return_history=True)
         return response, conv_history
 
@@ -265,15 +268,13 @@ def http_bot(
         logger.info(f"==== Image paths ====\n{all_image_paths}")
 
         response, _ = predict(message, 
-                                         all_image_paths[0],
-                                         state,
-                                         max_input_tiles, 
-                                         temperature, 
-                                         max_new_tokens,
-                                         top_p, 
-                                         repetition_penalty)
-        # logger.info(f"==== AI history ====\n{conv_history}")
-
+                                all_image_paths if len(all_image_paths) > 0 else None,
+                                state,
+                                max_input_tiles, 
+                                temperature, 
+                                max_new_tokens,
+                                top_p, 
+                                repetition_penalty)
 
         # response = "This is a test response"
         buffer = ""