Update app.py

app.py

@@ -7,7 +7,6 @@ import os
 import sys
 import time
 import spaces
-
 import gradio as gr
 import torch
 from PIL import Image
@@ -18,7 +17,6 @@ from transformers import (
     LlavaOnevisionForConditionalGeneration
 )
 from qwen_vl_utils import process_vision_info
-
 from taxonomy import policy_v1
 
 # Set up logging
@@ -92,11 +90,28 @@ class SimpleConversation:
     def dict(self):
         # Simplified serialization for logging
         image_info = "[WITH_IMAGE]" if self.current_image is not None else "[NO_IMAGE]"
+
+        # Handle prompt which might be a tuple containing an image
+        prompt = self.get_prompt()
+        if isinstance(prompt, tuple):
+            prompt = prompt[0]  # Just take the text part
+
+        # Create JSON-safe message representations
+        safe_messages = []
+        for msg in self.messages:
+            msg_prompt = msg[0]
+            # Handle tuple prompts that contain images
+            if isinstance(msg_prompt, tuple) and len(msg_prompt) > 0:
+                msg_prompt = msg_prompt[0]  # Take just the text part
+
+            # Add the message with safe values
+            safe_messages.append([msg_prompt, "[RESPONSE]" if msg[1] else None])
+
         return {
-            "prompt": self.get_prompt(),
+            "prompt": prompt,
             "image": image_info,
             "response": self.current_response,
-            "messages": [[m[0], "[RESPONSE]" if m[1] else None] for m in self.messages]
+            "messages": safe_messages
         }
 
     def copy(self):