Refactor predict function to remove message parameter and retrieve message from history

app.py

@@ -185,8 +185,7 @@ model = AutoModel.from_pretrained(
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True, use_fast=False)
 
 @spaces.GPU
-def predict(message, 
-            image_path, 
+def predict(image_path, 
             state,
             max_input_tiles=6, 
             temperature=1.0,
@@ -218,6 +217,7 @@ def predict(message,
 
         history = state.get_history()
         logger.info(f"====  History ====\n{history}")
+        message = history[-1][0] if len(history) > 0 else ""
 
         logger.info(f"==== Lenght Pixel values ====\n{len(pixel_values)}")
         
@@ -282,15 +282,14 @@ def ai_bot(
         logger.info(f"==== User message ====\n{message}")
         logger.info(f"==== Image paths ====\n{all_image_paths}")
 
-        response, _ = predict(message, 
-                                all_image_paths[0] if len(all_image_paths) > 0 else None,
-                                state,
-                                max_input_tiles, 
-                                temperature, 
-                                max_new_tokens,
-                                top_p, 
-                                repetition_penalty,
-                                do_sample)
+        response, _ = predict(state,
+                            all_image_paths[0] if len(all_image_paths) > 0 else None,
+                            max_input_tiles, 
+                            temperature, 
+                            max_new_tokens,
+                            top_p, 
+                            repetition_penalty,
+                            do_sample)
 
         # response = "This is a test response"
         buffer = ""