fix: find free frontend port, stop containers with the username

refactor: app -> assign_availale_gpu (comments added)

Author: FerTV

nebula/controller.py

@@ -366,7 +366,7 @@ def __init__(self, args):
             self.controller_port = SocketUtils.find_free_port()
 
         if not SocketUtils.is_port_open(self.frontend_port):
-            self.frontend_port = SocketUtils.find_free_port()
+            self.frontend_port = SocketUtils.find_free_port(self.controller_port + 1)
 
         if not SocketUtils.is_port_open(self.statistics_port):
             self.statistics_port = SocketUtils.find_free_port(self.frontend_port + 1)
@@ -716,8 +716,7 @@ def stop_waf():
     @staticmethod
     def stop():
         logging.info("Closing NEBULA (exiting from components)... Please wait")
-        DockerUtils.remove_containers_by_prefix(f"{os.environ['USER']}_nebula")
-        DockerUtils.remove_containers_by_prefix(f"nebula_")
+        DockerUtils.remove_containers_by_prefix(f"{os.environ['USER']}_")
         ScenarioManagement.stop_blockchain()
         ScenarioManagement.stop_participants()
         Controller.stop_waf()

nebula/frontend/app.py

@@ -1240,13 +1240,15 @@ async def assign_available_gpu(scenario_data, role):
         # Obtain currently used gpus
         if running_scenarios:
             running_gpus = []
+            # Obtain associated gpus of the running scenarios
             for scenario in running_scenarios:
                 scenario_gpus = json.loads(scenario["gpu_id"])
+                # Obtain the list of gpus in use without duplicates
                 for gpu in scenario_gpus:
                     if gpu not in running_gpus:
                         running_gpus.append(gpu)
 
-            # Obtain gpus that are not in use
+            # Add available system gpus if they are not in use
             for gpu in available_system_gpus:
                 if gpu not in running_gpus:
                     available_gpus.append(gpu)