concurrent execution

app.py

@@ -1,14 +1,12 @@
 import netron
-import threading
 import gradio as gr
 import os
+import threading
 from PIL import Image
 import cv2
 import numpy as np
 from yolov5 import xai_yolov5
 from yolov8 import xai_yolov8s
-import time
-import tempfile
 
 # Sample images directory
 sample_images = {
@@ -23,7 +21,7 @@ def load_sample_image(sample_name):
         return Image.open(image_path)
     return None
 
-def process_image(sample_choice, uploaded_image, yolo_versions):
+def process_image(sample_choice, uploaded_image, yolo_versions, result_callback):
     """Process the image using selected YOLO models."""
     if uploaded_image is not None:
         image = uploaded_image  # Use the uploaded image
@@ -42,9 +40,9 @@ def process_image(sample_choice, uploaded_image, yolo_versions):
         else:
             result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))
 
-    return result_images
+    result_callback(result_images)
 
-def view_model(selected_models):
+def view_model(selected_models, netron_callback):
     """Generate Netron visualization for the selected models."""
     for model in selected_models:
         if model == "yolov5":
@@ -56,8 +54,9 @@ def view_model(selected_models):
                 frameborder="0">
             </iframe>
             """
-            return iframe_html
-    return "<p>Please select a valid model for Netron visualization.</p>"
+            netron_callback(iframe_html)
+            return
+    netron_callback("<p>Please select a valid model for Netron visualization.</p>")
 
 # Custom CSS for styling (optional)
 custom_css = """
@@ -122,16 +121,35 @@ with gr.Blocks(css=custom_css) as interface:
         outputs=sample_display,
     )
 
-    # Process image and display results, also trigger Netron visualization when run button is clicked
+    # Parallel execution with threading
+    def run_both(sample_choice, uploaded_image, selected_models):
+        results = []
+        netron_html = ""
+
+        def update_results(res):
+            nonlocal results
+            results = res
+            result_gallery.update(res)
+
+        def update_netron(html):
+            nonlocal netron_html
+            netron_html = html
+            netron_display.update(html)
+
+        # Run both functions in parallel
+        thread1 = threading.Thread(target=process_image, args=(sample_choice, uploaded_image, selected_models, update_results))
+        thread2 = threading.Thread(target=view_model, args=(selected_models, update_netron))
+        thread1.start()
+        thread2.start()
+        thread1.join()
+        thread2.join()
+
     run_button.click(
-        fn=lambda sample_choice, uploaded_image, yolo_versions: [
-            process_image(sample_choice, uploaded_image, yolo_versions),  # Process image
-            view_model(yolo_versions)  # Display model visualization
-        ],
+        fn=run_both,
         inputs=[sample_selection, upload_image, selected_models],
-        outputs=[result_gallery, netron_display],
+        outputs=[],
     )
 
 # Launching Gradio app
 if __name__ == "__main__":
-    interface.launch(share=True)
+    interface.launch(share=True)