Update app.py

app.py

@@ -69,10 +69,26 @@ class Conversation:
         if "<image>" in message:
             return message.replace("<image>", "")
         return message
-    
+
     def dict(self):
+        # Create a serializable version of messages
+        serialized_messages = []
+        for role, message in self.messages:
+            if isinstance(message, tuple) and len(message) > 1:
+                # If the message contains an image (tuple format)
+                if isinstance(message[1], Image.Image):
+                    # Just keep the text part and ignore the image
+                    serialized_message = (message[0], "[IMAGE_IGNORED]")
+                else:
+                    # For non-image tuples, keep as is
+                    serialized_message = message
+            else:
+                # For non-tuple messages, keep as is
+                serialized_message = message
+            serialized_messages.append([role, serialized_message])
+
         return {
-            "messages": self.messages,
+            "messages": serialized_messages,
             "roles": self.roles,
             "offset": self.offset,
             "skip_next": self.skip_next,
@@ -278,8 +294,8 @@ def run_inference(prompt, image, temperature=0.2, top_p=0.95, max_tokens=512):
         return response.strip()
         
     except Exception as e:
-        logger.error(f"Error during inference: {str(e)}")
-        return f"Error during inference: {str(e)}"
+        logger.error(f"Error during inference: {e}")
+        return f"Error during inference: {e}"
 
 # Gradio UI functions
 get_window_url_params = """
@@ -456,16 +472,7 @@ block_css = """
 
 def build_demo(embed_mode, cur_dir=None, concurrency_count=10):
     models = get_model_list()
-    
-    with gr.Accordion("Safety Risk Taxonomy", open=False) as accordion:
-        textbox = gr.Textbox(
-            label="Safety Risk Taxonomy",
-            show_label=True,
-            placeholder="Enter your safety policy here",
-            container=True,
-            value=default_taxonomy,
-            lines=50)
-            
+
     with gr.Blocks(title="LlavaGuard", theme=gr.themes.Default(), css=block_css) as demo:
         state = gr.State()
 
@@ -502,6 +509,14 @@ def build_demo(embed_mode, cur_dir=None, concurrency_count=10):
                     max_output_tokens = gr.Slider(minimum=0, maximum=1024, value=512, step=64, interactive=True,
                                                  label="Max output tokens")
 
+                with gr.Accordion("Safety Risk Taxonomy", open=False):
+                    taxonomy_textbox = gr.Textbox(
+                        label="Safety Risk Taxonomy",
+                        show_label=True,
+                        placeholder="Enter your safety policy here",
+                        value=default_taxonomy,
+                        lines=20)
+
             with gr.Column(scale=8):
                 chatbot = gr.Chatbot(
                     elem_id="chatbot",
@@ -511,7 +526,13 @@ def build_demo(embed_mode, cur_dir=None, concurrency_count=10):
                 )
                 with gr.Row():
                     with gr.Column(scale=8):
-                        textbox.render()
+                        textbox = gr.Textbox(
+                            show_label=False,
+                            placeholder="Enter your message here",
+                            container=True,
+                            value=default_taxonomy,
+                            lines=3,
+                        )
                     with gr.Column(scale=1, min_width=50):
                         submit_btn = gr.Button(value="Send", variant="primary")
                 with gr.Row(elem_id="buttons") as button_row:
@@ -649,4 +670,4 @@ if __name__ == "__main__":
         )
     except Exception as e:
         logger.error(f"Error launching demo: {e}")
-        sys.exit(1)
+        sys.exit(1)