app.py

import gradio as gr
import json
from datetime import datetime

# 임시 데이터 저장소
users = []
contributions = []
nfts = []

def create_user(name, email):
    user = {
        'id': f'user_{len(users) + 1}',
        'name': name,
        'email': email,
        'points': 0,
        'level': 1,
        'created_at': datetime.now().isoformat()
    }
    users.append(user)
    return json.dumps(user, indent=2)

def create_contribution(user_id, contribution_type, description):
    # 포인트 계산
    points = {
        'POST_CREATION': 10,
        'COMMENT': 5,
        'REVIEW': 15,
        'BUG_REPORT': 20,
        'FEATURE_SUGGESTION': 15,
        'CODE_CONTRIBUTION': 30
    }
    
    # 사용자 찾기
    user = next((u for u in users if u['id'] == user_id), None)
    if not user:
        return json.dumps({'error': 'User not found'})
    
    # 기여도 생성
    contribution = {
        'id': f'contrib_{len(contributions) + 1}',
        'user_id': user_id,
        'type': contribution_type,
        'description': description,
        'points': points.get(contribution_type, 0),
        'created_at': datetime.now().isoformat()
    }
    contributions.append(contribution)
    
    # 사용자 포인트 업데이트
    user['points'] += contribution['points']
    user['level'] = (user['points'] // 100) + 1
    
    return json.dumps(contribution, indent=2)

def mint_nft(title, description, rarity):
    if not users:
        return json.dumps({'error': 'No users available'})
    
    nft = {
        'id': f'nft_{len(nfts) + 1}',
        'title': title,
        'description': description,
        'rarity': rarity,
        'owner_id': users[0]['id'],
        'created_at': datetime.now().isoformat()
    }
    nfts.append(nft)
    return json.dumps(nft, indent=2)

def get_leaderboard():
    sorted_users = sorted(users, key=lambda x: x['points'], reverse=True)
    leaderboard = []
    for i, user in enumerate(sorted_users, 1):
        leaderboard.append({
            'rank': i,
            'name': user['name'],
            'points': user['points'],
            'level': user['level']
        })
    return json.dumps(leaderboard, indent=2)

# Gradio 인터페이스 생성
with gr.Blocks(title="AI Tree Incentive System") as demo:
    gr.Markdown("""
    # 🎯 AI Tree Incentive System
    
    블로그 기여도 기반의 인센티브 시스템입니다.
    """)
    
    with gr.Tab("사용자 관리"):
        with gr.Row():
            with gr.Column():
                name_input = gr.Textbox(label="이름")
                email_input = gr.Textbox(label="이메일")
                create_user_btn = gr.Button("사용자 생성")
            with gr.Column():
                user_output = gr.JSON(label="생성된 사용자 정보")
        
        create_user_btn.click(
            create_user,
            inputs=[name_input, email_input],
            outputs=user_output
        )
    
    with gr.Tab("기여도 등록"):
        with gr.Row():
            with gr.Column():
                user_id_input = gr.Textbox(label="사용자 ID")
                contrib_type = gr.Dropdown(
                    choices=["POST_CREATION", "COMMENT", "REVIEW", "BUG_REPORT", "FEATURE_SUGGESTION", "CODE_CONTRIBUTION"],
                    label="기여 유형"
                )
                desc_input = gr.Textbox(label="설명")
                create_contrib_btn = gr.Button("기여도 등록")
            with gr.Column():
                contrib_output = gr.JSON(label="기여도 정보")
        
        create_contrib_btn.click(
            create_contribution,
            inputs=[user_id_input, contrib_type, desc_input],
            outputs=contrib_output
        )
    
    with gr.Tab("NFT 발행"):
        with gr.Row():
            with gr.Column():
                nft_title = gr.Textbox(label="NFT 제목")
                nft_desc = gr.Textbox(label="NFT 설명")
                nft_rarity = gr.Dropdown(
                    choices=["COMMON", "RARE", "EPIC", "LEGENDARY"],
                    label="희귀도"
                )
                mint_nft_btn = gr.Button("NFT 발행")
            with gr.Column():
                nft_output = gr.JSON(label="발행된 NFT 정보")
        
        mint_nft_btn.click(
            mint_nft,
            inputs=[nft_title, nft_desc, nft_rarity],
            outputs=nft_output
        )
    
    with gr.Tab("리더보드"):
        refresh_btn = gr.Button("새로고침")
        leaderboard_output = gr.JSON(label="리더보드")
        
        refresh_btn.click(
            get_leaderboard,
            inputs=[],
            outputs=leaderboard_output
        )

demo.launch()