test run

app.py

@@ -161,20 +161,40 @@ if __name__ == "__main__":
 
 import gradio as gr
 import netron
+import os
+import threading
+from flask import Flask, send_from_directory
+
+# Flask app to serve the Netron model visualization
+app = Flask(__name__)
+
+def start_netron(model_path):
+    # Start a new thread to serve the Netron visualization on a local server
+    netron.start(model_path, port=8080)
 
-# Define a function to visualize the model
+# Define a function to visualize the model and generate an iframe
 def visualize_model(model_file):
-    # Launch the Netron viewer in a web browser
-    netron.start(model_file)
-    return "Model visualization launched in browser."
+    # Save the model file temporarily
+    temp_model_path = os.path.join(os.getcwd(), "weight_files/yolov5.onnx")
+    with open(temp_model_path, "wb") as f:
+        f.write(model_file.read())
 
-# Create a Gradio interface
+    # Start Netron in a separate thread
+    threading.Thread(target=start_netron, args=(temp_model_path,)).start()
+
+    # Return an iframe embedding the Netron visualization
+    iframe_html = f'<iframe src="http://localhost:8080" width="800" height="600"></iframe>'
+    return iframe_html
+
+# Gradio interface to upload the model and show the visualization
 interface = gr.Interface(
     fn=visualize_model,
-    inputs=gr.File(label="Upload Model File"),  # Model file input
-    outputs="text",  # Just a text output to indicate launch
+    inputs=gr.File(label="Upload Model File"),
+    outputs=gr.HTML(),  # Output is HTML to embed iframe
     live=True
 )
 
+# Start the Gradio interface
 interface.launch()
 
+