daily_things_maker.py

import random
from pydantic import BaseModel
from openai_adapter import OpenAIAdapter


class Event(BaseModel):
    file_name: str
    place: str
    thing: str


def load_random_events() -> list[Event]:
    event_list = [
        {"file_name": "home_to_school", "place": "家→学校"},
        {"file_name": "school", "place": "学校"},
        {"file_name": "school_to_cram_school", "place": "学校→予備校"},
        {"file_name": "cram_school", "place": "予備校"},
        {"file_name": "cram_school_to_home", "place": "予備校→家"},
        {"file_name": "home", "place": "家"},
    ]
    events = []
    for event in event_list:
        with open(f"storage/daily_things_words/{event['file_name']}.txt", "r", encoding="utf-8") as f:
            random_event = random.choice(f.read().splitlines())
            events.append(
                Event(file_name=event["file_name"], place=event["place"], thing=random_event))
    return events


class DailyThingsMaker:
    def make() -> str:
        events = load_random_events()
        with open("storage/daily_things_make_prompt.txt", "r", encoding="utf-8") as f:
            system_prompt = f.read()

        adapter = OpenAIAdapter()
        daily_things: list[Event] = []
        for event in events:
            res = adapter.create_structured_output([adapter.create_message(
                "system", system_prompt), adapter.create_message("user", event.place+"："+event.thing)], Event)
            daily_things.append(res)
        return "\n".join([f"{event.place}:{event.thing}" for event in daily_things])


if __name__ == "__main__":
    res = DailyThingsMaker.make()
    print(res)