Update app.py

app.py

@@ -4,25 +4,19 @@ import sys  # Add this import to fix the error
 import datetime
 import time
 import requests
-
-from ui.statusui import StatusUI
-from ui.contentagentui import ContentAgentUI
-from ui.gradioui import GradioUI, create_failed_gradio_ui
-
-
-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
-
 import os
 import pytz # Had to give it permission in Code agent  
 
-from tools.tools import load_tools
-
 from agents.model import load_huggingface_model
 from agents.prompts import load_prompts
 
-#from ContentGradio import ContentAgentUI
+from ui.contentagentui import ContentAgentUI
+from ui.gradioui import GradioUI, create_failed_gradio_ui
+
+from smolagents import CodeAgent, HfApiModel
+from status_check import run_status_checks
+from tools.tools import load_tools
+from agents.prompts import load_prompts
    
 def initialize_agent(endpoint_uri: str):
 
@@ -50,81 +44,16 @@ def initialize_agent(endpoint_uri: str):
         additional_authorized_imports=["pytz"]
     ) 
     
-
-def wake_up_endpoint(endpoint_uri, ui, max_wait=300):
-    """Poll the endpoint until it responds OK or timeout."""
-
-    
-    # Enforce minimum wait time of 60 seconds
-    max_wait = max(max_wait, 60)
-    # Get token from environment variable or secrets store
-    hf_token = os.environ.get("HF_TOKEN")  # Or replace with your secret loading logic
-    headers = {"Authorization": f"Bearer {hf_token}"} if hf_token else {}
-    
-    payload = {"inputs": "ping"}  # Adjust to minimal valid input
-    start = time.time()
-
-    while time.time() - start < max_wait:
-        try:
-            ui.append("Pinging endpoint...")
-            ui.append(endpoint_uri)
-            response = requests.post(endpoint_uri, headers=headers, json=payload, timeout=5)
-            
-            if response.ok:
-                ui.append("✅ Endpoint is awake.")
-                return True
-
-            # If it's a 503 or 504, it's likely still warming up
-            ui.append(f"Response: {response.status_code}, retrying...")
-        except requests.exceptions.RequestException as e:
-            ui.append(f"Exception: {e}, retrying...")
-
-        time.sleep(3)  # fixed short backoff; can be exponential if needed
-
-    ui.append("❌ Timed out waiting for endpoint to wake up.")
-    return False
-
 def main():
-    
-    ui = StatusUI(" Content Agent ")
-    ui.launch()
-     
-# Elsewhere in your code:
-#ui.append("Checking endpoint…")
-
-#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, ui)
-    
-    if not wake_up_successful:
-        ui.append("Warning: Could not wake up the endpoint.  Continuing.")
-    else:     
-        ui.append("✅ End point responded OK.")
-    
-    is_healthy, status_info = check_model_endpoint(endpoint_uri) # Test the endpoint
-    
-    if not is_healthy:
-        interface = create_failed_gradio_ui(status_info)
-        interface.launch(show_error=True, share=True)
-        return
-    
-    # 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
-    
+    endpoint_uri = run_status_checks()
+    
+    if endpoint_uri:
+        # 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 
 
 if __name__ == "__main__":