Update app.py

app.py

@@ -938,8 +938,9 @@ def chat_response_stream_multiturn(
             if STREAM_CHECK_MULTIPLE > 0 and j % STREAM_CHECK_MULTIPLE == 0:
                 message_safety = safety_check(cur_out, history=None)
                 if message_safety is not None and not is_by_pass:
-                    yield message_safety
-                    return
+                    # yield message_safety
+                    raise gr.Error(message_safety)
+                    # return
 
             yield cur_out
         assert len(gen) == 1, f'{gen}'
@@ -964,8 +965,9 @@ def chat_response_stream_multiturn(
     
     message_safety = safety_check(cur_out, history=None)
     if message_safety is not None and not is_by_pass:
-        yield message_safety
-        return
+        # yield message_safety
+        raise gr.Error(message_safety)
+        # return