Update app.py

app.py

@@ -79,12 +79,22 @@ def generate_ocr(method, img):
 
     return text_output, label
 
+# Clear Button Functionality
+def clear_and_save(output_text, output_label):
+    # Save the results to the repository before clearing
+    save_results_to_repo(output_text, output_label)
+    
+    # Return empty values to clear the Gradio interface
+    return "", ""
+
 # Gradio Interface
 image_input = gr.Image()
 method_input = gr.Radio(["PaddleOCR", "EasyOCR", "KerasOCR"], value="PaddleOCR")
 output_text = gr.Textbox(label="Extracted Text")
 output_label = gr.Textbox(label="Spam Classification")
 
+clear_button = gr.Button("Clear")  # Add the Clear button
+
 demo = gr.Interface(
     generate_ocr,
     inputs=[method_input, image_input],
@@ -94,6 +104,13 @@ demo = gr.Interface(
     theme="compact",
 )
 
+# Set up the Clear button functionality
+clear_button.click(
+    clear_and_save, 
+    inputs=[output_text, output_label],
+    outputs=[output_text, output_label]
+)
+
 # Launch App
 if __name__ == "__main__":
     demo.launch()