Upload gradio_app.py

gradio_app.py

@@ -118,7 +118,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     return seed
 
 
-def build_model_viewer_html(save_folder, height=660, width=790, textured=False):
+def build_model_viewer_html(save_folder, height=660, width=790, textured=False, mesh_path):
     # Remove first folder from path to make relative path
     if textured:
         related_path = f"./textured_mesh.glb"
@@ -139,7 +139,7 @@ def build_model_viewer_html(save_folder, height=660, width=790, textured=False):
         template_html = template_html.replace('#height#', f'{height - offset}')
         template_html = template_html.replace('#width#', f'{width}')
         #template_html = template_html.replace('#src#', f'{related_path}/')
-        template_html = template_html.replace('#src#', f'https://nicolasg2523-picsto3d.hf.space/gradio_api/file=/tmp/{mesh_file_path}')
+        template_html = template_html.replace('#src#', f'https://nicolasg2523-picsto3d.hf.space/gradio_api/file=/tmp/{mesh_path}')
         f.write(template_html)
 
     tenplate_html_encoded = html.escape(template_html, quote=True).replace("'", "'")
@@ -359,7 +359,8 @@ def shape_generation(
     mesh.metadata['extras'] = stats
 
     path = export_mesh(mesh, save_folder, textured=False)
-    model_viewer_html = build_model_viewer_html(save_folder, height=HTML_HEIGHT, width=HTML_WIDTH)
+    print(path)
+    model_viewer_html = build_model_viewer_html(save_folder, height=HTML_HEIGHT, width=HTML_WIDTH, path)
     if args.low_vram_mode:
         torch.cuda.empty_cache()
     return (
@@ -575,7 +576,7 @@ def build_app():
                 save_folder = gen_save_folder()
                 _ = export_mesh(mesh, save_folder, textured=True)
                 model_viewer_html = build_model_viewer_html(save_folder, height=HTML_HEIGHT, width=HTML_WIDTH,
-                                                            textured=True)
+                                                            textured=True, path)
             else:
                 mesh = trimesh.load(file_out)
                 mesh = floater_remove_worker(mesh)
@@ -589,7 +590,7 @@ def build_app():
                 save_folder = gen_save_folder()
                 _ = export_mesh(mesh, save_folder, textured=False)
                 model_viewer_html = build_model_viewer_html(save_folder, height=HTML_HEIGHT, width=HTML_WIDTH,
-                                                            textured=False)
+                                                            textured=False, path)
             print(f'export to {path}')
             return model_viewer_html, gr.update(value=path, interactive=True)