File size: 803 Bytes
22abb9f
ce1805e
 
8991658
 
 
 
 
 
ce1805e
8991658
 
 
 
 
 
ce1805e
8991658
 
ce1805e
8991658
 
 
 
ce1805e
8991658
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
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()