app.py

"""
Chatbot Nuevo Régimen Académico
"""
from chatbot.ui import ChatbotInterface
from chatbot.llm import GeminiAI
from langchain.globals import set_verbose, set_debug


def respond(message, history):
    prompt = llm.getMainTemplate()
    chain = prompt | llm.llm
    response = chain.invoke({"message": message, "history": history})
    return response.content


if __name__ == "__main__":
    set_verbose(True)
    set_debug(True)
    llm = GeminiAI("gemini-1.5-flash")
    ui = ChatbotInterface(respond)

    ui.app.launch()