Update app.py

app.py

@@ -7,6 +7,7 @@ import asyncio
 import tempfile
 from threading import Thread
 import base64
+import shutil
 
 import gradio as gr
 import spaces
@@ -43,7 +44,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 def glb_to_data_url(glb_path: str) -> str:
     """
     Reads a GLB file from disk and returns a data URL with a base64 encoded representation.
-    This data URL can be used as the `src` for an HTML <model-viewer> tag.
+    (Not used in this method.)
     """
     with open(glb_path, "rb") as f:
         data = f.read()
@@ -332,22 +333,18 @@ def generate(
             num_steps=64,
             randomize_seed=True,
         )
-        # Convert the GLB file to a base64 data URL.
-        data_url = glb_to_data_url(glb_path)
-        # Create an HTML snippet that loads model-viewer and converts the data URL to a Blob URL.
+        # Instead of embedding via data URL, copy the GLB file to a static folder.
+        static_folder = os.path.join(os.getcwd(), "static")
+        if not os.path.exists(static_folder):
+            os.makedirs(static_folder)
+        new_filename = f"mesh_{uuid.uuid4()}.glb"
+        new_filepath = os.path.join(static_folder, new_filename)
+        shutil.copy(glb_path, new_filepath)
+        # Reference the file via a relative URL.
         html_output = f'''
         <script type="module" src="https://unpkg.com/@google/model-viewer/dist/model-viewer.min.js"></script>
-        <model-viewer id="model-viewer" alt="3D Model" auto-rotate camera-controls 
+        <model-viewer src="/static/{new_filename}" alt="3D Model" auto-rotate camera-controls 
                       style="width: 100%; height: 400px;"></model-viewer>
-        <script>
-          (async () => {{
-            const response = await fetch("{data_url}");
-            const arrayBuffer = await response.arrayBuffer();
-            const blob = new Blob([arrayBuffer], {{type: 'model/gltf-binary'}});
-            const blobUrl = URL.createObjectURL(blob);
-            document.getElementById("model-viewer").src = blobUrl;
-          }})();
-        </script>
         '''
         yield gr.HTML(html_output)
         return
@@ -480,5 +477,5 @@ demo = gr.ChatInterface(
 )
 
 if __name__ == "__main__":
-    # To create a public link, set share=True in launch().
-    demo.queue(max_size=20).launch(share=True)
+    # To serve the generated GLB files, add the static directory.
+    demo.queue(max_size=20).launch(share=True, static_dirs={"static": "static"})