import os
from logger import log_response
from custom_agent import CustomHfAgent 
image = []
def handle_submission(user_message, selected_tools, url_endpoint, document, image):
    
    log_response("User input \n {}".format(user_message))
    log_response("selected_tools \n {}".format(selected_tools))
    log_response("url_endpoint \n {}".format(url_endpoint))
    log_response("url_endpoint \n {}".format(document))
    log_response("url_endpoint \n {}".format(image))
    image = image
    
    agent = CustomHfAgent(
        url_endpoint=url_endpoint,
        token=os.environ['HF_token'],
        additional_tools=selected_tools,
        input_params={"max_new_tokens": 192},
    )

    response = agent.chat(user_message,document=document,image=image)

    log_response("Agent Response\n {}".format(response))

    return response