Update app.py

app.py

@@ -164,28 +164,36 @@ import os
 import threading
 import time
 
-# Function to start Netron server
-def start_netron_server(model_file, port=8080):
+# Function to start Netron server and get its port
+def start_netron_server(model_file):
     if not os.path.exists(model_file):
         return "Error: Model file does not exist."
     
     # Start Netron in a thread
-    thread = threading.Thread(target=netron.start, args=(model_file,), kwargs={"port": port, "host": "0.0.0.0"}, daemon=True)
+    def run_netron():
+        netron.start(model_file, browse=False)  # Prevent opening in a local browser
+    
+    thread = threading.Thread(target=run_netron, daemon=True)
     thread.start()
     
-    # Wait for the server to initialize
+    # Wait for the server to start
     time.sleep(2)
+    
+    # Netron typically uses port 8080 by default, but you may need to confirm this dynamically
+    port = 8080
+    return port
 
-# Function to return the Netron viewer iframe
+# Gradio function to visualize the model
 def visualize_model(file):
     if file:
         file_path = file.name
-        port = 8080  # Port for Netron server
-        start_netron_server(file_path, port)
+        port = start_netron_server(file_path)
         
+        # Construct the iframe URL using Hugging Face's public domain
+        space_url = os.getenv("SPACE_ID", "your-space-id")  # Replace 'your-space-id' with actual ID if testing locally
         iframe_url = f"https://bhumikamak-neuralvista.hf.space:{port}/"
         
-        # Embed the Netron viewer using iframe
+        # Embed the iframe into the Gradio app
         iframe_html = f"""
         <iframe 
             src="{iframe_url}"