Update app.py

app.py

@@ -86,13 +86,18 @@ def generate_ocr(method, img):
     return text_output, label
 
 # Save results to JSON file
+RESULTS_JSON = "ocr_results.json"
+
 def save_to_json(text, label):
-    data = {"extracted_text": text, "classification": label}
-    with open(RESULTS_JSON, "w") as f:
-        json.dump(data, f, indent=4)
-    return "Results saved to JSON file!"
+    data = {"Extracted Text": text, "Spam Classification": label}
+
+    # Save to JSON file
+    with open(RESULTS_JSON, "w") as json_file:
+        json.dump(data, json_file, indent=4)
+
+    return f"Results saved to {RESULTS_JSON}"
 
-# Gradio Interface
+# Create Gradio Interface
 image_input = gr.Image()
 method_input = gr.Radio(["PaddleOCR", "EasyOCR", "KerasOCR"], value="PaddleOCR")
 output_text = gr.Textbox(label="Extracted Text")
@@ -100,7 +105,7 @@ output_label = gr.Textbox(label="Spam Classification")
 save_button = gr.Button("Save to JSON")
 save_output = gr.Textbox(label="Save Status")
 
-# Define the save button within the interface
+# Main OCR Interface
 demo = gr.Interface(
     fn=generate_ocr,
     inputs=[method_input, image_input],
@@ -110,14 +115,12 @@ demo = gr.Interface(
     theme="compact",
 )
 
-# Create a separate interface for saving the results
-save_interface = gr.Interface(
+# *Attach Save Button to Function*
+save_button.click(
     fn=save_to_json,
     inputs=[output_text, output_label],
-    outputs=[save_output],
-    live=False
+    outputs=[save_output]
 )
 
-# Launch both interfaces together
-demo.launch()
-save_interface.launch()
+# Launch App
+demo.launch()