Cancel fix

app.py

@@ -9,25 +9,6 @@ LOADED_TTF_KEY = "loaded_ttf"
 SET_IMG_KEY = "set_img"
 OUTPUT_IMG_KEY = "output_img"
 
-# For multithreading toggle (prevent function from running too many time)
-process_running = False
-process_thread = None
-cancel_label = st.empty()
-
-def toggle_process(process_running, process_thread, run_process):
-    if process_running:
-        # Toggle off
-        process_running = False
-        cancel_label.text("Canceled")
-        if process_thread: # Kill Thread
-            process_thread.join()
-    else:
-        # Toggle on
-        cancel_label.empty()
-        process_running = True
-        process_thread = Thread(target=run_process)
-        process_thread.start()
-
 def get_ttf(key: str) -> Optional[any]:
     if key in st.session_state:
         return st.session_state[key]
@@ -66,9 +47,28 @@ def generate_button(prefix, file_input, version, **kwargs):
         key=f"{prefix}-ref_char_ids",
         )
     
+    # For multithreading toggle (prevent function from running too many time)
+    process_running = False
+    process_thread = None
+    cancel_label = st.empty()
+
+    def toggle_process(process_running, process_thread, run_process):
+        if process_running:
+            # Toggle off
+            process_running = False
+            cancel_label.text("Canceled")
+            if process_thread: # Kill Thread
+                process_thread.join()
+        else:
+            # Toggle on
+            cancel_label.empty()
+            process_running = True
+            process_thread = Thread(target=run_process)
+            process_thread.start()
+            
     if st.button("Generate image", key=f"{prefix}-btn") and process_running == False:
         with st.spinner(f"⏳ Generating image (5 minutes per n_sample estimated time)"):
-            image = toggle_process( ttf_to_image(file_input, OUTPUT_IMG_KEY, n_samples, ref_char_ids, version) )
+            image = toggle_process(process_running, process_thread, ttf_to_image(file_input, OUTPUT_IMG_KEY, n_samples, ref_char_ids, version) )
             set_img(OUTPUT_IMG_KEY, image.copy())
         st.image(image)