from chatbots.chatbot_base import ChatbotBase
from utils.tools import txt2string
from pathlib import Path
import os

current_dir = Path(__file__).resolve().parent

class Chatbot(ChatbotBase):
    def __init__(self) -> None:
        super().__init__(txt2string(os.path.join(current_dir,'system.txt')))

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