model.py

import gradio as gr
class Model:
    def __init__(self,
            name:str="Model",
            placeholder:str="Input"):
        self.name = name
        self.placeholder = placeholder
        self.SPETIAL_TOKENS = ["#화자#", "#청자#", "#(남자)청자#", "#(남자)화자#", "#(여자)청자#", "(여자)화자"]
    
    def generate(self, inputs:str) -> str:
        outputs = inputs
        return outputs
    
    def chat(self, message, chat_history):
        response = self.generate(message)
        for special_token in self.SPETIAL_TOKENS:
            response = response.replace(special_token, '너')
        chat_history.append((message, response))
        return "", gr.Chatbot(chat_history, label=self.name, bubble_full_width=False)