resolved server issue

app.py

@@ -8,6 +8,7 @@ import cv2
 import numpy as np
 from yolov5 import xai_yolov5
 from yolov8 import xai_yolov8s
+import requests
 
 # Sample images directory
 sample_images = {
@@ -18,12 +19,14 @@ sample_images = {
 # Preloaded model file path
 preloaded_model_file = os.path.join(os.getcwd(), "weight_files/yolov5.onnx")  # Example path
 
+# Function to load sample image
 def load_sample_image(sample_name):
     image_path = sample_images.get(sample_name)
     if image_path and os.path.exists(image_path):
         return Image.open(image_path)
     return None
 
+# Function to process the image
 def process_image(sample_choice, uploaded_image, yolo_versions):
     # Use uploaded or sample image
     if uploaded_image is not None:
@@ -46,14 +49,30 @@ def process_image(sample_choice, uploaded_image, yolo_versions):
 
     return result_images
 
+# Start Netron backend
 def start_netron_backend(model_file):
     def serve_netron():
         netron.start(model_file, host='0.0.0.0', port=8080)  # Start Netron on port 8080
-    
+
     # Launch Netron in a separate thread
     threading.Thread(target=serve_netron, daemon=True).start()
-    time.sleep(2)  # Allow server to initialize
 
+    # Wait until Netron server is ready
+    def wait_for_netron(url, timeout=10):
+        start_time = time.time()
+        while time.time() - start_time < timeout:
+            try:
+                response = requests.get(url)
+                if response.status_code == 200:
+                    return True
+            except requests.ConnectionError:
+                time.sleep(0.5)
+        return False
+
+    # Check server readiness
+    wait_for_netron("http://localhost:8080/")
+
+# View Netron model
 def view_netron_model():
     # Ensure model exists
     if not os.path.exists(preloaded_model_file):
@@ -61,7 +80,9 @@ def view_netron_model():
 
     # Start Netron backend
     start_netron_backend(preloaded_model_file)
-    return gr.HTML(f'<iframe src="http://localhost:8080/" width="100%" height="600px"></iframe>')
+    return gr.Markdown(
+        "### [View Model in Netron](http://localhost:8080/) - Click the link if Netron does not load properly."
+    )
 
 # Custom CSS for styling (optional)
 custom_css = """
@@ -117,7 +138,10 @@ with gr.Blocks(css=custom_css) as interface:
             height=500,
         )
 
-        gr.HTML(view_netron_model)
+        # Display Netron link
+        netron_display = gr.Markdown(
+            "### [View Model in Netron](http://localhost:8080/) - Click the link if Netron does not load properly."
+        )
 
     # Sample selection update
     sample_selection.change(
@@ -133,10 +157,6 @@ with gr.Blocks(css=custom_css) as interface:
         outputs=[result_gallery],
     )
 
-    # Netron model visualization
-    # interface.add_component(gr.HTML(view_netron_model))
-    # netron_display.value = view_netron_model()
-
 # Launch Gradio app
 if __name__ == "__main__":
     interface.launch(share=True)