Update app.py

app.py

@@ -1,8 +1,6 @@
 import gradio as gr
 from openai import OpenAI
 import os
-from gradio.components import ChatMessage
-
 
 client = OpenAI(
     base_url="https://integrate.api.nvidia.com/v1",
@@ -10,11 +8,11 @@ client = OpenAI(
 )
 
 def generate_response(message, history):
-    history_openai_format = []
     if history:
-        for chat_message in history:
-            history_openai_format.append(chat_message.dict())
-    
+        history_openai_format = history
+    else:
+        history_openai_format = []
+
     history_openai_format.append({"role": "user", "content": message})
 
     completion = client.chat.completions.create(
@@ -32,8 +30,7 @@ def generate_response(message, history):
             response += chunk.choices[0].delta.content
 
     history_openai_format.append({"role": "assistant", "content": response})
-    chat_history = [ChatMessage(role=msg["role"], content=msg["content"]) for msg in history_openai_format]
-    return response, chat_history
+    return response, history_openai_format
 
 iface = gr.ChatInterface(
     generate_response,
@@ -45,7 +42,7 @@ iface = gr.ChatInterface(
         "Kuantum bilgisayarların geleceği hakkında ne düşünüyorsun?"
     ],
     cache_examples=False,
-    type='chat'
+    type='messages'
 )
 
 iface.launch()