Replace LitModel3D with gr.Model3D and restore functionality for GLB extraction

app.py

@@ -239,6 +239,7 @@ with gr.Blocks(title="Game Items Generator") as demo:
         with gr.Column():
             video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
             # model_output = LitModel3D(label="Extracted GLB", exposure=20.0, height=300)
+            model_output = gr.Model3D(label="Extracted GLB", height=300)
             download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
             
     trial_id = gr.Textbox(visible=False)
@@ -293,19 +294,19 @@ with gr.Blocks(title="Game Items Generator") as demo:
         outputs=[extract_glb_btn],
     )
 
-    # extract_glb_btn.click(
-    #     extract_glb,
-    #     inputs=[output_buf, mesh_simplify, texture_size],
-    #     outputs=[model_output, download_glb],
-    # ).then(
-    #     activate_button,
-    #     outputs=[download_glb],
-    # )
-
-    # model_output.clear(
-    #     deactivate_button,
-    #     outputs=[download_glb],
-    # )
+    extract_glb_btn.click(
+        extract_glb,
+        inputs=[output_buf, mesh_simplify, texture_size],
+        outputs=[model_output, download_glb],
+    ).then(
+        activate_button,
+        outputs=[download_glb],
+    )
+
+    model_output.clear(
+        deactivate_button,
+        outputs=[download_glb],
+    )
     
 
 # Launch the Gradio app