from chatbots.chatbot_base import ChatbotBase
from utils.tools import txt2string


class Chatbot(ChatbotBase):
    def __init__(self) -> None:
        super().__init__(txt2string('./chatbots/improvement_bot/system.txt'))

        
    def generate_code(self,code,base64_img):
        self.add_message('Generate improved the source code for a web page that looks exactly like this.\n'+code,base64_img,role='user')
        return self.generate()