Update app.py

app.py

@@ -1,14 +1,14 @@
-from smolagents import DuckDuckGoSearchTool, load_tool,tool
+# from smolagents import DuckDuckGoSearchTool, load_tool,tool
 
-import sys  # Add this import to fix the error
-import datetime
+import sys   
+# import datetime
 import time
-import requests
+#import requests
 import os
 import pytz # Had to give it permission in Code agent  
 
 from ui.contentagentui import ContentAgentUI
-from ui.gradioui import GradioUI
+# from ui.gradioui import GradioUI
 
 from smolagents import CodeAgent, HfApiModel
 from status_check import resolve_endpoint, is_endpoint_healthy 
@@ -43,18 +43,23 @@ def initialize_agent(endpoint_uri: str):
     
 def main():
     uri = resolve_endpoint()
+    if not uri:
+        print("❌ Endpoint not ready: no URI configured.")
+        sys.exit(1)   # exit with non-zero status for clarity
     ok, msg = is_endpoint_healthy(uri)
     if not ok:
-        print f"Endpoint not ready: {msg}"
-
-    else:
-        # Initialize and run the agent
-        #agent = initialize_agent(endpoint_uri)
-        # Create an instance of the ContentAgentUI class
-        #ui = ContentAgentUI()
-        #ui.pass_through_agent(agent)  # Pass through the agent
-        pass 
+        print("Endpoint not ready: {msg}")
+        
+    #  We want to create a basic GUI calling the ContentAgentUI.. 
+    #  The default flow should be to show the "Wake Endpoint" button
+    #  If the end point is ready, don't show it.
+    # Initialize and run the agent
+    #agent = initialize_agent(endpoint_uri)
+    # Create an instance of the ContentAgentUI class
+    #ui = ContentAgentUI()
+    #ui.pass_through_agent(agent)  # Pass through the agent
 
+    
 if __name__ == "__main__":
     main()