dynamic port allocation

app.py

@@ -158,41 +158,56 @@ with gr.Blocks(css=custom_css) as interface:
 if __name__ == "__main__":
     interface.launch(share=True)
 """
-
 import gradio as gr
 import netron
 import os
 import threading
+import logging
+import time
 
-# Define the function to visualize the model
-def visualize_model(model_file):
-    # If the model file is a path (file uploaded), read it
-    if isinstance(model_file, str):
-        model_path = model_file
-    else:
-        # If it's a NamedString (new behavior), save it to a temporary file
-        temp_model_path = f"./temp_model.{model_file.name.split('.')[-1]}"
-        with open(temp_model_path, "wb") as f:
-            f.write(model_file.encode())  # Use encode() to write the string content
-        model_path = temp_model_path
+logging.basicConfig(level=logging.DEBUG)
 
-    # Start Netron visualization in a separate thread, specifying a custom port (e.g., 8081)
-    threading.Thread(target=netron.start, args=(model_path, 8081)).start()
+# Store the last Netron thread and port
+last_thread = None
+last_port = None
 
-    # Return an iframe embedding the Netron visualization on the correct port
-    iframe_html = f'<iframe src="http://localhost:8081" width="800" height="600"></iframe>'
-    return iframe_html
+def visualize_model(model_file):
+    global last_thread, last_port
 
-# Create the Gradio interface
+    try:
+        # Generate a temporary file for the model
+        temp_model_path = f"./temp_model.{model_file.name.split('.')[-1]}"
+        with open(temp_model_path, "wb") as f:
+            f.write(model_file.read())
+
+        # Stop previous Netron server if running
+        if last_port:
+            logging.info(f"Stopping Netron on port {last_port}")
+            netron.stop()
+            time.sleep(1)  # Small delay for proper shutdown
+
+        # Start a new Netron server on a dynamic port
+        new_port = 8081 + int(time.time()) % 1000  # Avoid conflicts
+        logging.info(f"Starting Netron on port {new_port}")
+        last_port = new_port
+        last_thread = threading.Thread(target=netron.start, args=(temp_model_path, new_port))
+        last_thread.start()
+
+        # Return iframe with dynamic port
+        iframe_html = f'<iframe src="http://localhost:{new_port}" width="800" height="600"></iframe>'
+        return iframe_html
+
+    except Exception as e:
+        logging.error(f"Error in visualization: {e}")
+        return "An error occurred while visualizing the model."
+
+# Create Gradio interface
 interface = gr.Interface(
     fn=visualize_model,
-    inputs=gr.File(label="Upload Model File"),  # Model file input
-    outputs=gr.HTML(),  # Output is HTML to embed the iframe
+    inputs=gr.File(label="Upload Model File"),  # Input as a model file
+    outputs=gr.HTML(),  # Output the embedded Netron visualization
     live=True
 )
 
-# Launch Gradio interface on localhost:7860
-interface.launch(server_name="0.0.0.0", server_port=7860, share=True)
-
-
-
+# Launch Gradio server on a fixed port
+interface.launch(server_name="0.0.0.0", server_port=7860)