import os
import asyncio
from image_processor3 import ImageProcessor3
from test_papago_translator import PapagoTranslator
from test_loggerModule import Logger1

class SimpleLogger:
    def log(self, msg, level=None, exc_info=None):
        print(msg)

async def main():
    logger = SimpleLogger()
    base_dir = os.path.dirname(os.path.abspath(__file__))
    file_num = 6
    img_path = os.path.join(base_dir, 'img', f"{file_num}.jpg")

    # ImageProcessor3 인스턴스 생성
    toggle_states = {
        'TEMP_IMAGE_DIR': os.path.join(base_dir, 'img'),
        #'image_font_path': os.path.join(base_dir, 'fonts', 'HakgyoansimDunggeunmisoTTFB.ttf'),
        #'image_font_path': os.path.join(base_dir, 'fonts', 'NanumBarunGothic.ttf'),
        #'image_font_path': os.path.join(base_dir, 'fonts', 'NanumSquareRoundR.ttf'),
        #'image_font_path': os.path.join(base_dir, 'fonts', 'gamtanload.ttf'),
        'image_font_path': os.path.join(base_dir, 'fonts', 'Cafe24Ohsquare-v2.0.ttf'),
        
        'ocr': True,
        'membership_level': 'vip',
        'request_inpainting_server_url': 'http://192.168.0.150:35756'
    }
    unwanted_words = {
        "고품질": "저품질",
        "세탁기": "세세탁기"
    }

    papago_translator = PapagoTranslator(logger)
    logger = Logger1()
    image_processor = ImageProcessor3(
        logger=logger,
        page=None,
        toggle_states=toggle_states,
        unwanted_words=unwanted_words,
        base_dir=base_dir,
        papago_translator=papago_translator,
        authenticated_by_admin=True,
    )

    image_processor.update_unwanted_texts(unwanted_words)

    # process_single_image 호출
    result = await image_processor.process_single_image(
        page=None, original_image_url=img_path, delay=0, index=file_num-1, file_prefix=""
    )

    #result = await image_processor.remove_background(page=None, original_image_url=img_path, index=file_num, file_prefix="")

    print(f"처리 결과: {result}")

if __name__ == '__main__':
    asyncio.run(main())