Upload gradio_app.py

gradio_app.py

@@ -142,14 +142,7 @@ def build_model_viewer_html(save_folder, path, height=660, width=790, textured=F
         template_html = template_html.replace('#src#', f'https://nicolasg2523-picsto3d.hf.space/gradio_api/file=/tmp/{path}')
         f.write(template_html)
 
-    tenplate_html_encoded = html.escape(template_html, quote=True).replace("'", "'")
-    iframe_tag = f'<iframe srcdoc="{tenplate_html_encoded}" height="{height}" width="100%" frameborder="0"></iframe>'
-
-    return f"""
-        <div style='height: {height}; width: 100%;'>
-        {iframe_tag}
-        </div>
-    """
+    return output_html_path
 
 @spaces.GPU(duration=40)
 def _gen_shape(
@@ -484,6 +477,26 @@ def build_app():
                 with gr.Tabs(selected='gen_mesh_panel') as tabs_output:
                     with gr.Tab('Generated Mesh', id='gen_mesh_panel'):
                         html_gen_mesh = gr.HTML(HTML_OUTPUT_PLACEHOLDER, label='Output')
+                        file_output = gr.File(label="Generated GLB")
+                        iframe_html = """
+                        <iframe id="model_iframe" style="width:500px;height:500px;"></iframe>
+                        <script>
+                            function updateIframe(srcUrl) {
+                                document.getElementById('model_iframe').srcdoc = `
+                                    <script type="module" src="https://unpkg.com/@google/model-viewer/dist/model-viewer.min.js"></script>
+                                    <model-viewer src="${srcUrl}" camera-controls auto-rotate style="height:500px;width:500px;"></model-viewer>
+                                `;
+                            }
+                        </script>
+                        """
+                        iframe = gr.HTML(iframe_html)
+                        file_output.change(fn=js_update, inputs=file_output, outputs=js_injection)
+
+                        def js_update(file):
+                            return f"<script>updateIframe('{file}');</script>"
+
+                        js_injection = gr.HTML()
+
                     with gr.Tab('Exporting Mesh', id='export_mesh_panel'):
                         html_export_mesh = gr.HTML(HTML_OUTPUT_PLACEHOLDER, label='Output')
                     with gr.Tab('Mesh Statistic', id='stats_panel'):
@@ -507,7 +520,7 @@ def build_app():
                 num_chunks,
                 randomize_seed,
             ],
-            outputs=[file_out, html_gen_mesh, stats, seed]
+            outputs=[file_out, html_gen_mesh, stats, seed, file_output]
         ).then(
             lambda: (gr.update(visible=False, value=False), gr.update(interactive=True), gr.update(interactive=True),
                      gr.update(interactive=False)),