Update app.py

app.py

@@ -435,12 +435,19 @@ with gr.Blocks() as interface:
                     return False, None
         
                 # Stop flag check
+                # if stop_value is not None and stop_value.value:
+                #     p.terminate()
+                #     p.join()
+                #     print("🛑 Stop flag detected — function killed early.")
+                #     return False, None
                 if stop_value is not None and stop_value.value:
-                    p.terminate()
-                    p.join()
-                    print("🛑 Stop flag detected — function killed early.")
+                    print("🛑 Stop flag detected — waiting for child to exit gracefully.")
+                    p.join(timeout=3)  # short wait for graceful exit
+                    if p.is_alive():
+                        print("⚠️ Child still alive, forcing termination.")
+                        p.terminate()
+                        p.join()
                     return False, None
-        
                 time.sleep(0.1)  # avoid busy waiting
         
             # Process finished naturally