Update app.py

app.py

@@ -5,6 +5,8 @@ import datetime
 import time
 import requests
 
+from checks.check_gui import StatusUI
+
 from checks.failed_check import  create_failed_gradio_ui
 from checks.endpoint_check import check_public_endpoint
 from checks.health_check import check_model_endpoint, should_launch_ui
@@ -44,6 +46,7 @@ def initialize_agent(endpoint_uri: str):
         prompt_templates=combined_prompts(),
         additional_authorized_imports=["pytz"]
     ) 
+    
 
 def wake_up_endpoint(endpoint_uri, max_wait=30):
     """Poll the endpoint until it responds OK or timeout."""
@@ -76,8 +79,19 @@ def wake_up_endpoint(endpoint_uri, max_wait=30):
     return False
 
 def main():
+    
+    
+    ui = StatusUI("🚦 Endpoint Logger")
+    ui.launch()
+
+# Elsewhere in your code:
+#ui.append("Checking endpoint…")
+#ui.append("✅ Model responded OK.")
+#ui.shutdown()   
+    
     # Load endpoint and check health
 
+    
     # Prechecks 
     endpoint_uri = load_huggingface_model() # Get the URI for the endpoint