LLama Index를 활용한 더 지니어스 게임 AI로 제작 (게임 UI 반영)

캐릭터 얼굴 이미지만 제작
1. 마인크래프트 스타일
2. 32x32 pixel size
3. 캐릭터 성격
    {
        "name": "Hyun-Ah",
        "age": 30,
        "role": "프로 마술사",
        "background": "세계 각지를 돌며 공연을 펼친 유명 마술사. TV 쇼에서도 활약 중이며, 사람의 시선을 교묘히 조작하는 데 능숙함.",
        "traits": [
            "유머러스하고 외향적임",
            "타인을 쉽게 매료시킴",
            "재치 있고 순발력이 뛰어남",
            "창의적이고 혁신적인 사고방식",
            "강한 직관력"
        ],
        "strengths": [
            "상대의 주의를 분산시키는 능력",
            "의도적으로 혼란을 유발하는 플레이",
            "눈속임과 기만 전술에 능함",
            "예측 불가능한 행동으로 상대를 혼란스럽게 함"
        ],
        "weaknesses": [
            "때로는 과도한 쇼맨십으로 진실성을 의심받음",
            "장기적인 계획보다는 즉흥적인 행동을 선호함"
        ],
        "favorite_quote": "불가능은 단지 환상일 뿐이다.",
        "hidden_talent": "뛰어난 픽포켓 기술 보유",
        "strategies": [
            {
              "name": "환영 카드",
              "description": "상대방을 혼란시키기 위해 거짓으로 제시하는 카드"
            },
            {
              "name": "목매듭 승점",
              "description": "예상치 못한 방법으로 획득한 승점"
            }
          ]        
    },

class GameData:
    def __init__(self):
        if not os.path.exists("./game_data/storage"):
            self.documents = SimpleDirectoryReader("./game_data/game_scenario").load_data()
            self.index = VectorStoreIndex.from_documents(self.documents)
            self.index.storage_context.persist()    
        else:
            self.storage_context = StorageContext.from_defaults(persist_dir="./game_data/storage")
            self.index = load_index_from_storage(self.storage_context)        
        self.query_engine = self.index.as_query_engine()

For example                
- 20대 남성
사회 초년생: 20대 남성은 대부분 대학을 졸업하고 막 사회에 진출한 신입사원으로 인식됩니다. 취업 준비나 군복무 경험을 바탕으로 성장을 시작하는 시기입니다.
자유를 중시: 대학 생활이나 사회생활 초기의 자유를 중요시하며, 여행, 취미생활, 자기 계발에 관심이 많습니다.
현실적 고민: 많은 20대 남성들이 경제적 부담, 취업난, 미래의 불안정성 등 현실적인 고민을 가지고 있습니다.
- 20대 여성
자기 계발: 20대 여성은 자기 계발을 중요시하며, 학업, 직업, 외모 관리에 많은 시간을 투자하는 시기로 인식됩니다.
독립과 자유: 여성들도 대학 생활에서 독립을 경험하고 자유로운 생활을 즐기며, 여행과 취미 활동에 대한 관심이 큽니다.
커리어 지향: 20대 여성 중 많은 이들이 자신의 커리어를 확립하기 위해 노력하며, 사회적 성취에 대한 열망이 큽니다.
- 30대 남성
안정적 직업: 30대 남성은 사회적으로 안정된 직장을 가진 가장으로 인식되며, 직업적 성공과 경제적 안정을 추구하는 이미지가 강합니다.
가정 중심: 결혼과 자녀 양육을 시작하는 시기인 만큼, 가정을 꾸리고 경제적 책임을 지는 가장으로서의 이미지가 있습니다.
사회적 성공 압박: 30대 남성은 커리어에서 성공을 이루어야 한다는 압박을 받기도 하며, 승진이나 자산 축적에 대한 욕구가 큽니다.
- 30대 여성
커리어와 가정의 균형: 30대 여성은 커리어와 가정 사이에서 균형을 잡으려는 경향이 있으며, 직장에서의 성취와 가정에서의 역할을 동시에 이루려는 이미지가 있습니다.
결혼과 육아: 30대 중반부터 결혼과 출산을 경험하는 여성들이 많으며, 이에 따라 일과 가정의 조화를 이뤄야 하는 부담이 있습니다.
자아 실현: 여전히 자기 계발에 대한 욕구가 강하며, 직업적으로도 더 많은 성장을 이루려는 경향이 있습니다.
- 40대 남성
성공의 상징: 40대 남성은 직업적으로 높은 위치에 올라 있으며, 경제적 안정성을 갖춘 성공적인 인물로 인식되곤 합니다.
가정의 기둥: 가정을 이끄는 역할이 강조되며, 자녀 교육과 가정의 경제적 안정에 대해 많은 책임을 느끼는 시기입니다.
건강 관리: 40대가 되면 자신의 건강에 대한 관심이 증가하고, 운동이나 건강관리에 시간을 투자하는 이미지도 있습니다.
- 40대 여성
가정 관리: 40대 여성은 자녀 양육에 많은 시간을 보내며, 가정의 중심 역할을 담당하는 이미지가 강합니다.
직업적 성취: 경력이 있는 40대 여성들은 직장에서 중간 관리자급으로 성장해 있으며, 직업적 성취를 이어가려는 이미지도 있습니다.
사회적 기여: 40대 여성들은 사회나 커뮤니티 활동에 참여하여 자원봉사, 사회 기여 활동을 활발히 하는 경향이 있습니다.

class Room:
    def add_characters(self, character_names):
        self.characters = []
        for name in character_names:
            x = random.randint(self.rect.left, self.rect.right - 32)
            y = random.randint(self.rect.top, self.rect.bottom - 32)
            self.characters.append((x, y, name, character_images[name]))  # 이름도 저장

    def generate_dialogue(self, ai_agent, game_members_info, main_match_description):
        if len(self.characters) == 2:
            _, _, char1_name, _ = self.characters[0]  # 캐릭터 1의 이름
            _, _, char2_name, _ = self.characters[1]  # 캐릭터 2의 이름
            char_info = {char1_name: game_members_info[char1_name], char2_name: game_members_info[char2_name]}
            
            # 대화 생성
            dialogue_gen = ai_agent.ai_generate_dialogue(char_info, main_match_description)
            full_dialogue = ''.join(token for token in dialogue_gen)

            # 대화를 캐릭터별로 분리 (예시: "**캐릭터명**: 대화내용")
            dialogue_lines = full_dialogue.split("**")[1:]  # "**"로 시작하는 대화 부분만 추출
            dialogue_dict = {}

            for i in range(0, len(dialogue_lines), 2):
                    character_name = dialogue_lines[i].strip(":")  # 캐릭터 이름 추출
                    character_dialogue = dialogue_lines[i + 1].strip()  # 대화 내용 추출
                    
                    # 대화가 이미 존재하는 경우 리스트에 추가, 처음인 경우 리스트 생성
                    if character_name not in dialogue_dict:
                        dialogue_dict[character_name] = []
                    
                    dialogue_dict[character_name].append(character_dialogue)
                    # JSON 저장을 위한 형식 변환
            dialogue_data = {
                "room": self.name,
                "dialogue": [
                    {
                        "character": char_name,
                        "text": char_dialogue
                    }
                    for char_name, char_dialogue in dialogue_dict.items()
                ]
            }

            # 대화 저장 (방 이름과 대화 내용을 저장)
            self.save_dialogue_to_json(dialogue_data, self.name)


            # 현재 대화 내용 설정
            self.dialogue = full_dialogue  

class Game:
    def run(self):
        running = True
        while running:
            for event in pygame.event.get():
                if event.type == pygame.QUIT:
                    running = False
                elif event.type == pygame.MOUSEBUTTONDOWN:
                    self.handle_mouse_click(event.pos)
                elif event.type == pygame.MOUSEWHEEL:
                    # 마우스 휠 이벤트로 스크롤 오프셋 조절
                    self.scroll_offset += event.y * 10  # 마우스 휠 값에 따라 스크롤 이동 속도 조절
                    self.scroll_offset = max(self.scroll_offset, 0)  # 최상단으로 스크롤 제한
                    

            self.draw()
            pygame.display.flip()

        pygame.quit()