import os
from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import yaml
from tools.final_answer import FinalAnswerTool
from tools.visit_webpage import VisitWebpageTool
from tools.web_search import DuckDuckGoSearchTool

from Gradio_UI import GradioUI
import socket

if os.getenv('HF_SPACE') == "HF_SPACE":
    token = os.getenv('HF_READ_First_Agent')
else:
    f = open("hf_token_read.txt")
    d = f.read()
    f.close()
    os.environ["HF_TOKEN"] = d


visit_webpage_tool = VisitWebpageTool()
websearch_tool = DuckDuckGoSearchTool()
final_answer = FinalAnswerTool()

model = HfApiModel(
    max_tokens=1000,
    temperature=0.5,
    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
    custom_role_conversions=None,
)

with open("prompts.yaml", 'r') as stream:
    prompt_templates = yaml.safe_load(stream)
    
agent = CodeAgent(
    model=model,
    tools=[final_answer, websearch_tool, visit_webpage_tool], ## add your tools here (don't remove final answer)
    max_steps=6,
    verbosity_level=1,
    grammar=None,
    planning_interval=None,
    name=None,
    description=None,
    prompt_templates=prompt_templates
)

GradioUI(agent).launch()