# app_agent_config.py
import streamlit as st
from tool_loader import ToolLoader
from tool_config import tool_names
from logger import log_enabled

from PIL import Image 
import numpy as np


class AgentConfig:
    def __init__(self):
        self.tool_checkboxes = []
        self.url_endpoint = ""
        self.image = []
        self.document = ""
        self.log_enabled = False
        self.tool_loader = ToolLoader(tool_names)

    def configure(self):
        st.markdown("Change the agent's configuration here.")
        self.url_endpoint = st.selectbox("Select Inference URL", [
            "https://api-inference.huggingface.co/models/bigcode/starcoder",
            "https://api-inference.huggingface.co/models/OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5",
            "https://api-inference.huggingface.co/models/gpt2"
        ])
        tool_loader = ToolLoader(tool_names)
        self.log_enabled = st.checkbox("Enable Logging")
        self.tool_checkboxes = [st.checkbox(f"{tool.name} --- {tool.description} ") for tool in tool_loader.tools]