from smolagents import DuckDuckGoSearchTool, load_tool,tool

import sys  # Add this import to fix the error
import datetime
import time
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 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):

     # Load prompts
    combined_prompts = load_prompts()
    combined_tools = load_tools() ## from the tools folder (don't remove final answer)

    """Initialize and configure the CodeAgent"""
    model = HfApiModel(
        max_tokens=2096,
        temperature=0.5,
        model_id=endpoint_uri,
        custom_role_conversions=None,
    )
    
    return CodeAgent(
        model=model,
        tools=combined_tools,
        max_steps=6,
        verbosity_level=3,
        name="content_agent",
        description="Evaluates whether text is polite or impolite.",
        prompt_templates=combined_prompts,
        additional_authorized_imports=["pytz"]
    ) 
    
def main():
    uri = resolve_endpoint()
    ok, msg = is_endpoint_healthy(uri)
    if not ok:
        return f"Endpoint not ready: {msg}"

    
    if ok:
        # 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__":
    main()