import core.contentGen as contentGen
import core.posterGen as posterGen
import core.simple_collage as simple_collage
import json
class poster_gen_config:
    def __init__(self, json_file):
        self.json_file = json_file

    def get_info_directory(self):
        return self.json_file['info_directory']

    def get_poster_num(self):
        return self.json_file['poster_num']

    def get_input_dir(self):
        return self.json_file['input_dir']

    def get_target_size(self):
        return self.json_file['target_size']
    
    def get_output_dir(self):
        return self.json_file['output_dir']
    
    def get_tweet_content(self):
        return self.json_file['tweet_content']

def generate_poster(poster_gen_config):
    info_directory = poster_gen_config.get_info_directory()
    poster_num = poster_gen_config.get_poster_num()
    input_dir = poster_gen_config.get_input_dir()
    target_size = poster_gen_config.get_target_size()
    output_dir = poster_gen_config.get_output_dir()
    tweet_content = poster_gen_config.get_tweet_content()
    content_gen = contentGen.ContentGenerator()
    response = content_gen.run(info_directory, poster_num, tweet_content)
    # print(response)
    img_list = simple_collage.process_directory(input_dir, target_size=target_size, output_count=poster_num, output_dir=output_dir)
    poster_gen = posterGen.PosterGenerator()
    poster_config = posterGen.PosterConfig(response)
    result_path = []
    for index, item in enumerate(poster_config.get_config()):
        text_data = {
                    "title": f"{item['main_title']}",
                    "subtitle": "",
                    "additional_texts": [
                    {"text": f"{item['texts'][0]}", "position": "bottom", "size_factor": 0.5},
                    {"text": f"{item['texts'][1]}", "position": "bottom", "size_factor": 0.5}
                            ]
        }
        img_path = img_list[index]['path']
        print(img_path)
        result_path.append(poster_gen.create_poster(img_path, text_data, f"{index}.jpg"))
    return result_path

def main():
    json_file_path = "/root/autodl-tmp/TravelContentCreator/poster_gen_config.json"
    with open(json_file_path, 'r') as f:
        json_file = json.load(f)
    poster_config = poster_gen_config(json_file)
    generate_poster(poster_config)
    # info_directory = [
    #     "/root/autodl-tmp/sanming_img/相机/甘露寺/description.txt"
    # ]
    # poster_num = 1
    # tweet_content = """
 
    # """
    # input_dir = "/root/autodl-tmp/sanming_img/modify/甘露寺"
    # output_dir = "/root/autodl-tmp/poster_generate_result"
    # target_size = (900, 1200)
    # result_path = []
    
    
    # content_gen = contentGen.ContentGenerator()
    # response = content_gen.run(info_directory, poster_num, tweet_content)
    # print(response)
    # img_list = simple_collage.process_directory(input_dir, target_size=target_size, output_count=poster_num, output_dir=output_dir)
    # print(img_list)
    # poster_gen = posterGen.PosterGenerator()
    # poster_config = posterGen.PosterConfig(response)
    # # config_path = "/root/autodl-tmp/poster_generate_result/2025-04-17_09-31-03.json"
    # # poster_config = posterGen.PosterConfig(config_path)
    # for index, item in enumerate(poster_config.get_config()):
    #     text_data = {
    #                 "title": f"{item['main_title']}",
    #                 "subtitle": "",
    #                 "additional_texts": [
    #                 {"text": f"{item['texts'][0]}", "position": "bottom", "size_factor": 0.5},
    #                 {"text": f"{item['texts'][1]}", "position": "bottom", "size_factor": 0.5}
    #                         ]
    #     }
    #     img_path = img_list[index]['path']
    #     print(img_path)
    #     result_path.append(poster_gen.create_poster(img_path, text_data, f"{index}.jpg"))
        
if __name__ == "__main__":
    main()