import os

import gradio as gr

# 设置行数和列数, 可以根据需要调整
M, N = 2, 3
scene_name = [["渐缩木屋.glb", "渐缩木屋.glb", "渐缩木屋.glb"], ["渐缩木屋.glb", "渐缩木屋.glb", "渐缩木屋.glb"]]
scene_path = [[os.path.join(os.path.dirname(__file__), "scenes", col) for col in row] for row in scene_name]

with gr.Blocks() as demo:
    gr.Markdown("<h1 style='text-align: center;'>场景生成用户调研</h1>")
    
    for i in range(M):
        with gr.Row():
            for j in range(N):
                model_path = scene_path[i][j]
                with gr.Column():
                    gr.Model3D(value=model_path, label=f"模型 {i*N + j + 1}", height=500)
                    gr.Slider(minimum=1, maximum=5, step=1, value=3, label="按照美学评分", interactive=True)
                    gr.Slider(minimum=1, maximum=5, step=1, value=3, label="按照保真性评分", interactive=True)

    demo.launch(server_name='0.0.0.0', server_port=7860)