Update app.py
Browse files
app.py
CHANGED
@@ -7,62 +7,143 @@ import base64
|
|
7 |
import sys
|
8 |
import shutil
|
9 |
import re
|
|
|
|
|
10 |
from streamlit_ace import st_ace
|
11 |
|
|
|
|
|
|
|
|
|
12 |
st.set_page_config(page_title="Blender 3D Viewer", layout="wide")
|
13 |
st.title("🌍 Blender Script → 3D Viewer")
|
14 |
|
15 |
-
#
|
|
|
|
|
|
|
|
|
|
|
16 |
def find_blender():
|
17 |
-
# Common paths to check
|
18 |
paths_to_check = [
|
19 |
"blender", # default PATH
|
20 |
"/usr/bin/blender",
|
21 |
-
"/opt/blender/blender"
|
|
|
|
|
|
|
22 |
]
|
23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
24 |
for path in paths_to_check:
|
25 |
try:
|
|
|
26 |
result = subprocess.run([path, "--version"],
|
27 |
capture_output=True,
|
28 |
text=True,
|
29 |
-
timeout=
|
30 |
if result.returncode == 0:
|
31 |
st.success(f"Found Blender: {result.stdout.strip()}")
|
|
|
32 |
return path
|
33 |
-
except:
|
|
|
34 |
continue
|
35 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
36 |
return None
|
37 |
|
38 |
-
#
|
39 |
def preprocess_script(script_text, tmp_dir):
|
40 |
-
|
41 |
-
|
42 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
43 |
|
44 |
-
#
|
45 |
-
|
46 |
-
|
47 |
|
48 |
-
#
|
49 |
if 'bpy.ops.wm.save_as_mainfile' not in script_text:
|
50 |
-
# Add file saving code at the end
|
51 |
script_text += f'''
|
52 |
|
53 |
-
# Save the .blend file
|
54 |
-
|
55 |
-
|
56 |
-
|
57 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
58 |
'''
|
|
|
59 |
else:
|
60 |
-
# Make sure existing save code uses absolute paths correctly
|
61 |
-
|
62 |
-
|
63 |
-
|
64 |
-
|
65 |
-
|
|
|
|
|
|
|
66 |
|
67 |
return script_text
|
68 |
|
@@ -70,13 +151,20 @@ print(f"Saved blend file to: {{blend_file_path}}")
|
|
70 |
blender_path = find_blender()
|
71 |
if not blender_path:
|
72 |
st.error("❌ Blender not found! The app requires Blender to be installed.")
|
73 |
-
|
|
|
|
|
|
|
|
|
74 |
# Continue anyway to show the interface
|
75 |
|
76 |
# Hardcoded default texture URLs (with fallbacks)
|
77 |
DEFAULT_TEXTURE_URLS = [
|
78 |
"https://eoimages.gsfc.nasa.gov/images/imagerecords/57000/57730/land_ocean_ice_2048.jpg",
|
79 |
-
"https://svs.gsfc.nasa.gov/vis/a000000/a002900/a002915/bluemarble-2048.png"
|
|
|
|
|
|
|
80 |
]
|
81 |
|
82 |
# Sidebar: Blender script editor with example script
|
@@ -256,6 +344,34 @@ bpy.context.scene.cycles.device = 'CPU'
|
|
256 |
print('Solar system animation setup complete.')"""
|
257 |
}
|
258 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
259 |
# Add example selector
|
260 |
selected_example = st.sidebar.selectbox(
|
261 |
"Load example script:",
|
@@ -287,6 +403,19 @@ custom_packages = st.sidebar.text_area(
|
|
287 |
height=100
|
288 |
)
|
289 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
290 |
# Main action
|
291 |
if st.sidebar.button("Run & Export GLB"):
|
292 |
if not blender_path:
|
@@ -300,12 +429,15 @@ if st.sidebar.button("Run & Export GLB"):
|
|
300 |
with st.spinner("Processing your 3D scene..."):
|
301 |
# Use a more robust temporary directory approach
|
302 |
tmp_dir = tempfile.mkdtemp(prefix="blender_app_")
|
|
|
|
|
303 |
try:
|
304 |
# 1) Preprocess and save user script
|
305 |
processed_script = preprocess_script(script_text, tmp_dir)
|
306 |
script_path = os.path.join(tmp_dir, "user_script.py")
|
307 |
with open(script_path, "w") as f:
|
308 |
f.write(processed_script)
|
|
|
309 |
|
310 |
# 2) Collect textures
|
311 |
texture_paths = []
|
@@ -316,20 +448,23 @@ if st.sidebar.button("Run & Export GLB"):
|
|
316 |
with open(path, "wb") as tf:
|
317 |
tf.write(upload.read())
|
318 |
texture_paths.append(path)
|
|
|
319 |
else:
|
320 |
# Try multiple default textures if some fail
|
321 |
for url in DEFAULT_TEXTURE_URLS:
|
322 |
try:
|
323 |
-
r = requests.get(url, timeout=
|
324 |
r.raise_for_status()
|
325 |
ext = os.path.splitext(url)[-1] or ".jpg"
|
326 |
path = os.path.join(tmp_dir, f"default{ext}")
|
327 |
with open(path, "wb") as tf:
|
328 |
tf.write(r.content)
|
329 |
texture_paths.append(path)
|
|
|
330 |
# If we got one texture successfully, that's enough
|
331 |
break
|
332 |
except Exception as e:
|
|
|
333 |
st.warning(f"Could not download texture {url}: {str(e)}")
|
334 |
|
335 |
# Display texture file information
|
@@ -337,14 +472,17 @@ if st.sidebar.button("Run & Export GLB"):
|
|
337 |
st.subheader("Texture Files")
|
338 |
for tp in texture_paths:
|
339 |
st.code(os.path.basename(tp))
|
|
|
340 |
else:
|
341 |
st.warning("No textures were loaded. Your 3D model may appear without textures.")
|
|
|
342 |
|
343 |
# 3) Install custom Python libraries if specified
|
344 |
if custom_packages and custom_packages.strip():
|
345 |
pkgs = [l.strip() for l in custom_packages.splitlines() if l.strip()]
|
346 |
if pkgs:
|
347 |
st.info(f"Installing: {', '.join(pkgs)}")
|
|
|
348 |
try:
|
349 |
# Use Python executable from current environment for pip
|
350 |
pip_cmd = [sys.executable, "-m", "pip", "install", "--user"] + pkgs
|
@@ -353,52 +491,123 @@ if st.sidebar.button("Run & Export GLB"):
|
|
353 |
check=True,
|
354 |
capture_output=True,
|
355 |
text=True,
|
356 |
-
timeout=
|
357 |
)
|
358 |
st.text_area("pip install output", pip_res.stdout + pip_res.stderr, height=100)
|
|
|
359 |
except Exception as e:
|
360 |
st.warning(f"pip install failed: {str(e)}")
|
|
|
361 |
|
362 |
# 4) Prepare environment with necessary variables
|
363 |
env = os.environ.copy()
|
364 |
env["TEXTURE_PATHS"] = ",".join(texture_paths)
|
365 |
env["BLENDER_OUTPUT_DIR"] = tmp_dir # Pass the tmp_dir to the Blender script
|
|
|
366 |
|
367 |
-
# 5) Run Blender to build .blend file
|
368 |
blend_path = os.path.join(tmp_dir, "scene.blend")
|
369 |
|
370 |
-
|
371 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
372 |
try:
|
373 |
-
|
374 |
-
|
|
|
375 |
cwd=tmp_dir,
|
376 |
env=env,
|
377 |
check=True,
|
378 |
capture_output=True,
|
379 |
text=True,
|
380 |
-
timeout=
|
381 |
)
|
382 |
-
|
383 |
-
|
384 |
-
|
385 |
-
|
386 |
-
|
387 |
-
|
388 |
-
|
389 |
-
|
390 |
-
|
|
|
391 |
except Exception as e:
|
392 |
-
st.error(f"
|
|
|
393 |
st.stop()
|
394 |
-
|
395 |
-
# Check if blend file was created
|
396 |
-
if not os.path.exists(blend_path):
|
397 |
-
st.error("Blender did not create the expected scene.blend file.")
|
398 |
-
st.info("This might be due to an issue with the script or Blender configuration.")
|
399 |
-
st.stop()
|
400 |
|
401 |
-
# 6) Export GLB with animation
|
402 |
glb_path = os.path.join(tmp_dir, "animation.glb")
|
403 |
|
404 |
with st.status("Exporting to GLB format...") as status:
|
@@ -408,19 +617,57 @@ if st.sidebar.button("Run & Export GLB"):
|
|
408 |
f.write(f"""
|
409 |
import bpy
|
410 |
import os
|
|
|
411 |
|
412 |
-
#
|
413 |
-
|
414 |
-
|
415 |
-
|
416 |
-
|
417 |
-
|
418 |
-
|
419 |
-
|
420 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
421 |
""")
|
422 |
|
423 |
cmd2 = [blender_path, "--background", "--python", export_script]
|
|
|
|
|
424 |
try:
|
425 |
r2 = subprocess.run(
|
426 |
cmd2,
|
@@ -429,12 +676,14 @@ bpy.ops.export_scene.gltf(
|
|
429 |
check=True,
|
430 |
capture_output=True,
|
431 |
text=True,
|
432 |
-
timeout=
|
433 |
)
|
434 |
status.update(label="GLB export completed successfully", state="complete")
|
435 |
st.text_area("Export output", r2.stdout + r2.stderr, height=100)
|
|
|
436 |
except Exception as e:
|
437 |
st.error(f"GLB export failed: {str(e)}")
|
|
|
438 |
st.stop()
|
439 |
|
440 |
# 7) Embed GLB inline if it exists
|
@@ -444,10 +693,12 @@ bpy.ops.export_scene.gltf(
|
|
444 |
|
445 |
file_size = len(data) / (1024 * 1024) # Size in MB
|
446 |
st.success(f"Successfully created GLB file ({file_size:.1f} MB)")
|
|
|
447 |
|
448 |
# Check if file isn't too large for embedding
|
449 |
if file_size > 50:
|
450 |
st.warning("The GLB file is quite large. The viewer might be slow to load.")
|
|
|
451 |
|
452 |
b64 = base64.b64encode(data).decode()
|
453 |
|
@@ -476,15 +727,24 @@ bpy.ops.export_scene.gltf(
|
|
476 |
)
|
477 |
else:
|
478 |
st.error("GLB file was not generated successfully.")
|
|
|
|
|
479 |
if os.path.exists(blend_path):
|
480 |
st.info("The Blender file was created, but the GLB export failed.")
|
|
|
481 |
|
482 |
finally:
|
483 |
-
# Clean up - remove temporary directory
|
484 |
-
|
485 |
-
|
486 |
-
|
487 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
488 |
|
489 |
# Add helpful information at the bottom
|
490 |
st.divider()
|
@@ -497,8 +757,11 @@ with st.expander("About this app"):
|
|
497 |
- Generate and visualize 3D models with animations
|
498 |
- Download the results as GLB files for use in other applications
|
499 |
|
500 |
-
The app automatically adds necessary imports
|
501 |
so you can focus on creating 3D content without worrying about environment details.
|
|
|
|
|
|
|
502 |
""")
|
503 |
|
504 |
with st.expander("Script Tips"):
|
@@ -525,4 +788,37 @@ with st.expander("Script Tips"):
|
|
525 |
obj.rotation_euler = (0, 0, math.radians(360))
|
526 |
obj.keyframe_insert(data_path="rotation_euler", frame=250)
|
527 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
528 |
""")
|
|
|
7 |
import sys
|
8 |
import shutil
|
9 |
import re
|
10 |
+
import logging
|
11 |
+
import platform
|
12 |
from streamlit_ace import st_ace
|
13 |
|
14 |
+
# Set up logging
|
15 |
+
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
16 |
+
logger = logging.getLogger('blender_app')
|
17 |
+
|
18 |
st.set_page_config(page_title="Blender 3D Viewer", layout="wide")
|
19 |
st.title("🌍 Blender Script → 3D Viewer")
|
20 |
|
21 |
+
# Detect environment
|
22 |
+
IS_HUGGINGFACE = os.environ.get('SPACE_ID') is not None
|
23 |
+
if IS_HUGGINGFACE:
|
24 |
+
st.info("Running in Hugging Face Spaces environment")
|
25 |
+
|
26 |
+
# Find Blender executable with enhanced detection for Hugging Face
|
27 |
def find_blender():
|
28 |
+
# Common paths to check, expanded for Hugging Face environment
|
29 |
paths_to_check = [
|
30 |
"blender", # default PATH
|
31 |
"/usr/bin/blender",
|
32 |
+
"/opt/blender/blender",
|
33 |
+
"/app/bin/blender", # Common in Docker containers
|
34 |
+
"/usr/local/bin/blender",
|
35 |
+
os.path.join(os.path.expanduser("~"), "blender/blender")
|
36 |
]
|
37 |
|
38 |
+
# On HF, Blender might be installed in a custom location
|
39 |
+
if IS_HUGGINGFACE:
|
40 |
+
# Check if Blender was installed via apt.txt
|
41 |
+
hf_paths = [
|
42 |
+
"/opt/conda/bin/blender",
|
43 |
+
"/home/user/blender/blender"
|
44 |
+
]
|
45 |
+
paths_to_check = hf_paths + paths_to_check
|
46 |
+
|
47 |
+
# On Windows systems
|
48 |
+
if platform.system() == "Windows":
|
49 |
+
program_files = os.environ.get("ProgramFiles", "C:\\Program Files")
|
50 |
+
paths_to_check.append(os.path.join(program_files, "Blender Foundation", "Blender", "blender.exe"))
|
51 |
+
|
52 |
for path in paths_to_check:
|
53 |
try:
|
54 |
+
logger.info(f"Checking Blender at: {path}")
|
55 |
result = subprocess.run([path, "--version"],
|
56 |
capture_output=True,
|
57 |
text=True,
|
58 |
+
timeout=10)
|
59 |
if result.returncode == 0:
|
60 |
st.success(f"Found Blender: {result.stdout.strip()}")
|
61 |
+
logger.info(f"Blender found at {path}: {result.stdout.strip()}")
|
62 |
return path
|
63 |
+
except Exception as e:
|
64 |
+
logger.debug(f"Failed to find Blender at {path}: {str(e)}")
|
65 |
continue
|
66 |
|
67 |
+
# If not found, try using the 'which' command on Unix-like systems
|
68 |
+
if platform.system() != "Windows":
|
69 |
+
try:
|
70 |
+
result = subprocess.run(["which", "blender"],
|
71 |
+
capture_output=True,
|
72 |
+
text=True,
|
73 |
+
timeout=5)
|
74 |
+
if result.returncode == 0 and result.stdout.strip():
|
75 |
+
path = result.stdout.strip()
|
76 |
+
st.success(f"Found Blender using 'which': {path}")
|
77 |
+
logger.info(f"Blender found using 'which': {path}")
|
78 |
+
return path
|
79 |
+
except Exception as e:
|
80 |
+
logger.debug(f"Failed to use 'which' to find Blender: {str(e)}")
|
81 |
+
|
82 |
return None
|
83 |
|
84 |
+
# Enhanced script preprocessor to ensure reliable .blend file saving
|
85 |
def preprocess_script(script_text, tmp_dir):
|
86 |
+
logger.info("Preprocessing script...")
|
87 |
+
|
88 |
+
# Check if necessary imports exist in the script
|
89 |
+
required_imports = {
|
90 |
+
'import bpy': 'import bpy',
|
91 |
+
'import os': 'import os',
|
92 |
+
'import math': 'import math' if 'math.' in script_text else None,
|
93 |
+
'import sys': 'import sys'
|
94 |
+
}
|
95 |
+
|
96 |
+
for import_check, import_statement in required_imports.items():
|
97 |
+
if import_statement and import_check not in script_text:
|
98 |
+
script_text = import_statement + '\n' + script_text
|
99 |
+
logger.info(f"Added {import_statement} to script")
|
100 |
|
101 |
+
# Ensure absolute paths for file operations in Hugging Face environment
|
102 |
+
absolute_tmp_dir = os.path.abspath(tmp_dir)
|
103 |
+
logger.info(f"Using absolute temp directory: {absolute_tmp_dir}")
|
104 |
|
105 |
+
# More robust file saving code with error handling
|
106 |
if 'bpy.ops.wm.save_as_mainfile' not in script_text:
|
107 |
+
# Add enhanced file saving code at the end with detailed error handling
|
108 |
script_text += f'''
|
109 |
|
110 |
+
# Save the .blend file with error handling
|
111 |
+
try:
|
112 |
+
# Ensure the output directory exists
|
113 |
+
output_dir = os.environ.get('BLENDER_OUTPUT_DIR', '{absolute_tmp_dir}')
|
114 |
+
os.makedirs(output_dir, exist_ok=True)
|
115 |
+
|
116 |
+
# Use absolute path for the blend file
|
117 |
+
blend_file_path = os.path.abspath(os.path.join(output_dir, "scene.blend"))
|
118 |
+
|
119 |
+
# Print debug info for troubleshooting
|
120 |
+
print(f"Attempting to save blend file to: {{blend_file_path}}")
|
121 |
+
print(f"Current working directory: {{os.getcwd()}}")
|
122 |
+
|
123 |
+
# Save the file with absolute path - using both filepath and check_existing parameters
|
124 |
+
bpy.ops.wm.save_as_mainfile(filepath=blend_file_path, check_existing=False)
|
125 |
+
|
126 |
+
# Verify the file was saved
|
127 |
+
if os.path.exists(blend_file_path):
|
128 |
+
print(f"Successfully saved blend file to: {{blend_file_path}}")
|
129 |
+
else:
|
130 |
+
print(f"ERROR: Failed to save blend file - file does not exist after save operation")
|
131 |
+
except Exception as e:
|
132 |
+
print(f"ERROR saving .blend file: {{str(e)}}")
|
133 |
+
import traceback
|
134 |
+
traceback.print_exc()
|
135 |
'''
|
136 |
+
logger.info("Added enhanced file saving code to script")
|
137 |
else:
|
138 |
+
# Make sure existing save code uses absolute paths correctly and has error handling
|
139 |
+
save_pattern = r'bpy\.ops\.wm\.save_as_mainfile\s*\(\s*filepath\s*=\s*["\']([^"\']+)["\']'
|
140 |
+
if re.search(save_pattern, script_text):
|
141 |
+
script_text = re.sub(
|
142 |
+
save_pattern,
|
143 |
+
f'bpy.ops.wm.save_as_mainfile(filepath=os.path.abspath(os.path.join(os.environ.get("BLENDER_OUTPUT_DIR", "{absolute_tmp_dir}"), "scene.blend")), check_existing=False',
|
144 |
+
script_text
|
145 |
+
)
|
146 |
+
logger.info("Modified existing save code to use absolute paths")
|
147 |
|
148 |
return script_text
|
149 |
|
|
|
151 |
blender_path = find_blender()
|
152 |
if not blender_path:
|
153 |
st.error("❌ Blender not found! The app requires Blender to be installed.")
|
154 |
+
if IS_HUGGINGFACE:
|
155 |
+
st.info("For Hugging Face Spaces, make sure to include 'blender' in your apt.txt file to install Blender.")
|
156 |
+
st.code("blender", language="text")
|
157 |
+
else:
|
158 |
+
st.info("Please install Blender and make sure it's in your PATH.")
|
159 |
# Continue anyway to show the interface
|
160 |
|
161 |
# Hardcoded default texture URLs (with fallbacks)
|
162 |
DEFAULT_TEXTURE_URLS = [
|
163 |
"https://eoimages.gsfc.nasa.gov/images/imagerecords/57000/57730/land_ocean_ice_2048.jpg",
|
164 |
+
"https://svs.gsfc.nasa.gov/vis/a000000/a002900/a002915/bluemarble-2048.png",
|
165 |
+
# Additional fallbacks
|
166 |
+
"https://upload.wikimedia.org/wikipedia/commons/thumb/c/cd/Blue_Marble_2007_East.jpg/600px-Blue_Marble_2007_East.jpg",
|
167 |
+
"https://www.solarsystemscope.com/textures/download/2k_earth_daymap.jpg"
|
168 |
]
|
169 |
|
170 |
# Sidebar: Blender script editor with example script
|
|
|
344 |
print('Solar system animation setup complete.')"""
|
345 |
}
|
346 |
|
347 |
+
# Add a simple script example that's lightweight for testing
|
348 |
+
script_examples["Simple Cube (Test)"] = """import bpy
|
349 |
+
|
350 |
+
# Clear the scene
|
351 |
+
bpy.ops.object.select_all(action='SELECT')
|
352 |
+
bpy.ops.object.delete()
|
353 |
+
|
354 |
+
# Create a cube
|
355 |
+
bpy.ops.mesh.primitive_cube_add(size=1, location=(0, 0, 0))
|
356 |
+
cube = bpy.context.active_object
|
357 |
+
cube.name = 'TestCube'
|
358 |
+
|
359 |
+
# Add animation
|
360 |
+
cube.rotation_euler = (0, 0, 0)
|
361 |
+
cube.keyframe_insert(data_path="rotation_euler", frame=1)
|
362 |
+
cube.rotation_euler = (0, math.radians(360), 0)
|
363 |
+
cube.keyframe_insert(data_path="rotation_euler", frame=100)
|
364 |
+
|
365 |
+
# Setup camera
|
366 |
+
bpy.ops.object.camera_add(location=(0, -3, 0))
|
367 |
+
camera = bpy.context.active_object
|
368 |
+
camera.rotation_euler = (math.radians(90), 0, 0)
|
369 |
+
bpy.context.scene.camera = camera
|
370 |
+
|
371 |
+
# Setup lighting
|
372 |
+
bpy.ops.object.light_add(type='SUN', location=(5, -5, 5))
|
373 |
+
"""
|
374 |
+
|
375 |
# Add example selector
|
376 |
selected_example = st.sidebar.selectbox(
|
377 |
"Load example script:",
|
|
|
403 |
height=100
|
404 |
)
|
405 |
|
406 |
+
# Add options specific to troubleshooting in Hugging Face
|
407 |
+
if IS_HUGGINGFACE:
|
408 |
+
st.sidebar.header("Hugging Face Options")
|
409 |
+
debug_mode = st.sidebar.checkbox("Enable Debug Mode", value=False)
|
410 |
+
if debug_mode:
|
411 |
+
log_level = st.sidebar.selectbox(
|
412 |
+
"Log Level",
|
413 |
+
options=["INFO", "DEBUG"],
|
414 |
+
index=0
|
415 |
+
)
|
416 |
+
if log_level == "DEBUG":
|
417 |
+
logger.setLevel(logging.DEBUG)
|
418 |
+
|
419 |
# Main action
|
420 |
if st.sidebar.button("Run & Export GLB"):
|
421 |
if not blender_path:
|
|
|
429 |
with st.spinner("Processing your 3D scene..."):
|
430 |
# Use a more robust temporary directory approach
|
431 |
tmp_dir = tempfile.mkdtemp(prefix="blender_app_")
|
432 |
+
logger.info(f"Created temporary directory: {tmp_dir}")
|
433 |
+
|
434 |
try:
|
435 |
# 1) Preprocess and save user script
|
436 |
processed_script = preprocess_script(script_text, tmp_dir)
|
437 |
script_path = os.path.join(tmp_dir, "user_script.py")
|
438 |
with open(script_path, "w") as f:
|
439 |
f.write(processed_script)
|
440 |
+
logger.info(f"Saved processed script to: {script_path}")
|
441 |
|
442 |
# 2) Collect textures
|
443 |
texture_paths = []
|
|
|
448 |
with open(path, "wb") as tf:
|
449 |
tf.write(upload.read())
|
450 |
texture_paths.append(path)
|
451 |
+
logger.info(f"Saved {len(texture_paths)} uploaded textures")
|
452 |
else:
|
453 |
# Try multiple default textures if some fail
|
454 |
for url in DEFAULT_TEXTURE_URLS:
|
455 |
try:
|
456 |
+
r = requests.get(url, timeout=15)
|
457 |
r.raise_for_status()
|
458 |
ext = os.path.splitext(url)[-1] or ".jpg"
|
459 |
path = os.path.join(tmp_dir, f"default{ext}")
|
460 |
with open(path, "wb") as tf:
|
461 |
tf.write(r.content)
|
462 |
texture_paths.append(path)
|
463 |
+
logger.info(f"Downloaded texture from {url}")
|
464 |
# If we got one texture successfully, that's enough
|
465 |
break
|
466 |
except Exception as e:
|
467 |
+
logger.warning(f"Could not download texture {url}: {str(e)}")
|
468 |
st.warning(f"Could not download texture {url}: {str(e)}")
|
469 |
|
470 |
# Display texture file information
|
|
|
472 |
st.subheader("Texture Files")
|
473 |
for tp in texture_paths:
|
474 |
st.code(os.path.basename(tp))
|
475 |
+
logger.info(f"Using texture: {tp}")
|
476 |
else:
|
477 |
st.warning("No textures were loaded. Your 3D model may appear without textures.")
|
478 |
+
logger.warning("No textures were loaded")
|
479 |
|
480 |
# 3) Install custom Python libraries if specified
|
481 |
if custom_packages and custom_packages.strip():
|
482 |
pkgs = [l.strip() for l in custom_packages.splitlines() if l.strip()]
|
483 |
if pkgs:
|
484 |
st.info(f"Installing: {', '.join(pkgs)}")
|
485 |
+
logger.info(f"Installing packages: {', '.join(pkgs)}")
|
486 |
try:
|
487 |
# Use Python executable from current environment for pip
|
488 |
pip_cmd = [sys.executable, "-m", "pip", "install", "--user"] + pkgs
|
|
|
491 |
check=True,
|
492 |
capture_output=True,
|
493 |
text=True,
|
494 |
+
timeout=180 # 3 minute timeout
|
495 |
)
|
496 |
st.text_area("pip install output", pip_res.stdout + pip_res.stderr, height=100)
|
497 |
+
logger.info(f"Pip install result: {pip_res.returncode}")
|
498 |
except Exception as e:
|
499 |
st.warning(f"pip install failed: {str(e)}")
|
500 |
+
logger.error(f"pip install failed: {str(e)}")
|
501 |
|
502 |
# 4) Prepare environment with necessary variables
|
503 |
env = os.environ.copy()
|
504 |
env["TEXTURE_PATHS"] = ",".join(texture_paths)
|
505 |
env["BLENDER_OUTPUT_DIR"] = tmp_dir # Pass the tmp_dir to the Blender script
|
506 |
+
logger.info(f"Set environment variable BLENDER_OUTPUT_DIR={tmp_dir}")
|
507 |
|
508 |
+
# 5) Run Blender to build .blend file with enhanced error handling and retry
|
509 |
blend_path = os.path.join(tmp_dir, "scene.blend")
|
510 |
|
511 |
+
max_retries = 3
|
512 |
+
for attempt in range(1, max_retries + 1):
|
513 |
+
with st.status(f"Running Blender to create scene (attempt {attempt}/{max_retries})...") as status:
|
514 |
+
cmd1 = [blender_path, "--background", "--python", script_path]
|
515 |
+
logger.info(f"Running Blender command: {' '.join(cmd1)}")
|
516 |
+
|
517 |
+
try:
|
518 |
+
r1 = subprocess.run(
|
519 |
+
cmd1,
|
520 |
+
cwd=tmp_dir,
|
521 |
+
env=env,
|
522 |
+
check=True,
|
523 |
+
capture_output=True,
|
524 |
+
text=True,
|
525 |
+
timeout=300 # 5 minute timeout
|
526 |
+
)
|
527 |
+
status.update(label=f"Blender scene created successfully (attempt {attempt})", state="complete")
|
528 |
+
st.text_area("Blender output", r1.stdout + r1.stderr, height=150)
|
529 |
+
logger.info("Blender script execution completed successfully")
|
530 |
+
break
|
531 |
+
except subprocess.TimeoutExpired:
|
532 |
+
st.error(f"Blender process timed out after 5 minutes (attempt {attempt}/{max_retries}).")
|
533 |
+
logger.error(f"Blender timeout on attempt {attempt}")
|
534 |
+
if attempt == max_retries:
|
535 |
+
st.stop()
|
536 |
+
except subprocess.CalledProcessError as e:
|
537 |
+
st.error(f"Blender build failed with error code {e.returncode} (attempt {attempt}/{max_retries})")
|
538 |
+
st.text_area("Error details", e.stdout + e.stderr, height=150)
|
539 |
+
logger.error(f"Blender error on attempt {attempt}: {e.returncode}")
|
540 |
+
logger.debug(f"Blender stderr: {e.stderr}")
|
541 |
+
|
542 |
+
# If it's the last attempt, stop execution
|
543 |
+
if attempt == max_retries:
|
544 |
+
st.stop()
|
545 |
+
except Exception as e:
|
546 |
+
st.error(f"Blender build failed: {str(e)} (attempt {attempt}/{max_retries})")
|
547 |
+
logger.error(f"Unexpected error on attempt {attempt}: {str(e)}")
|
548 |
+
if attempt == max_retries:
|
549 |
+
st.stop()
|
550 |
+
|
551 |
+
# Check if blend file was created with better diagnostics
|
552 |
+
if not os.path.exists(blend_path):
|
553 |
+
st.error("Blender did not create the expected scene.blend file.")
|
554 |
+
logger.error(f"Blend file not found at {blend_path}")
|
555 |
+
|
556 |
+
# Check directory contents for troubleshooting
|
557 |
+
dir_contents = os.listdir(tmp_dir)
|
558 |
+
logger.info(f"Temporary directory contents: {dir_contents}")
|
559 |
+
st.info(f"Temporary directory contents: {', '.join(dir_contents)}")
|
560 |
+
|
561 |
+
# Try an alternative approach by writing a minimal script just to save a file
|
562 |
+
st.warning("Attempting alternative approach to create .blend file...")
|
563 |
+
logger.info("Trying alternative approach with minimal script")
|
564 |
+
|
565 |
+
minimal_script_path = os.path.join(tmp_dir, "minimal_save.py")
|
566 |
+
with open(minimal_script_path, "w") as f:
|
567 |
+
f.write(f"""
|
568 |
+
import bpy
|
569 |
+
import os
|
570 |
+
|
571 |
+
# Make sure the directory exists
|
572 |
+
os.makedirs(r'{tmp_dir}', exist_ok=True)
|
573 |
+
|
574 |
+
# Create a simple cube
|
575 |
+
bpy.ops.mesh.primitive_cube_add(size=1, location=(0, 0, 0))
|
576 |
+
|
577 |
+
# Save the file
|
578 |
+
blend_path = r'{blend_path}'
|
579 |
+
print(f"Saving to {{blend_path}}")
|
580 |
+
bpy.ops.wm.save_as_mainfile(filepath=blend_path, check_existing=False)
|
581 |
+
print(f"Save operation completed, checking if file exists: {{os.path.exists(blend_path)}}")
|
582 |
+
""")
|
583 |
+
|
584 |
try:
|
585 |
+
cmd_min = [blender_path, "--background", "--python", minimal_script_path]
|
586 |
+
r_min = subprocess.run(
|
587 |
+
cmd_min,
|
588 |
cwd=tmp_dir,
|
589 |
env=env,
|
590 |
check=True,
|
591 |
capture_output=True,
|
592 |
text=True,
|
593 |
+
timeout=120
|
594 |
)
|
595 |
+
st.text_area("Minimal script output", r_min.stdout + r_min.stderr, height=150)
|
596 |
+
logger.info("Minimal script execution completed")
|
597 |
+
|
598 |
+
if os.path.exists(blend_path):
|
599 |
+
st.success("Alternative approach created a .blend file successfully!")
|
600 |
+
logger.info("Alternative approach succeeded")
|
601 |
+
else:
|
602 |
+
st.error("Alternative approach also failed to create .blend file.")
|
603 |
+
logger.error("Alternative approach failed")
|
604 |
+
st.stop()
|
605 |
except Exception as e:
|
606 |
+
st.error(f"Alternative approach failed: {str(e)}")
|
607 |
+
logger.error(f"Alternative approach error: {str(e)}")
|
608 |
st.stop()
|
|
|
|
|
|
|
|
|
|
|
|
|
609 |
|
610 |
+
# 6) Export GLB with animation - enhanced error handling
|
611 |
glb_path = os.path.join(tmp_dir, "animation.glb")
|
612 |
|
613 |
with st.status("Exporting to GLB format...") as status:
|
|
|
617 |
f.write(f"""
|
618 |
import bpy
|
619 |
import os
|
620 |
+
import sys
|
621 |
|
622 |
+
# Print debug info
|
623 |
+
print(f"Python version: {{sys.version}}")
|
624 |
+
print(f"Blender version: {{bpy.app.version_string}}")
|
625 |
+
print(f"Attempting to load blend file: {blend_path}")
|
626 |
+
print(f"Current working directory: {{os.getcwd()}}")
|
627 |
+
print(f"File exists check: {{os.path.exists(r'{blend_path}')}}")
|
628 |
+
|
629 |
+
try:
|
630 |
+
# Load the blend file
|
631 |
+
bpy.ops.wm.open_mainfile(filepath=r'{blend_path}')
|
632 |
+
print("Blend file loaded successfully")
|
633 |
+
|
634 |
+
# Export to GLB
|
635 |
+
glb_path = r'{glb_path}'
|
636 |
+
print(f"Exporting to GLB: {{glb_path}}")
|
637 |
+
|
638 |
+
# Check if glTF export is available
|
639 |
+
if hasattr(bpy.ops.export_scene, 'gltf'):
|
640 |
+
bpy.ops.export_scene.gltf(
|
641 |
+
filepath=glb_path,
|
642 |
+
export_format='GLB',
|
643 |
+
export_animations=True
|
644 |
+
)
|
645 |
+
print(f"GLB export completed, file exists: {{os.path.exists(glb_path)}}")
|
646 |
+
else:
|
647 |
+
print("ERROR: glTF export operator not available. Check if the add-on is enabled.")
|
648 |
+
# Try to enable the addon
|
649 |
+
bpy.ops.preferences.addon_enable(module='io_scene_gltf2')
|
650 |
+
print("Attempted to enable glTF add-on")
|
651 |
+
|
652 |
+
# Try export again
|
653 |
+
if hasattr(bpy.ops.export_scene, 'gltf'):
|
654 |
+
bpy.ops.export_scene.gltf(
|
655 |
+
filepath=glb_path,
|
656 |
+
export_format='GLB',
|
657 |
+
export_animations=True
|
658 |
+
)
|
659 |
+
print(f"GLB export completed after enabling add-on")
|
660 |
+
else:
|
661 |
+
print("ERROR: Could not enable glTF export add-on")
|
662 |
+
except Exception as e:
|
663 |
+
print(f"ERROR during export: {{str(e)}}")
|
664 |
+
import traceback
|
665 |
+
traceback.print_exc()
|
666 |
""")
|
667 |
|
668 |
cmd2 = [blender_path, "--background", "--python", export_script]
|
669 |
+
logger.info(f"Running GLB export command: {' '.join(cmd2)}")
|
670 |
+
|
671 |
try:
|
672 |
r2 = subprocess.run(
|
673 |
cmd2,
|
|
|
676 |
check=True,
|
677 |
capture_output=True,
|
678 |
text=True,
|
679 |
+
timeout=180 # 3 minute timeout
|
680 |
)
|
681 |
status.update(label="GLB export completed successfully", state="complete")
|
682 |
st.text_area("Export output", r2.stdout + r2.stderr, height=100)
|
683 |
+
logger.info("GLB export completed successfully")
|
684 |
except Exception as e:
|
685 |
st.error(f"GLB export failed: {str(e)}")
|
686 |
+
logger.error(f"GLB export error: {str(e)}")
|
687 |
st.stop()
|
688 |
|
689 |
# 7) Embed GLB inline if it exists
|
|
|
693 |
|
694 |
file_size = len(data) / (1024 * 1024) # Size in MB
|
695 |
st.success(f"Successfully created GLB file ({file_size:.1f} MB)")
|
696 |
+
logger.info(f"Created GLB file, size: {file_size:.1f} MB")
|
697 |
|
698 |
# Check if file isn't too large for embedding
|
699 |
if file_size > 50:
|
700 |
st.warning("The GLB file is quite large. The viewer might be slow to load.")
|
701 |
+
logger.warning(f"Large GLB file: {file_size:.1f} MB")
|
702 |
|
703 |
b64 = base64.b64encode(data).decode()
|
704 |
|
|
|
727 |
)
|
728 |
else:
|
729 |
st.error("GLB file was not generated successfully.")
|
730 |
+
logger.error(f"GLB file not found at {glb_path}")
|
731 |
+
|
732 |
if os.path.exists(blend_path):
|
733 |
st.info("The Blender file was created, but the GLB export failed.")
|
734 |
+
logger.info("Blend file exists but GLB export failed")
|
735 |
|
736 |
finally:
|
737 |
+
# Clean up - remove temporary directory if not in debug mode
|
738 |
+
if IS_HUGGINGFACE and debug_mode:
|
739 |
+
st.info(f"Debug mode: Temporary files kept at {tmp_dir}")
|
740 |
+
logger.info(f"Keeping temp files for debugging: {tmp_dir}")
|
741 |
+
else:
|
742 |
+
try:
|
743 |
+
shutil.rmtree(tmp_dir, ignore_errors=True)
|
744 |
+
logger.info(f"Cleaned up temporary directory: {tmp_dir}")
|
745 |
+
except Exception as e:
|
746 |
+
st.warning(f"Failed to clean up temporary files: {str(e)}")
|
747 |
+
logger.error(f"Cleanup error: {str(e)}")
|
748 |
|
749 |
# Add helpful information at the bottom
|
750 |
st.divider()
|
|
|
757 |
- Generate and visualize 3D models with animations
|
758 |
- Download the results as GLB files for use in other applications
|
759 |
|
760 |
+
The app automatically adds necessary imports and file saving code to your scripts,
|
761 |
so you can focus on creating 3D content without worrying about environment details.
|
762 |
+
|
763 |
+
Running on Hugging Face Spaces: This app requires Blender to be installed. Make sure
|
764 |
+
your Space includes 'blender' in the apt.txt file.
|
765 |
""")
|
766 |
|
767 |
with st.expander("Script Tips"):
|
|
|
788 |
obj.rotation_euler = (0, 0, math.radians(360))
|
789 |
obj.keyframe_insert(data_path="rotation_euler", frame=250)
|
790 |
```
|
791 |
+
|
792 |
+
4. **Troubleshooting in Hugging Face**: If you're having issues with the app in Hugging Face Spaces:
|
793 |
+
|
794 |
+
- Enable Debug Mode in the sidebar
|
795 |
+
- Check the error messages in the app output
|
796 |
+
- Make sure your apt.txt file includes 'blender'
|
797 |
+
- Try the "Simple Cube (Test)" example first to verify Blender is working
|
798 |
+
""")
|
799 |
+
|
800 |
+
# Add a section at the bottom for Hugging Face Spaces setup
|
801 |
+
with st.expander("Hugging Face Spaces Setup"):
|
802 |
+
st.markdown("""
|
803 |
+
### Setting up this app on Hugging Face Spaces
|
804 |
+
|
805 |
+
To ensure this app works correctly on Hugging Face:
|
806 |
+
|
807 |
+
1. **Create an apt.txt file** in your repository with:
|
808 |
+
```
|
809 |
+
blender
|
810 |
+
```
|
811 |
+
|
812 |
+
2. **Create a requirements.txt file** with:
|
813 |
+
```
|
814 |
+
streamlit
|
815 |
+
streamlit-ace
|
816 |
+
requests
|
817 |
+
```
|
818 |
+
|
819 |
+
3. **Set the Space SDK** to "Streamlit" in the Space settings
|
820 |
+
|
821 |
+
4. **Set resource allocation** to at least 2 CPU + 4GB RAM for better performance
|
822 |
+
|
823 |
+
Troubleshooting: If you encounter "Blender not found" errors, check your apt.txt file and make sure Blender is being installed correctly.
|
824 |
""")
|