Add health check for the virtual tryon model

app.py

@@ -1,8 +1,11 @@
 from __future__ import annotations
 
 import os
+import threading
 from typing import TYPE_CHECKING, Any
 
+from utils import health_check_virtual_try_model
+
 
 IS_HF_ZERO_GPU = os.getenv("SPACE_ID", "").startswith("sitatech/")
 IS_LOCAL = os.getenv("LOCALE_RUN") is not None
@@ -146,6 +149,8 @@ def set_client_for_session(request: gr.Request):
         raise WebRTCError(
             f"Inference server is not available. Status code: {health_check_response.status_code}"
         )
+    
+    threading.Thread(target=health_check_virtual_try_model).start()
 
     if not vibe_shopping_agent.clients_connected:
         vibe_shopping_agent.connect_clients()

mcp_host/ui.py

@@ -67,7 +67,7 @@ def ProductList(products: list[dict[str, str]]):
                 object-fit: contain;
                 border-radius: 8px 8px 0 0;
             " />
-            <div style="width: 100%; padding: 8px; display: flex; flex-direction: column; align-items: center; justify-content: space-between; height: 50px;">
+            <div style="width: 100%; padding: 8px; display: flex; flex-direction: column; align-items: center; justify-content: space-between; height: 52px;">
                 <h5 style="
                     opacity: 0.7;
                     margin: 0 5px;
@@ -90,7 +90,7 @@ def ProductList(products: list[dict[str, str]]):
         display: grid;
         grid-template-columns: repeat(auto-fit, minmax(200px, 1fr));
         gap: 0.8rem;
-        padding: 1rem 0;
+        padding: 0.5rem 0;
         height: 600px;
         width: 100%;
         overflow-y: auto;

utils.py

@@ -1,6 +1,8 @@
 import os
 from io import BytesIO
 from PIL import Image
+from fastrtc import WebRTCError
+import requests
 
 
 class ImageUploader:
@@ -49,3 +51,14 @@ def pil_to_bytes(image, format: str = "PNG") -> bytes:
 def get_hf_space_file_url_prefix() -> str:
     space_host = os.getenv("SPACE_HOST")
     return f"https://{space_host}/gradio_api/file="
+
+
+def health_check_virtual_try_model():
+    try:
+        virtual_try_hc_response = requests.get(
+            "https://sita-berete-3-vibe-shopping--health-check.modal.run"
+        )
+        virtual_try_hc_response.raise_for_status()
+    except Exception as e:
+        print(f"Virtual try-on model health check failed: {e}")
+        raise WebRTCError("Error: Virtual try-on server failed to start")