Upload gradio_app.py

gradio_app.py

@@ -364,12 +364,12 @@ def shape_generation(
         model_viewer_html,
         stats,
         seed,
-        model_viewer_html
+        gr.update(value=path)
     )
 
 def js_update(file):
     return f"<script>updateIframe('{file}');</script>"
-                            
+
 def build_app():
     title = 'Generación de modelo basado en imágenes (de 1 a 4 vistas)'
 
@@ -481,21 +481,7 @@ def build_app():
                     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)
-                        js_injection = gr.HTML()
-                        file_output.change(fn=js_update, inputs=file_output, outputs=js_injection)
-
+                        file_output.change(fn=js_update, inputs=file_output, outputs=html_gen_mesh)
 
                     with gr.Tab('Exporting Mesh', id='export_mesh_panel'):
                         html_export_mesh = gr.HTML(HTML_OUTPUT_PLACEHOLDER, label='Output')
@@ -661,6 +647,15 @@ if __name__ == '__main__':
     <div style='height: {650}px; width: 100%; border-radius: 8px; border-color: #e5e7eb; border-style: solid; border-width: 1px; display: flex; justify-content: center; align-items: center;'>
       <div style='text-align: center; font-size: 16px; color: #6b7280;'>
         <p style="color: #8d8d8d;">Bienvenido a 3DMarket</p>
+        <iframe id="model_iframe" style="width:500px;height:500px;">
+            <script type="module" src="https://unpkg.com/@google/model-viewer/dist/model-viewer.min.js"></script>
+            <model-viewer id="model_viewer" src="" camera-controls auto-rotate style="height:500px;width:500px;"></model-viewer>
+        </iframe>
+        <script>
+            function updateIframe(srcUrl) {
+                document.getElementById('model_viewer').src = '${srcUrl}';
+            }
+        </script>
       </div>
     </div>
     """