MCP활용 강사정보 추출하기

@mcp.tool()
def get_coachInfo_csv_to_md(name: str) -> str:
    """
    주어진 코치 이름에 해당하는 정보를 md 파일로 추출하여 다운로드할 수 있도록 하는 함수.
    
    :param name: 코치의 이름
    :return: md 파일의 경로 또는 오류 메시지
    """
    import os
    import csv
    
    try:
        file_path = os.path.join(os.path.dirname(__file__), 'data', 'coachInfo.csv')
        
        # CSV 파일을 읽고 코치 정보를 찾기
        with open(file_path, 'r', encoding='utf-8') as file:
            csv_reader = csv.reader(file, delimiter='\t')
            
            # 첫 줄은 헤더일 수 있으므로 확인용 변수
            header_checked = False
            indices = {}
            
            for row in csv_reader:
                # 첫 줄이 헤더인 경우 인덱스 설정
                if not header_checked:
                    header_checked = True
                    headers = row
                    indices = {
                        'name': headers.index('name'),
                        'birth': headers.index('birth'),
                        'gender': headers.index('gender'),
                        'phone': headers.index('phone'),
                        'email': headers.index('email'),
                        'coachFrom': headers.index('coachFrom'),
                        'qualificationKCA': headers.index('qualificationKCA'),
                        'qualificationICF': headers.index('qualificationICF'),
                        '2024Activity': headers.index('2024Activity'),
                        'coach': headers.index('coach')
                    }
                    continue
                
                # 코치 이름이 일치하는지 확인
                if row[indices['name']] == name:
                    # 코칭 자격 정보 처리
                    qualification_kca = row[indices['qualificationKCA']] if row[indices['qualificationKCA']] else ""
                    qualification_icf = row[indices['qualificationICF']] if row[indices['qualificationICF']] else ""
                    
                    # 2024 활동 내역 처리
                    activity_2024 = row[indices['2024Activity']].strip() if row[indices['2024Activity']].strip() else "해당사항 없음"
                    activity_2024_lines = "\n".join([f"- {line.strip()}" for line in activity_2024.split('\n')])
                    
                    # 전체 활동 내역 처리
                    coach = row[indices['coach']].strip() if row[indices['coach']].strip() else "해당사항 없음"
                    coach_lines = "\n".join([f"- {line.strip()}" for line in coach.split('\n')])
                    
                    # md 파일 생성
                    md_content = f"## {row[indices['name']]} ( {row[indices['birth']]}, {row[indices['gender']]} )\n"
                    md_content += f"- 전화번호 : {row[indices['phone']]}\n"
                    md_content += f"- email : {row[indices['email']]}\n"
                    md_content += f"- 기본정보 : {row[indices['coachFrom']]}\n"
                    md_content += f"- 코칭자격 : {qualification_kca}. {qualification_icf}\n\n"
                    md_content += "## 자사 계열사 코칭활동\n"
                    md_content += "#### 2024년 활동내역\n"
                    md_content += f"{activity_2024_lines}\n"
                    md_content += "#### 전체 활동내역\n"
                    md_content += f"{coach_lines}\n"
                    
                    md_file_path = os.path.join(os.path.dirname(__file__), f"{name}.md")
                    with open(md_file_path, 'w', encoding='utf-8') as md_file:
                        md_file.write(md_content)
                    
                    return f"MD 파일이 생성되었습니다: {md_file_path}"
        
        # 코치 이름이 발견되지 않으면 메시지 반환
        return f"코치 '{name}'을(를) 찾을 수 없습니다."
        
    except FileNotFoundError:
        return f"데이터 파일을 찾을 수 없습니다. 경로: {os.path.join('data', 'coachInfo.csv')}"
    except Exception as e:
        return f"오류가 발생했습니다: {str(e)}"