Upload gradio_app.py

gradio_app.py

@@ -28,6 +28,7 @@ from fastapi.staticfiles import StaticFiles
 import uuid
 
 import base64
+from html import escape
 
 from hy3dgen.shapegen.utils import logger
 
@@ -138,15 +139,12 @@ 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'{mesh_file_path}/')
-        
+        template_html = template_html.replace('#src#', f'{mesh_file_path}/')        
         f.write(template_html)
 
-    print(related_path)
-    print(mesh_file_path)
-    print(output_html_path)
-
-    iframe_tag = f'<iframe src="file=https://nicolasg2523-picsto3d.hf.space/gradio_api/file=/tmp{output_html_path}" height="{height}" width="100%" frameborder="0"></iframe>'
+    tenplate_html_encoded = encode(tenplate_html)
+    iframe_tag = f'<iframe srcdoc="{tenplate_html_encoded}" height="{height}" width="100%" frameborder="0"></iframe>'
+    #iframe_tag = f'<iframe src="file=https://nicolasg2523-picsto3d.hf.space/gradio_api/file=/tmp{output_html_path}" height="{height}" width="100%" frameborder="0"></iframe>'
     
     print(f'Find html file {output_html_path}, {os.path.exists(output_html_path)}')
     print(iframe_tag)