from chatbot import ChatBotManager
import gradio as gr

class Gradio_App():

    def __init__(self):
        self.server_name = "0.0.0.0"
        self.server_port = 7860
        self.chatbot_manager = ChatBotManager()

    def build_ui(self):
        with gr.Blocks() as self.demo:
            dropdown = gr.Dropdown(self.chatbot_manager.get_available_model_types(), label='Model type')
            chat_iface = gr.ChatInterface(self.chatbot_manager.obtain_answer, \
                                          additional_inputs=[dropdown], \
                                          type="messages")

    def run(self):
        self.demo.launch(server_name=self.server_name, server_port=self.server_port)

def main():
    gradio_app = Gradio_App()
    gradio_app.build_ui()
    gradio_app.run()

main()