fix refresh

app.py

@@ -69,19 +69,13 @@ class OCRDataCollector:
         return block
 
     def submit_image(self, image, text_block):
-        if image is None:
-            message = "No image uploaded. Please try again or use 'Skip' to move on."
-        else:
+        if image is not None:
             timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
             self.collected_pairs.append({"text": text_block, "image": image, "timestamp": timestamp})
-            message = "Thank you! Your submission has been saved."
-        new_text = self.get_random_text_block()
-        return new_text, message
+        return self.get_random_text_block()
 
     def skip_text(self, text_block):
-        new_text = self.get_random_text_block()
-        message = "Skipped. Here is the next text."
-        return new_text, message
+        return self.get_random_text_block()
 
 def create_gradio_interface():
     collector = OCRDataCollector()
@@ -93,7 +87,6 @@ def create_gradio_interface():
         text_box = gr.Textbox(value=collector.current_text_block, label="Text to Handwrite", interactive=False)
         image_input = gr.Image(type="pil", label="Upload Handwritten Image", sources=["upload"])
         
-        
         with gr.Row():
             submit_btn = gr.Button("Submit")
             skip_btn = gr.Button("Skip")
@@ -101,13 +94,13 @@ def create_gradio_interface():
         submit_btn.click(
             fn=collector.submit_image,
             inputs=[image_input, text_box],
-            outputs=[text_box]
+            outputs=text_box
         )
         
         skip_btn.click(
             fn=collector.skip_text,
             inputs=text_box,
-            outputs=[text_box]
+            outputs=text_box
         )