mattb512 commited on
Commit
a1091d4
1 Parent(s): 3b95d85

add validation

Browse files
Files changed (1) hide show
  1. serve/gradio_web_server.py +6 -0
serve/gradio_web_server.py CHANGED
@@ -114,6 +114,12 @@ def clear_history(request: gr.Request):
114
 
115
  def add_text(state, text, image, image_process_mode, request: gr.Request):
116
  # logger.info(f"add_text. ip: {request.client.host}. len: {len(text)}")
 
 
 
 
 
 
117
  if len(text) <= 0 and image is None:
118
  state.skip_next = True
119
  return (state, state.to_gradio_chatbot(), "", None) + (no_change_btn,) * 5
 
114
 
115
  def add_text(state, text, image, image_process_mode, request: gr.Request):
116
  # logger.info(f"add_text. ip: {request.client.host}. len: {len(text)}")
117
+ if not text:
118
+ raise gr.Error("Please provide a prompt.")
119
+
120
+ if not image:
121
+ raise gr.Error("Please provide an image.")
122
+
123
  if len(text) <= 0 and image is None:
124
  state.skip_next = True
125
  return (state, state.to_gradio_chatbot(), "", None) + (no_change_btn,) * 5