import os 
import gradio as gr 
from src.chains import build_chain
import langchain

css = """
    footer {visibility: hidden}
    """

js = """ 
    """


class Agent:
    def __init__(self):
        self.agent = None 
        self.data = None 

agent = Agent()
agent.agent = build_chain()

def answer_query(message, history):
    message = agent.agent.run(message)
    return message