import gradio as gr
import random

# 定義抽籤的邏輯
def draw_lottery(max_seat_number):
    return random.randint(1, max_seat_number)

# 設置 Gradio 的界面
with gr.Blocks() as demo:
    gr.Markdown("## 抽籤系統")
    
    # 輸入框讓用戶輸入班級座號最大值
    max_seat_number = gr.Number(label="輸入班級座號最大值", value=30)
    
    # 按鈕用來觸發抽籤
    draw_button = gr.Button("抽籤")
    
    # 顯示抽出的號碼
    result = gr.Textbox(label="抽中的號碼")
    
    # 當按下按鈕時執行抽籤
    draw_button.click(draw_lottery, inputs=max_seat_number, outputs=result)

# 運行 Gradio 應用
demo.launch()