Update app.py

app.py

@@ -100,12 +100,14 @@ def bot(history, api_kind):
     else:
          gr.Warning(f"API {api_kind} is not supported")
          raise ValueError(f"API {api_kind} is not supported")
-
-    history[-1][1] = ""
-    for character in generate_fn(prompt, history[:-1]):
-        history[-1][1] = character
-        yield history, prompt_html
-
+    try:
+        print(prompt_html)
+        history[-1][1] = ""
+        for character in generate_fn(prompt, history[:-1]):
+            history[-1][1] = character
+            yield history, prompt_html
+     except Exception as e:  # Catch any exception
+        yield f"An unexpected error occurred during generation: {str(e)}"
 
 with gr.Blocks() as demo:
     # Beautiful heading with logo