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=[] | |
| print(bpy.app.version_string) | |
| 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 | |
| bpy.context.view_layer.objects.active = planet | |
| planet.select_set(True) | |
| planet.animation_data_create() | |
| planet.animation_data.action = bpy.data.actions.new(name="RotationAction_"+planet.name) | |
| start_frame = 1 | |
| duration=2 | |
| end_frame = int(duration * 10) | |
| planet.rotation_mode = 'XYZ' | |
| planet.keyframe_insert(data_path="rotation_euler", index=2, frame=start_frame) | |
| for frame in range(start_frame, end_frame + 1): | |
| ##angle = (frame / end_frame) * 2 * math.pi | |
| ##x = radius * math.cos(angle) | |
| ##y = planet.location.y | |
| ##z = radius * math.sin(angle) | |
| ##planet.location = (x, y, z) | |
| ##planet.keyframe_insert(data_path="location", index=-1, frame=frame) | |
| angle = math.radians(frame * 18) | |
| planet.rotation_euler = (0, 0, angle) | |
| planet.keyframe_insert(data_path="rotation_euler", index=-1, frame=frame) | |
| ##return planet | |
| 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 = planet.location.y | |
| z = radius * math.sin(angle) | |
| planet.location = (x, y, z) | |
| planet.keyframe_insert(data_path="location", index=-1, frame=frame) | |
| def add_animated_space_rocks(name): | |
| bpy.context.scene.render.fps = 10 | |
| duration = 2 | |
| radius=2 | |
| bpy.ops.mesh.primitive_uv_sphere_add(radius=random.uniform(0.03, 0.1), location=(random.uniform(-10, 10), random.uniform(-10, 10), random.uniform(-4, 9))) | |
| space_rock = bpy.context.active_object | |
| space_rock.name = ""+name+"_"+str(random.randint(1,100))+"" | |
| 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=20) | |
| bpy.context.view_layer.objects.active = space_rock | |
| space_rock.select_set(True) | |
| def create_particle_effects_animation(): | |
| ##wip | |
| 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=20) | |
| def create_meteor_with_particle_system(): | |
| ##wip | |
| 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]) | |
| meteor.location = (random.uniform(-10, 10), random.uniform(-10, 10), random.uniform(-10, 10)) | |
| bpy.context.view_layer.objects.active = meteor | |
| meteor.select_set(True) | |
| meteor.animation_data_create() | |
| meteor.animation_data.action = bpy.data.actions.new(name="OrbitAction") | |
| start_frame = 1 | |
| duration=2 | |
| radius=3 | |
| 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 = meteor.location.z | |
| meteor.location = (x, y, z) | |
| meteor.keyframe_insert(data_path="location", index=-1, frame=frame) | |
| ###meteor.keyframe_insert(data_path="location", frame=1) | |
| ##meteor.location = (random.uniform(-10, 10), random.uniform(-10, 10), random.uniform(-10, 10)) | |
| ##meteor.keyframe_insert(data_path="location", frame=20) | |
| def create_scene(): | |
| world = bpy.context.scene.world | |
| if not world.use_nodes: | |
| world.use_nodes = True | |
| ##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 | |
| bg_node = world.node_tree.nodes.get('Background') | |
| bg_node.inputs['Color'].default_value = (0, 0, 0, 1) | |
| ##tex_environment = world.node_tree.nodes.new('ShaderNodeTexEnvironment') | |
| ##tex_environment.image = img | |
| ##world = bpy.context.scene.world | |
| ##tx_node = world.node_tree.nodes['Background'] | |
| ##world.node_tree.links.new(tex_environment.outputs[0], tx_node.inputs[0]) | |
| ##img = bpy.data.images.load(image_path) | |
| 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) | |
| for i in range(30): | |
| add_animated_space_rocks("spacerock") | |
| create_meteor_with_particle_system() | |
| ##create_particle_effects_animation() | |
| ##create_planet_flyby_animation(planet1, 4, 2) | |
| ##visualize_planets() | |
| scene = bpy.context.scene | |
| render = scene.render | |
| render.resolution_x = 1920 | |
| render.resolution_y = 1080 | |
| render.engine = 'BLENDER_EEVEE' | |
| 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) |