Upload app.py

app.py

@@ -652,11 +652,10 @@ def optimize_prompt(initial_prompt: str, dataset_name: str, dataset_split: str,
         try:
             # Monkey-patch to disable process pool in Gradio environment
             # This prevents "signal only works in main thread" errors
-            import sys
-            import openevolve.controller.parallel_controller as pc_module
+            import openevolve.process_parallel as pp_module
 
             # Save original class
-            OriginalProcessController = pc_module.ProcessParallelController
+            OriginalProcessController = pp_module.ProcessParallelController
 
             # Create a no-op version that doesn't use multiprocessing
             class NoOpProcessController:
@@ -679,7 +678,7 @@ def optimize_prompt(initial_prompt: str, dataset_name: str, dataset_split: str,
                     return [func(*a) for a in zip(*args)]
 
             # Replace the process controller
-            pc_module.ProcessParallelController = NoOpProcessController
+            pp_module.ProcessParallelController = NoOpProcessController
 
             try:
                 # Run evolution
@@ -691,7 +690,7 @@ def optimize_prompt(initial_prompt: str, dataset_name: str, dataset_split: str,
                 )
             finally:
                 # Restore original class
-                pc_module.ProcessParallelController = OriginalProcessController
+                pp_module.ProcessParallelController = OriginalProcessController
 
             progress(0.80, desc="Parsing evolution history...")