adding check port functionality

vizro-ai/examples/dashboard_ui/_utils.py

@@ -3,6 +3,7 @@
 import base64
 import io
 import logging
+import socket
 
 import pandas as pd
 
@@ -82,3 +83,14 @@ def format_output(generated_code):
     generated_code += "    app.run(port=8051)\n"
 
     return generated_code
+
+
+def check_available_port(port):
+    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sk:
+        return sk.connect_ex(('127.0.0.1', port)) != 0
+
+
+def find_available_port(base_port=8051):
+    while not check_available_port(base_port):
+        base_port += 1
+    return base_port

vizro-ai/examples/dashboard_ui/app.py

@@ -5,7 +5,7 @@
 
 import dash_bootstrap_components as dbc
 import vizro.models as vm
-from _utils import format_output
+from _utils import format_output, find_available_port
 from actions import data_upload_action, display_filename, save_files
 from components import (
     CodeClipboard,
@@ -264,10 +264,11 @@ def show_button(ai_response):
 )
 def run_generated_dashboard(n_clicks):
     """Runs vizro-ai generated dashboard in an iframe window."""
+    port = find_available_port()
     if not n_clicks:
         raise PreventUpdate
     else:
-        subprocess.Popen(["python", "output_files/run_vizro_ai_output.py"])
+        subprocess.Popen(["python", "output_files/run_vizro_ai_output.py", str(port)])
         iframe = html.Iframe(src="http://localhost:8051/", height="600px")
         return True, iframe