Spaces:
Sleeping
Sleeping
add showing cancel
Browse files
app.py
CHANGED
|
@@ -10,18 +10,20 @@ SET_IMG_KEY = "set_img"
|
|
| 10 |
OUTPUT_IMG_KEY = "output_img"
|
| 11 |
|
| 12 |
# For multithreading toggle (prevent function from running too many time)
|
| 13 |
-
|
| 14 |
process_thread = None
|
|
|
|
| 15 |
|
| 16 |
def toggle_process(process_running, process_thread, run_process):
|
| 17 |
-
if
|
| 18 |
# Toggle off
|
| 19 |
process_running = False
|
| 20 |
-
|
| 21 |
if process_thread: # Kill Thread
|
| 22 |
process_thread.join()
|
| 23 |
else:
|
| 24 |
# Toggle on
|
|
|
|
| 25 |
process_running = True
|
| 26 |
process_thread = Thread(target=run_process)
|
| 27 |
process_thread.start()
|
|
@@ -63,8 +65,8 @@ def generate_button(prefix, file_input, version, **kwargs):
|
|
| 63 |
value="1,2,3,4,5,6,7,8",
|
| 64 |
key=f"{prefix}-ref_char_ids",
|
| 65 |
)
|
| 66 |
-
|
| 67 |
-
if st.button("Generate image", key=f"{prefix}-btn"):
|
| 68 |
with st.spinner(f"⏳ Generating image (5 minutes per n_sample estimated time)"):
|
| 69 |
image = toggle_process( ttf_to_image(file_input, OUTPUT_IMG_KEY, n_samples, ref_char_ids, version) )
|
| 70 |
set_img(OUTPUT_IMG_KEY, image.copy())
|
|
|
|
| 10 |
OUTPUT_IMG_KEY = "output_img"
|
| 11 |
|
| 12 |
# For multithreading toggle (prevent function from running too many time)
|
| 13 |
+
process_running = False
|
| 14 |
process_thread = None
|
| 15 |
+
cancel_label = st.empty()
|
| 16 |
|
| 17 |
def toggle_process(process_running, process_thread, run_process):
|
| 18 |
+
if process_running:
|
| 19 |
# Toggle off
|
| 20 |
process_running = False
|
| 21 |
+
cancel_label.text("Canceled")
|
| 22 |
if process_thread: # Kill Thread
|
| 23 |
process_thread.join()
|
| 24 |
else:
|
| 25 |
# Toggle on
|
| 26 |
+
cancel_label.empty()
|
| 27 |
process_running = True
|
| 28 |
process_thread = Thread(target=run_process)
|
| 29 |
process_thread.start()
|
|
|
|
| 65 |
value="1,2,3,4,5,6,7,8",
|
| 66 |
key=f"{prefix}-ref_char_ids",
|
| 67 |
)
|
| 68 |
+
|
| 69 |
+
if st.button("Generate image", key=f"{prefix}-btn") and process_running == False:
|
| 70 |
with st.spinner(f"⏳ Generating image (5 minutes per n_sample estimated time)"):
|
| 71 |
image = toggle_process( ttf_to_image(file_input, OUTPUT_IMG_KEY, n_samples, ref_char_ids, version) )
|
| 72 |
set_img(OUTPUT_IMG_KEY, image.copy())
|