Spaces:
Running
Running
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() | |