update interface after entering error caption or image

app.py

@@ -82,10 +82,10 @@ def get_image_bytes(image):
 def process_image_and_caption(username_input, image, caption):
     if image is None:
         gr.Warning("No image uploaded!", duration=5)
-        return "No image uploaded!"
+        return "No image uploaded!", image, caption
     if caption is None:
         gr.Warning("No caption uploaded!", duration=5)
-        return "No caption uploaded!"
+        return "No caption uploaded!", image, caption
     # Get current time in GMT+7
     current_time = datetime.now(gmt_plus7)
 
@@ -159,11 +159,16 @@ with gr.Blocks() as demo:
         inputs=[username_input, image_input, caption_input],
         outputs=[output, image_input, caption_input]
     )
+    caption_input.submit(
+        process_image_and_caption,
+        inputs=[username_input, image_input, caption_input],
+        outputs=[output, image_input, caption_input]
+    )
 
+    
     username_input.submit(
         login, inputs=[username_input, state, package], outputs=[login_output, login_section, upload_section] #, translation_section, en_input, vi_input]
     )
-
     login_button.click(
         login, inputs=[username_input, state, package], outputs=[login_output, login_section, upload_section] #, translation_section, en_input, vi_input]
     )