Update app.py

app.py

@@ -29,7 +29,7 @@ def respond(
     try:
         client = get_client(model_name)
     except ValueError as e:
-        return str(e)
+        return chat_history + [[message, str(e)]]
 
     messages = [{"role": "system", "content": system_message}]
 
@@ -39,21 +39,22 @@ def respond(
 
     messages.append({"role": "user", "content": message})
 
-    response = ""
+    try:
+        response = client.chat_completion(
+            messages,
+            max_tokens=max_tokens,
+            temperature=temperature,
+            top_p=top_p,
+        )
+        assistant_message = response.choices[0].message.content
+    except Exception as e:
+        assistant_message = f"An error occurred: {str(e)}"
 
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.content
-        response += token
-        yield response
+    chat_history.append((message, assistant_message))
+    return chat_history
 
 def clear_conversation():
-    return None
+    return []
 
 with gr.Blocks() as demo:
     gr.Markdown("# Advanced AI Chatbot")