import gradio as gr
import os

def load_mesh_with_info(mesh_file):
    if not mesh_file:
        return None, "No file selected"
    
    file_size = os.path.getsize(mesh_file) / 1024  # Convert to KB
    return mesh_file, f"File: {os.path.basename(mesh_file)}, Size: {file_size:.2f} KB"

demo = gr.Interface(
    fn=load_mesh_with_info,
    inputs=[gr.Model3D(label="Upload a 3D Model")],
    outputs=[gr.Model3D(clear_color=[0.1, 0.1, 0.1, 1.0], label="Preview Model"),
             gr.Text(label="File Info")],
    examples=[
        ["files/model1.glb"],
        ["files/model2.glb"],
        ["files/model3.glb"],
        ["files/model4.glb"],
    ],
    live=True
)

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