update interface beta test2

app.py

@@ -24,11 +24,10 @@ def respond(message, chat_history):
     message_content = message
 
     try:
-        # Generate response using Cohere
-        response = co.generate(
+        # Generate response using Cohere chat API
+        response = co.chat(
             model='command-xlarge-nightly',  # Ensure the correct model ID is used
-            prompt=f"{prompt} '{message_content}'",
-            max_tokens=100,
+            query=f"{prompt} '{message_content}'",
             temperature=0.3,
             k=0,
             p=0.75,
@@ -44,11 +43,6 @@ def respond(message, chat_history):
     except Exception as e:
         return str(e), chat_history
 
-def slow_echo(message, history):
-    for i in range(len(message)):
-        time.sleep(0.05)
-        yield "You typed: " + message[: i + 1]
-
 # Create Gradio interface
 demo = gr.ChatInterface(
     respond,