app.py

import gradio as gr
import requests

# Sleeper API 엔드포인트
BASE_URL = "https://api.sleeper.app/v1"

# 리그 이름을 가져오기 위한 함수
def fetch_league_name(league_id):
    url = f"{BASE_URL}/league/{league_id}"
    response = requests.get(url)
    if response.status_code == 200:
        league_data = response.json()
        return league_data.get("name", f"리그 이름 없음 ({league_id})")
    else:
        return f"리그 정보 가져오기 실패 ({league_id})"

# 선수 이름을 가져오기 위한 유틸리티 함수
def fetch_players():
    url = f"{BASE_URL}/players/nba"
    response = requests.get(url)
    if response.status_code == 200:
        players_data = response.json()
        return {player_id: data["full_name"] for player_id, data in players_data.items() if "full_name" in data.keys()}
    else:
        return {}

# 특정 주차의 매치업 정보 가져오기
def fetch_matchups(league_id, week):
    url = f"{BASE_URL}/league/{league_id}/matchups/{week}"
    response = requests.get(url)
    if response.status_code == 200:
        return response.json()
    else:
        return []

# 주차별 매치업 정보와 선수 이름 표시
def get_weekly_matchup(league_id, week):
    players_map = fetch_players()
    matchups = fetch_matchups(league_id, week)
    if not matchups:
        return {"error": "주차 매치업 정보를 가져오는 데 실패했습니다."}

    results = []
    for matchup in matchups:
        entry = {
            "Team ID": matchup.get("roster_id", "Unknown"),
            "Points": matchup.get("points", 0),
            "Players": [players_map.get(player_id, player_id) for player_id in matchup.get("players", [])],
        }
        results.append(entry)

    return results

# 리그 이름과 주차를 입력받아 결과를 반환하는 함수
def display_weekly_matchup(league_name, week):
    league_ids = {
        fetch_league_name("1149290874371846144"): "1149290874371846144",
        fetch_league_name("1083634148067074048"): "1083634148067074048",
    }
    league_id = league_ids.get(league_name)
    if not league_id:
        return {"error": "유효하지 않은 리그 이름입니다."}

    return get_weekly_matchup(league_id, int(week))

# 리그 이름을 드롭다운에 동적으로 추가
def get_league_choices():
    return [
        fetch_league_name("1149290874371846144"),
        fetch_league_name("1083634148067074048"),
    ]

# Gradio 인터페이스 설정
with gr.Blocks() as demo:
    gr.Markdown("# 판타지 NBA 리그 매치업 정보")
    gr.Markdown("Sleeper API를 사용하여 리그 이름과 주차별 매치업 정보를 확인합니다.")
    
    with gr.Row():
        league_dropdown = gr.Dropdown(choices=get_league_choices(), label="리그 선택")
        week_input = gr.Number(label="주차 입력 (숫자)")
    
    result_output = gr.JSON(label="결과")
    
    submit_button = gr.Button("결과 보기")
    
    # 버튼 클릭 시 데이터 처리
    submit_button.click(
        display_weekly_matchup,
        inputs=[league_dropdown, week_input],
        outputs=result_output
    )

if __name__ == "__main__":
    demo.launch()