Spaces:
Running
Running
import bpy | |
import gradio as gr | |
import math | |
import os | |
import random | |
from diffusers.utils import load_image | |
from PIL import Image | |
import PIL.Image | |
import tqdm | |
os.putenv("XDG_RUNTIME_DIR", '.') | |
os.environ['XDG_RUNTIME_DIR'] = '.' | |
fin_rend = './rendered_scene.png' | |
fin_blend = './exp_scene.blend' | |
fin_glb = './exp_scene.glb' | |
fin_anim = './animation.mp4' | |
img = bpy.data.images.load('./STScI-01G8H1NK4W8CJYHF2DDFD1W0DQ.png') | |
earth_texture = bpy.data.images.load('./ear0xuu2.jpg') | |
mars_texture = bpy.data.images.load('mar0kuu2.jpg') | |
apol=[] | |
def add_planet(name, radius, mass, distance_from_sun): | |
material = bpy.data.materials.new(name="Mymarsmat"+str(random.randint(1,36))) | |
material.use_nodes = True | |
image_texture_node = material.node_tree.nodes.new('ShaderNodeTexImage') | |
if name=="mars": | |
image_texture_node.image = bpy.data.images.load("mar0kuu2.jpg") | |
else: | |
image_texture_node.image = bpy.data.images.load("ear0xuu2.jpg") | |
bsdf_node = material.node_tree.nodes.get('Principled BSDF') | |
material.node_tree.links.new(bsdf_node.inputs['Base Color'], image_texture_node.outputs['Color']) | |
bpy.ops.mesh.primitive_uv_sphere_add(radius=radius, location=(distance_from_sun * random.uniform(-8, 8), random.uniform(-8, 8), random.uniform(-8, 3))) | |
planet = bpy.context.active_object | |
planet.name = name | |
planet.data.materials.append(material) | |
planet["mass"] = mass | |
return planet | |
def visualize_planets(): | |
bpy.ops.object.select_all(action='DESELECT') | |
for obj in bpy.context.scene.objects: | |
if obj.type == 'MESH' and "mass" in obj: | |
obj.select_set(True) | |
##for area in bpy.context.screen.areas: | |
## if area.type == 'VIEW_3D': | |
## override = bpy.context.copy() | |
## override['area'] = area | |
## bpy.ops.view3d.snap_cursor_to_selected(override) | |
bpy.ops.view3d.snap_cursor_to_selected() | |
##bpy.ops.object.origin_set(type='ORIGIN_CURSOR') | |
##bpy.context.scene.cursor.location = (0, 0, 0) | |
def create_planet_flyby_animation(planet, radius, duration): | |
bpy.context.view_layer.objects.active = planet | |
planet.select_set(True) | |
planet.animation_data_create() | |
planet.animation_data.action = bpy.data.actions.new(name="OrbitAction") | |
start_frame = 1 | |
end_frame = int(duration * 10) | |
for frame in range(start_frame, end_frame + 1): | |
angle = (frame / end_frame) * 2 * math.pi | |
x = radius * math.cos(angle) | |
y = radius * math.sin(angle) | |
z = planet.location.z | |
planet.location = (x, y, z) | |
planet.keyframe_insert(data_path="location", index=-1, frame=frame) | |
def add_animated_space_rocks(): | |
bpy.context.scene.render.fps = 10 | |
duration = 2 | |
for i in range(30): | |
bpy.ops.mesh.primitive_uv_sphere_add(radius=random.uniform(0.01, 0.1), location=(random.uniform(-10, 10), random.uniform(-10, 10), random.uniform(-4, 9))) | |
space_rock = bpy.context.active_object | |
space_rock.keyframe_insert(data_path="location", frame=1) | |
space_rock.location = (random.uniform(-10, 10), random.uniform(-10, 10), random.uniform(-10, 10)) | |
space_rock.keyframe_insert(data_path="location", frame=duration * bpy.context.scene.render.fps) | |
def create_particle_effects_animation(): | |
bpy.context.scene.render.fps = 10 | |
duration = 2 | |
bpy.ops.mesh.primitive_uv_sphere_add(radius=random.uniform(0.05, 0.10), location=(random.uniform(-1, 1), random.uniform(-1, 1), 1)) | |
emitter = bpy.context.active_object | |
emitter.name = "ParticleEmitter" | |
particle_system = emitter.modifiers.new(name="ParticleSettings", type='PARTICLE_SYSTEM') | |
particle_system.particle_system.settings.count = 30 | |
particle_system.particle_system.settings.frame_start = 1 | |
particle_system.particle_system.settings.frame_end = duration * bpy.context.scene.render.fps | |
particle_system.particle_system.settings.render_type = 'OBJECT' | |
particle_system.particle_system.settings.instance_object = bpy.data.objects["ParticleOne"] | |
emitter.keyframe_insert(data_path="location", index=-1, frame=1) | |
emitter.location = (random.uniform(-10, 10), random.uniform(-10, 10), random.uniform(-10, 10)) | |
emitter.keyframe_insert(data_path="location", index=-1, frame=duration * bpy.context.scene.render.fps) | |
def create_meteor_with_particle_system(): | |
bpy.ops.mesh.primitive_ico_sphere_add(subdivisions=4, radius=0.25, location=(random.uniform(-10, 10), random.uniform(-10, 10), random.uniform(-10, 10))) | |
meteor = bpy.context.active_object | |
meteor.name = "Meteor" | |
meteor_material = bpy.data.materials.new(name="MeteorMaterial") | |
meteor.data.materials.append(meteor_material) | |
meteor.active_material.use_nodes = True | |
nodes = meteor.active_material.node_tree.nodes | |
emission_node = nodes.new(type='ShaderNodeEmission') | |
emission_node.inputs[0].default_value = (0.0, 0.0, 1.0, 1) # Blue color | |
output_node = nodes.get("Material Output") | |
material_link = meteor.active_material.node_tree.links.new | |
material_link(emission_node.outputs[0], output_node.inputs[0]) | |
bpy.ops.mesh.primitive_uv_sphere_add(radius=random.uniform(0.01, 0.04), location=(random.uniform(-10, 10), random.uniform(-10, 10), random.uniform(-10, 10))) | |
partone = bpy.context.active_object | |
partone.name = "ParticleOne" | |
particle_settings = meteor.modifiers.new(name="EnergySpheres", type='PARTICLE_SYSTEM') | |
particle_settings.particle_system.settings.count = 100 | |
particle_settings.particle_system.settings.lifetime = random.uniform(10, 20) | |
particle_settings.particle_system.settings.frame_start = 1 | |
particle_settings.particle_system.settings.frame_end = 20 | |
particle_settings.particle_system.settings.use_emit_random = True | |
particle_settings.particle_system.settings.render_type = 'OBJECT' | |
particle_settings.particle_system.settings.instance_object = bpy.data.objects["ParticleOne"] | |
meteor.keyframe_insert(data_path="location", index=-1, frame=1) | |
meteor.location = (random.uniform(-10, 10), random.uniform(-10, 10), random.uniform(-10, 10)) | |
meteor.keyframe_insert(data_path="location", index=-1, frame=20) | |
def create_scene(): | |
scene = bpy.context.scene | |
render = scene.render | |
render.resolution_x = 1920 | |
render.resolution_y = 1080 | |
render.engine = 'BLENDER_EEVEE' | |
##world = bpy.context.scene.world | |
##bg_node = world.node_tree.nodes['Background'] | |
##bg_node.image = img | |
##env_data = bpy.data.worlds['World'].node_tree.nodes['Background'].input | |
##env_data.default_value = img | |
##tex_environment = bpy.data.worlds['World'].node_tree.nodes.new('ShaderNodeTexEnvironment') | |
##tex_environment.image = img | |
##world = bpy.context.scene.world | |
##bg_node = world.node_tree.nodes['Background'] | |
##world.node_tree.links.new(tex_environment.outputs[0], bg_node.inputs[0]) | |
##img = bpy.data.images.load(image_path) | |
world = bpy.context.scene.world | |
if not world.use_nodes: | |
world.use_nodes = True | |
##tree = world.node_tree | |
##bg_node = tree.nodes.new(type='ShaderNodeTexEnvironment') | |
##bg_node.image = img | |
##tree.links.new(bg_node.outputs[0], tree.nodes['Background'].inputs[0]) | |
bg_node = world.node_tree.nodes.get('Background') | |
bg_node.inputs['Color'].default_value = (0, 0, 0, 1) | |
bpy.data.objects['Cube'].select_set(True) | |
bpy.ops.object.delete() | |
bpy.data.objects['Camera'].select_set(True) | |
bpy.ops.object.delete() | |
light_data = bpy.data.lights.new(name="NewLight", type='SUN') | |
light_data.energy = 30 | |
light_object = bpy.data.objects.new(name="NewLight", object_data=light_data) | |
scene.collection.objects.link(light_object) | |
light_object.location = (3.0, 5.0, 11.0) | |
camera_data = bpy.data.cameras.new(name='Camera') | |
camera = bpy.data.objects.new('Camera', camera_data) | |
bpy.context.collection.objects.link(camera) | |
bpy.context.scene.camera = camera | |
camera.location = (0, 0, 10) | |
camera.rotation_euler = (0, 0, 0) | |
##planet1 = add_planet("earth", 2.5, 1, 0) | |
##planet2 = add_planet("mars", 1.75, 0.5, 1.5) | |
add_planet("earth", 2.5, 1, 0) | |
add_planet("mars", 1.75, 0.5, 1.5) | |
add_animated_space_rocks() | |
create_meteor_with_particle_system() | |
create_particle_effects_animation() | |
##create_planet_flyby_animation(planet1, 4, 2) | |
##visualize_planets() | |
create_scene() | |
def export_blend(): | |
bpy.ops.wm.save_as_mainfile(filepath='./exp_scene.blend') | |
last_created_time = 0 | |
last_created_file = None | |
apol=[] | |
for root, dirs, files in os.walk('.'): | |
for file in files: | |
file_path = os.path.join(root, file) | |
created_time = os.path.getctime(file_path) | |
if created_time > last_created_time: | |
last_created_time = created_time | |
last_created_file = file_path | |
apol.append(last_created_file) | |
return last_created_file | |
def render_scene(sny,progress=gr.Progress(track_tqdm=True)): | |
bpy.context.scene.render.filepath = './rendered_scene.png' | |
bpy.ops.render.render(animation=False, write_still=True) | |
last_created_time = 0 | |
last_created_file = None | |
apol=[] | |
for root, dirs, files in os.walk('.'): | |
for file in files: | |
file_path = os.path.join(root, file) | |
created_time = os.path.getctime(file_path) | |
if created_time > last_created_time: | |
last_created_time = created_time | |
last_created_file = file_path | |
imoge = load_image(last_created_file).convert('RGB') | |
apol.append(imoge) | |
moog = gr.Model3D(value=sny) | |
return apol, sny, moog | |
def export_glb(): | |
bpy.ops.export_scene.gltf(filepath='./exp_scene.glb', export_format='GLB')##, export_image_format='AUTO', export_materials='EXPORT', export_cameras=True, export_lights=True) | |
last_created_time = 0 | |
last_created_file = None | |
apol=[] | |
for root, dirs, files in os.walk('.'): | |
for file in files: | |
file_path = os.path.join(root, file) | |
created_time = os.path.getctime(file_path) | |
if created_time > last_created_time: | |
last_created_time = created_time | |
last_created_file = file_path | |
apol.append(last_created_file) | |
return last_created_file | |
def render_anim(sny,progress=gr.Progress(track_tqdm=True)): | |
bpy.context.scene.render.image_settings.file_format = 'FFMPEG' | |
bpy.context.scene.render.ffmpeg.format = 'MPEG4' | |
bpy.context.scene.render.ffmpeg.codec = 'H264' | |
bpy.context.scene.render.ffmpeg.constant_rate_factor = 'HIGH' | |
bpy.context.scene.frame_start = 1 | |
bpy.context.scene.frame_end = 20 | |
bpy.context.scene.render.filepath = './animation.mp4' | |
bpy.ops.render.render(animation=True) | |
last_created_time = 0 | |
last_created_file = None | |
apol=[] | |
for root, dirs, files in os.walk('.'): | |
for file in files: | |
file_path = os.path.join(root, file) | |
created_time = os.path.getctime(file_path) | |
if created_time > last_created_time: | |
last_created_time = created_time | |
last_created_file = file_path | |
apol.append(last_created_file) | |
moog = gr.Model3D(value=sny) | |
return apol[0], sny, moog | |
with gr.Blocks() as iface: | |
with gr.Column(): | |
moog = gr.Model3D() | |
ony = gr.Button("render Image") | |
twy = gr.Button("render Animation") | |
rani = gr.Video() | |
rdr = gr.Gallery(columns=1) | |
sny = gr.File(value=export_glb(),interactive=False) | |
siy = gr.File(value=export_blend(),interactive=False) | |
ony.click(fn=render_scene, inputs=[sny], outputs=[rdr,sny,moog]) | |
twy.click(fn=render_anim, inputs=[sny], outputs=[rani,sny,moog]) | |
iface.queue(max_size=1,api_open=False) | |
iface.launch(max_threads=20,inline=False,show_api=False) |