revert port

app.py

@@ -159,55 +159,28 @@ if __name__ == "__main__":
     interface.launch(share=True)
 """
 import gradio as gr
-import netron
 import os
-import threading
-import logging
-import time
-
-logging.basicConfig(level=logging.DEBUG)
-
-# Store the last Netron thread and port
-last_thread = None
-last_port = None
-
-def visualize_model(model_file):
-    global last_thread, last_port
-
-    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"),  # Input as a model file
-    outputs=gr.HTML(),  # Output the embedded Netron visualization
-    live=True
-)
-
-# Launch Gradio server on a fixed port
-interface.launch(server_name="0.0.0.0", server_port=7860)
+import subprocess
+
+# Path to the model file (change this to your actual model path)
+MODEL_FILE = os.path.join(os.getcwd(), "weight_files/yolov5.onnx") 
+
+# Function to visualize the model using Netron
+def visualize_model():
+    # Serve the model using Netron
+    netron_command = f"netron.start('{MODEL_FILE}', address='0.0.0.0', port=8081)"
+    subprocess.Popen(["python3", "-c", netron_command])
+    # Return the URL where the model is served
+    return f"http://localhost:8081"
+
+# Gradio Interface
+with gr.Blocks() as demo:
+    gr.Markdown("## Model Visualization with Netron")
+    output_link = gr.Textbox(label="Netron Model Link")
+    visualize_btn = gr.Button("Visualize Model")
+
+    # On button click, call the function and display the link
+    visualize_btn.click(fn=visualize_model, inputs=[], outputs=[output_link])
+
+# Launch the Gradio app
+demo.launch(server_name="0.0.0.0", server_port=7860, ssr=False)