app.py

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)
    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])

    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)
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').resize((512, 512))
    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(render_scene, [sny], [rdr,sny,moog])
    twy.click(render_anim, [sny], [rani,sny,moog])

iface.launch()