from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool

import sys  # Add this import to fix the error
import datetime
import requests
from checks.failed_check import  create_failed_gradio_ui
from checks.endpoint_check import check_public_endpoint

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 Gradio_UI import GradioUI

# Prechecks 

# Get the URI for the endpoint 
endpoint_uri = load_huggingface_model()

# Test the endpoint

status_info = check_public_endpoint(endpoint_uri)

print(status_info["status_code"])
print(status_info["response_data"])
print(status_info["message"])

if status_info["status"] is False:
     create_failed_gradio_ui()
else: 
    
    model = HfApiModel(
        max_tokens=2096,
        temperature=0.5,
        model_id= endpoint_uri,
         custom_role_conversions=None,
     )

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

    agent = CodeAgent(
        model=model,
        tools=combined_tools, 
        max_steps=6,
        verbosity_level=3,
        grammar=None,
        planning_interval=None,
        name="Content Agent",
        description="Evaluates whether text is polite or impolite. ",
        prompt_templates=combined_prompts,
        additional_authorized_imports=["pytz"]
     
    )
    
    # Launch the UI earlier or without passing through the Agent
    GradioUI(agent).launch()