Update app.py

app.py

@@ -5,8 +5,8 @@ import requests
 HF_API_URL = "https://api-inference.huggingface.co/models/driaforall/mem-agent"
 HF_TOKEN = "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"
 
-USER_AVATAR = "https://i.imgur.com/YourUserAvatar.png"  # replace with your avatar URL
-AI_AVATAR = "https://i.imgur.com/YourAIAvatar.png"      # replace with AI avatar URL
+USER_AVATAR = "https://i.imgur.com/YourUserAvatar.png"
+AI_AVATAR = "https://i.imgur.com/YourAIAvatar.png"
 
 GOOGLE_VERIFICATION = "OXT03M7HRqpImYW2myqTjRmEP_f8wZi2SnbrO0JmGcc"
 
@@ -20,7 +20,6 @@ def call_ai(message):
     try:
         response = requests.post(HF_API_URL, headers=headers, json=payload)
         data = response.json()
-        # Parse AI response safely
         reply = "⚠️ No response from AI"
         if isinstance(data, list) and len(data) > 0:
             if "generated_text" in data[0]:
@@ -45,9 +44,11 @@ with gr.Blocks(title="Mem-Agent AI Chat", theme="default") as demo:
     """)
 
     gr.Markdown("## 💬 Mem-Agent AI Chat")
-    
+
     with gr.Column():
-        chat_history = gr.Chatbot(elem_id="chatbot").style(height=500)
+        # Updated for modern Gradio
+        chat_history = gr.Chatbot(elem_id="chatbot", type="messages")  
+
         with gr.Row():
             user_input = gr.Textbox(
                 placeholder="Type your message here...",
@@ -58,8 +59,8 @@ with gr.Blocks(title="Mem-Agent AI Chat", theme="default") as demo:
     # Handle sending messages
     def respond(message, chat):
         ai_reply = call_ai(message)
-        chat.append((f'<img src="{USER_AVATAR}" width="40"/> {message}',
-                     f'<img src="{AI_AVATAR}" width="40"/> {ai_reply}'))
+        chat.append({"role": "user", "content": f"{message}"})
+        chat.append({"role": "assistant", "content": f"{ai_reply}"})
         return "", chat
 
     submit_btn.click(respond, [user_input, chat_history], [user_input, chat_history])