app.py

import gradio as gr
import pandas as pd
import numpy as np

# 데이터 정의 (하드코딩)
data = {
    "Company/Model": [
        "Anthropic/Claude 3 Opus",
        "OpenAI/GPT-4",
        "Google/Gemini Ultra",
        "Cohere/Command R+",
        "Naver/HyperCLOVA X",
        "Kakao/KoGPT"
    ],
    "URL": [
        "https://www.anthropic.com/claude",
        "https://openai.com/gpt-4",
        "https://deepmind.google/technologies/gemini/",
        "https://cohere.com/models/command-r-plus",
        "https://clova.ai/hyperclova",
        "https://kogpt.ai/"
    ],
    "Korean Bar Exam (변호사)": [85, 82, 80, 75, 79, 77],
    "Senior Civil Service Examination (국가직 5급)": [88, 84, 83, 76, 81, 78]
}

# DataFrame 생성
df = pd.DataFrame(data)

# Average 점수 계산
exam_columns = ["Korean Bar Exam (변호사)", "Senior Civil Service Examination (국가직 5급)"]
df["Average"] = df[exam_columns].mean(axis=1).round(1)

# 열 순서 재배치 (Company/Model, URL, Average, 그 다음 각 시험)
cols = ["Company/Model", "URL", "Average"] + exam_columns
df = df[cols]

# HTML로 렌더링하기 위한 함수 (URL을 클릭 가능한 링크로 변환)
def format_df_as_html(df):
    # DataFrame 복사본 생성
    display_df = df.copy()
    
    # URL 열을 클릭 가능한 링크로 변환
    for i, url in enumerate(display_df["URL"]):
        model_name = display_df.iloc[i]["Company/Model"]
        display_df.at[i, "Company/Model"] = f'<a href="{url}" target="_blank">{model_name}</a>'
    
    # URL 열 제거 (이미 Company/Model에 링크로 통합)
    display_df = display_df.drop("URL", axis=1)
    
    # 표 스타일 추가
    styled_html = """
    <style>
        table {
            width: 100%;
            border-collapse: collapse;
            font-family: Arial, sans-serif;
        }
        th {
            background-color: #4CAF50;
            color: white;
            font-weight: bold;
            text-align: left;
            padding: 12px;
        }
        td {
            padding: 10px;
            border-bottom: 1px solid #ddd;
        }
        tr:nth-child(even) {
            background-color: #f2f2f2;
        }
        tr:hover {
            background-color: #ddd;
        }
        .header {
            text-align: center;
            font-size: 24px;
            font-weight: bold;
            margin-bottom: 20px;
            color: #333;
        }
    </style>
    <div class="header">Korean Exam Leaderboard</div>
    """
    
    # DataFrame을 HTML로 변환하고 스타일 적용
    html_table = display_df.to_html(index=False, escape=False)
    return styled_html + html_table

# Gradio 인터페이스
def show_leaderboard():
    html_content = format_df_as_html(df)
    return html_content

# 인터페이스 생성
demo = gr.Interface(
    fn=show_leaderboard,
    inputs=None,
    outputs=gr.HTML(),
    title="Korean Exam Leaderboard",
    description="성능 비교: 한국 법학 및 행정고시 시험에서의 AI 모델 점수"
)

# 앱 실행
if __name__ == "__main__":
    demo.launch()