Update app.py

app.py

@@ -1,4 +1,4 @@
-"""SusAI ©2025 Intern Labs. v1.1.0"""
+#SusAI ©2025 Intern Labs. v1.1.0
 import os
 import gradio as gr
 from gradio_client import Client
@@ -22,10 +22,10 @@ def chat_with_ai(message, history):
         top_p=0.60,
         api_name="/chat"
     )
-    return result
+    return {"role": "assistant", "content": result}
 
 # Gradio Chat Interface
-demo = gr.ChatInterface(fn=chat_with_ai)
+demo = gr.ChatInterface(fn=chat_with_ai, type="messages")
 
 if __name__ == "__main__":
     demo.launch()