Update app.py

app.py

@@ -80,25 +80,32 @@ def wake_up_endpoint(endpoint_uri, max_wait=30):
 
 def main():
     
-    
+    ui = StatusUI(" Content Agent ")
+    ui.launch()
+     
 # Elsewhere in your code:
 #ui.append("Checking endpoint…")
-#ui.append("✅ Model responded OK.")
+
 #ui.shutdown()   
     
     # Load endpoint and check health
 
     
     # Prechecks 
+    ui.append("Starting prechecks...")
+    ui.append("Checking endpoint..")
+    
     endpoint_uri = load_huggingface_model() # Get the URI for the endpoint 
-   
+    ui.append(endpoint_uri)   
 
     # Wake it up before health check
     wake_up_successful = wake_up_endpoint(endpoint_uri)
+    
     if not wake_up_successful:
-        print("Warning: Could not wake up the endpoint. Exiting.")
-        exit(0)
-
+        ui.append("Warning: Could not wake up the endpoint.  Continuing.")
+    elif     
+        ui.append("✅ End point responded OK.")
+    
     is_healthy, status_info = check_model_endpoint(endpoint_uri) # Test the endpoint
     
     if not is_healthy:
@@ -114,8 +121,5 @@ def main():
     
 
 if __name__ == "__main__":
-    
-    ui = StatusUI("🚦 Endpoint Logger")
-    ui.launch()
 
-    #main()
+    main()