Upload gradio_app.py

gradio_app.py

@@ -137,25 +137,39 @@ def build_model_viewer_html(save_folder, height=660, width=790, textured=False):
         template_html = template_html.replace('#src#', f'{related_path}/')
         f.write(template_html)
 
-    iframe_tag = f'<iframe src="file={output_html_path}" height="{height}" width="100%" frameborder="0"></iframe>'
-    print(
-        f'Find html file {output_html_path}, {os.path.exists(output_html_path)}')
+    #iframe_tag = f'<iframe src="file={output_html_path}" height="{height}" width="100%" frameborder="0"></iframe>'
+    #print(
+    #    f'Find html file {output_html_path}, {os.path.exists(output_html_path)}')
+
+    #return f"""
+    #    <div style='height: {height}; width: 100%;'>
+    #    {iframe_tag}
+    #    </div>
+    #"""
 
-    return f"""
-        <div style='height: {height}; width: 100%;'>
-        {iframe_tag}
-        </div>
-    """
 
-    #if textured:
-    #    mesh_file = os.path.join(save_folder, f'textured_mesh.glb')
-    #else: 
-    #    mesh_file = os.path.join(save_folder, f'white_mesh.glb')
+    if textured:
+        mesh_file_path = os.path.join(save_folder, f'textured_mesh.glb')
+    else: 
+        mesh_file_path = os.path.join(save_folder, f'white_mesh.glb')
+
 
+    print(
+        f'Find html file {mesh_file_path}, {os.path.exists(mesh_file_path)}')
+
+    
     # Encode the GLB file as base64 to embed directly in the HTML
-    #with open(mesh_file, "rb") as f:
-    #    glb_data = f.read()
-    #    glb_base64 = base64.b64encode(glb_data).decode("utf-8")
+    with open(mesh_file_path, 'rb') as f:
+        encoded = base64.b64encode(f.read()).decode('utf-8')
+    
+    html = f"""
+    <model-viewer src="data:model/gltf-binary;base64,{encoded}" 
+                  alt="3D Model" camera-controls auto-rotate 
+                  style="height: 690px; width: 500px;">
+    </model-viewer>
+    """
+
+    return html
 
     #model_viewer_html = f"""
     #<script type="module" src="https://unpkg.com/@google/model-viewer/dist/model-viewer.min.js"></script>
@@ -166,7 +180,6 @@ def build_model_viewer_html(save_folder, height=660, width=790, textured=False):
     #              style="height: {height}px; width: {width}px;">
     #</model-viewer>
     #"""
-
     #return model_viewer_html
 
 @spaces.GPU(duration=40)