Update app.py

app.py

@@ -2,7 +2,14 @@ import os
 import base64
 import requests
 import gradio as gr
-from huggingface_hub import InferenceClient, ChatMessage
+from huggingface_hub import InferenceClient
+from dataclasses import dataclass
+
+@dataclass
+class ChatMessage:
+    """Custom ChatMessage class since huggingface_hub doesn't provide one"""
+    role: str
+    content: str
 
 class XylariaChat:
     def __init__(self):
@@ -26,7 +33,9 @@ class XylariaChat:
         self.persistent_memory = {}
         
         # System prompt with more detailed instructions
-        self.system_prompt = """You are a helpful and harmless assistant. You are Xylaria developed by Sk Md Saad Amin(india, 12 year old). You should think step-by-step."""
+        self.system_prompt = """You are a helpful and harmless AI assistant named Xylaria. 
+Always think step-by-step and provide clear, thoughtful responses. 
+Be kind, ethical, and supportive in your interactions."""
     
     def store_information(self, key, value):
         """Store important information in persistent memory"""
@@ -240,11 +249,16 @@ class XylariaChat:
                             placeholder="Type your message...", 
                             container=False
                         )
-                        img = gr.Image(
-                            sources=["upload", "webcam"], 
-                            type="filepath", 
-                            label="Upload or Capture Image"
-                        )
+                        
+                        # Image upload as a separate button
+                        with gr.Row():
+                            img = gr.Image(
+                                sources=["upload", "webcam"], 
+                                type="filepath", 
+                                label="Upload Image",
+                                visible=False
+                            )
+                            upload_btn = gr.Button("Upload Image")
                     
                     btn = gr.Button("Send", scale=1)
                 
@@ -253,6 +267,13 @@ class XylariaChat:
                     clear = gr.Button("Clear Conversation")
                     clear_memory = gr.Button("Clear Memory")
             
+            # Image upload toggle
+            upload_btn.click(
+                fn=lambda: gr.update(visible=True), 
+                inputs=None, 
+                outputs=[img]
+            )
+            
             # Submit functionality with streaming and image support
             btn.click(
                 fn=streaming_response,