Update app.py

app.py

@@ -19,16 +19,21 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
     messages.append({"role": "user", "content": message})
 
     response = ""
-    for message in client.chat_completion(
+    for chunk 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 history + [(message, response)]
+        if chunk.choices[0].delta.content is not None:
+            token = chunk.choices[0].delta.content
+            response += token
+            yield history + [(message, response)]
+
+    # If the response is empty, yield a default message
+    if not response:
+        yield history + [(message, "I apologize, but I couldn't generate a response.")]
 
 def clear_session():
     return "", []