Update app.py

app.py

@@ -1,4 +1,3 @@
-# Updated app.py
 import streamlit as st
 import tempfile
 import subprocess
@@ -6,8 +5,8 @@ import os
 import requests
 import base64
 import sys
-from streamlit_ace import st_ace
 import shutil
+from streamlit_ace import st_ace
 
 st.set_page_config(page_title="Blender 3D Viewer", layout="wide")
 st.title("🌍 Blender Script → 3D Viewer")
@@ -128,8 +127,9 @@ bpy.context.scene.render.engine = 'CYCLES'
 bpy.context.scene.cycles.device = 'CPU'
 bpy.context.scene.render.film_transparent = True
 
-# Save the .blend file in the current directory
-blend_file_path = bpy.path.abspath("//scene.blend")
+# Save the .blend file using an absolute path
+output_dir = os.environ.get('BLENDER_OUTPUT_DIR', os.getcwd())
+blend_file_path = os.path.join(output_dir, "scene.blend")
 bpy.ops.wm.save_as_mainfile(filepath=blend_file_path)
 """
 
@@ -229,14 +229,15 @@ if st.sidebar.button("Run & Export GLB"):
                     except Exception as e:
                         st.warning(f"pip install failed: {str(e)}")
 
-            # 4) Prepare environment with minimal variables
+            # 4) Prepare environment with necessary variables
             env = os.environ.copy()
             env["TEXTURE_PATHS"] = ",".join(texture_paths)
+            env["BLENDER_OUTPUT_DIR"] = tmp_dir  # Pass the tmp_dir to the Blender script
             
             # 5) Run Blender to build .blend file
             blend_path = os.path.join(tmp_dir, "scene.blend")
             
-            with st.status("Running Blender to create scene..."):
+            with st.status("Running Blender to create scene...") as status:
                 cmd1 = [blender_path, "--background", "--python", script_path]
                 try:
                     r1 = subprocess.run(
@@ -248,10 +249,15 @@ if st.sidebar.button("Run & Export GLB"):
                         text=True,
                         timeout=180  # 3 minute timeout
                     )
-                    st.code(r1.stdout.split("\n")[-10:], language="text")  # Show just the last few lines
+                    status.update(label="Blender scene created successfully", state="complete")
+                    st.text_area("Blender output", r1.stdout + r1.stderr, height=150)
                 except subprocess.TimeoutExpired:
                     st.error("Blender process timed out after 3 minutes.")
                     st.stop()
+                except subprocess.CalledProcessError as e:
+                    st.error(f"Blender build failed with error code {e.returncode}")
+                    st.text_area("Error details", e.stdout + e.stderr, height=150)
+                    st.stop()
                 except Exception as e:
                     st.error(f"Blender build failed: {str(e)}")
                     st.stop()
@@ -259,20 +265,32 @@ if st.sidebar.button("Run & Export GLB"):
             # Check if blend file was created
             if not os.path.exists(blend_path):
                 st.error("Blender did not create the expected scene.blend file.")
+                st.info("This might be due to an issue with the script or Blender configuration.")
                 st.stop()
                 
             # 6) Export GLB with animation
             glb_path = os.path.join(tmp_dir, "animation.glb")
             
-            with st.status("Exporting to GLB format..."):
-                expr = (
-                    "import bpy;"
-                    "bpy.ops.wm.open_mainfile(filepath=r'" + blend_path + "');"
-                    "bpy.ops.export_scene.gltf(filepath=r'" + glb_path + 
-                    "', export_format='GLB', export_animations=True)"
-                )
+            with st.status("Exporting to GLB format...") as status:
+                # Create a separate Python script for GLB export to avoid command line issues
+                export_script = os.path.join(tmp_dir, "export_glb.py")
+                with open(export_script, "w") as f:
+                    f.write(f"""
+import bpy
+import os
+
+# Load the blend file
+bpy.ops.wm.open_mainfile(filepath=r'{blend_path}')
+
+# Export to GLB
+bpy.ops.export_scene.gltf(
+    filepath=r'{glb_path}',
+    export_format='GLB',
+    export_animations=True
+)
+""")
                 
-                cmd2 = [blender_path, "--background", "--python-expr", expr]
+                cmd2 = [blender_path, "--background", "--python", export_script]
                 try:
                     r2 = subprocess.run(
                         cmd2,
@@ -283,6 +301,8 @@ if st.sidebar.button("Run & Export GLB"):
                         text=True,
                         timeout=120  # 2 minute timeout
                     )
+                    status.update(label="GLB export completed successfully", state="complete")
+                    st.text_area("Export output", r2.stdout + r2.stderr, height=100)
                 except Exception as e:
                     st.error(f"GLB export failed: {str(e)}")
                     st.stop()
@@ -292,6 +312,13 @@ if st.sidebar.button("Run & Export GLB"):
                 with open(glb_path, 'rb') as f:
                     data = f.read()
                 
+                file_size = len(data) / (1024 * 1024)  # Size in MB
+                st.success(f"Successfully created GLB file ({file_size:.1f} MB)")
+                
+                # Check if file isn't too large for embedding
+                if file_size > 50:
+                    st.warning("The GLB file is quite large. The viewer might be slow to load.")
+                
                 b64 = base64.b64encode(data).decode()
                 
                 # Display the 3D model viewer
@@ -303,7 +330,9 @@ if st.sidebar.button("Run & Export GLB"):
         alt="3D Model" 
         auto-rotate 
         camera-controls 
-        style="width:100%; height:600px;">
+        ar
+        shadow-intensity="1"
+        style="width:100%; height:600px; background-color: #f0f0f0;">
     </model-viewer>
     """
                 st.components.v1.html(html, height=650)
@@ -317,13 +346,15 @@ if st.sidebar.button("Run & Export GLB"):
                 )
             else:
                 st.error("GLB file was not generated successfully.")
+                if os.path.exists(blend_path):
+                    st.info("The Blender file was created, but the GLB export failed.")
                 
         finally:
             # Clean up - remove temporary directory
             try:
                 shutil.rmtree(tmp_dir, ignore_errors=True)
-            except:
-                pass
+            except Exception as e:
+                st.warning(f"Failed to clean up temporary files: {str(e)}")
 
 # Add helpful information at the bottom
 st.divider()
@@ -337,6 +368,36 @@ with st.expander("About this app"):
     - Download the results as GLB files for use in other applications
     
     The example script creates a rotating Earth with the uploaded texture.
+    
+    ### Troubleshooting
+    
+    - If you encounter errors, check the console output for details
+    - Make sure your script correctly saves files using absolute paths
+    - For complex models, allow more time for processing
+    - If the app doesn't work at all, the Hugging Face Space might need proper Blender installation
     """)
     
-    st.info("If the app isn't working, make sure 'blender' is properly installed in the Spaces environment.")
+with st.expander("Script Tips"):
+    st.markdown("""
+    ### Tips for writing Blender scripts
+    
+    1. **Accessing textures**: Use the environment variable `TEXTURE_PATHS` to get the paths to uploaded textures
+    
+    ```python
+    # Example code to load the first uploaded texture
+    texture_paths = os.environ.get('TEXTURE_PATHS', '').split(',')
+    if texture_paths and texture_paths[0]:
+        image = bpy.data.images.load(texture_paths[0])
+    ```
+    
+    2. **Saving files**: Always use absolute paths with the environment variable
+    
+    ```python
+    # Example code to save your blend file
+    output_dir = os.environ.get('BLENDER_OUTPUT_DIR', os.getcwd())
+    blend_file_path = os.path.join(output_dir, "scene.blend")
+    bpy.ops.wm.save_as_mainfile(filepath=blend_file_path)
+    ```
+    
+    3. **Animations**: Make sure to set keyframes if you want your model to animate
+    """)