Update app.py

app.py

@@ -20,6 +20,9 @@ def respond(
     temperature,
     top_p,
 ):
+    if not message:
+        return ""
+    
     messages = [{"role": "system", "content": system_message}]
 
     for val in history:
@@ -32,20 +35,22 @@ def respond(
 
     response = ""
 
-    for message in client.text_generation(
-        prompt="\n".join([m["content"] for m in messages]),
-        max_new_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.token.text
-
-        response += token
-        yield response
+    try:
+        for message in client.text_generation(
+            prompt="\n".join([m["content"] for m in messages]),
+            max_new_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        ):
+            token = message.token.text
+            response += token
+            yield response
+    except Exception as e:
+        yield f"An error occurred: {str(e)}"
 
 def update_textbox(prompt):
-    return prompt
+    return gr.update(value=prompt)
 
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()