Rename app (31).py to app.py

app (31).py → app.py

@@ -6,15 +6,12 @@ import random
 from diffusers.utils import load_image
 from PIL import Image
 import PIL.Image
-##from io_scene_gltf2.blender.exp import gltf2_blender_export
+import tqdm
 
 os.putenv("XDG_RUNTIME_DIR", '.')
 fin_rend = './rendered_scene.png'
-fin_pth = './path_to_exp.blend'
 fin_blend = './exp_scene.blend'
-fin_obj = './exp_scene.obj'
 fin_glb = './exp_scene.glb'
-fin_gltf = './exp_scene.gltf'
 fin_anim = './animation.mp4'
 img = bpy.data.images.load('./STScI-01G8H1NK4W8CJYHF2DDFD1W0DQ.png')
 earth_texture = bpy.data.images.load('./ear0xuu2.jpg')
@@ -23,28 +20,17 @@ apol=[]
 
 def add_planet(name, radius, mass, distance_from_sun):
     material = bpy.data.materials.new(name="Mymarsmat"+str(random.randint(1,36)))
-    ##material = bpy.data.materials.new(name="MyMaterial")
     material.use_nodes = True
-
-    ##for node in material.node_tree.nodes:
-    ##    material.node_tree.nodes.remove(node)
-
-
     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")
-
-    # Link the image texture node to the principled shader
     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_material = bpy.data.materials.new(name=name + "_texture")
     planet.data.materials.append(material)
     planet["mass"] = mass
     return planet
@@ -58,8 +44,6 @@ def visualize_planets():
     bpy.ops.object.origin_set(type='ORIGIN_CURSOR')
     bpy.context.scene.cursor.location = (0, 0, 0)
 
-##visualize_planets()
-
 def create_planet_flyby_animation(planet, radius, duration):
     bpy.context.view_layer.objects.active = planet
     planet.select_set(True)
@@ -71,12 +55,10 @@ def create_planet_flyby_animation(planet, radius, duration):
         angle = (frame / end_frame) * 2 * math.pi
         x = radius * math.cos(angle)
         y = radius * math.sin(angle)
-        z = planet.location.z  # Maintain the planet's z-coordinate
+        z = planet.location.z
         planet.location = (x, y, z)
         planet.keyframe_insert(data_path="location", index=-1, frame=frame)
 
-##create_planet_flyby_animation(planet, radius, 2)
-
 def add_animated_space_rocks():
     bpy.context.scene.render.fps = 10
     duration = 2
@@ -99,18 +81,13 @@ def create_particle_effects_animation():
     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
-
-    # Set the particle system to be rendered as an object
     particle_system.particle_system.settings.render_type = 'OBJECT'
     particle_system.particle_system.settings.instance_object = bpy.data.objects["ParticleOne"]
-
-    # Animate the emitter's location, rotation, and scale
     emitter.keyframe_insert(data_path="location", index=-1, frame=1)
-    emitter.location = (random.uniform(-10, 10), random.uniform(-10, 10), random.uniform(-10, 10))  # Set the new location for the end of the animation
+    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():
-    # Create a mesh object to serve as the meteor
     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"
@@ -170,12 +147,6 @@ def create_scene():
     bg_node.image = img
     tree.links.new(bg_node.outputs[0], tree.nodes['Background'].inputs[0])
 
-    ##world.light_settings.use_ambient_occlusion = True
-    ##world.light_settings.ao_factor = 0.5
-    ##world.light_settings.ao_factor = 0.5
-    ##world.light_settings.use_environment_light = True
-    ##world.light_settings.environment_energy = 0.5
-
     bpy.data.objects['Cube'].select_set(True)
     bpy.ops.object.delete()
     bpy.data.objects['Camera'].select_set(True)
@@ -197,9 +168,6 @@ def create_scene():
     ##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)
-    ##bpy.context.active_object=None
-
-
 
     add_animated_space_rocks()
     create_meteor_with_particle_system()
@@ -208,14 +176,7 @@ def create_scene():
 create_scene()
 
 def export_blend():
-    ##bpy.ops.wm.save_mainfile(filepath='./path_to_exp.blend')
     bpy.ops.wm.save_as_mainfile(filepath='./exp_scene.blend')
-    ##bpy.ops.export_scene.obj(filepath='./exp_scene.obj', export_format='OBJ', export_materials='EXPORT')
-    ##bpy.ops.export_scene.gltf(filepath='./exp_scene.glb', export_format='GLB', check_existing=False, export_image_format='AUTO', export_materials='EXPORT', export_cameras=True, export_lights=True)
-    ##bpy.ops.export_scene.gltf(filepath='./exp_scene.gltf', export_selected=False)
-    ##cntxt = bpy.context
-    ##bpy.ops.export_scene.gltf(filepath='./exp_scene.glb', export_format='GLB')
-    ##bpy.ops.export_scene.gltf(cntxt, RenderSettings)
     last_created_time = 0
     last_created_file = None
     apol=[]
@@ -228,21 +189,12 @@ def export_blend():
                 last_created_time = created_time
                 last_created_file = file_path
 
-    ##imoge = load_image(last_created_file).convert('RGB').resize((512, 512))
     apol.append(last_created_file)
-    ##return last_created_file
-    ##return fin_rend
-    ##return apol
     return last_created_file
 
-def render_scene(blk,sny):
+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)
-    ##bpy.ops.render.opengl(animation=False, render_keyed_only=False, sequencer=False, write_still=True, view_context=False)
-    ##imoge = load_image(fin_rend)
-    ##imoge.save('./rendered_scene.png', 'PNG')
-    ##imoge = PIL.Image(fin_rend)
-    ##imoge = load_image().convert('RGB').resize((512, 512))
     last_created_time = 0
     last_created_file = None
     apol=[]
@@ -257,93 +209,11 @@ def render_scene(blk,sny):
 
     imoge = load_image(last_created_file).convert('RGB').resize((512, 512))
     apol.append(imoge)
-    ##blk=gr.HTML(value="""<html><head><title>Three.js Blender Scene Viewer</title><style>body { margin: 0; } canvas { display: block; } </style></head><body><script src='https://threejs.org/build/three.js'></script><script src='https://threejs.org/examples/jsm/loaders/GLTFLoader.js'></script><script> var scene, camera, renderer; function init() { scene = new THREE.Scene(); camera = new THREE.PerspectiveCamera(75, window.innerWidth / window.innerHeight, 0.1, 1000); camera.position.z = 5; renderer = new THREE.WebGLRenderer({antialias: true}); renderer.setSize(window.innerWidth, window.innerHeight); document.body.appendChild(renderer.domElement); var loader = new THREE.GLTFLoader(); loader.load('"""+sny+"""', function(gltf) { scene.add(gltf.scene); }, undefined, function(error) { console.error(error); }); }; function animate(){ requestAnimationFrame(animate); renderer.render(scene, camera); }; init(); animate();</script></body></html>""")
-    pckg = f"/file={sny}"
     moog = gr.Model3D(value=sny)
-    return apol, f"""<iframe src="file=frame.html" width="100%" height="200px" data-glb="exp_scene.glb"></iframe>""", sny, moog
-    
-    ##blk=gr.HTML(value=html)
-    ##return last_created_file
-    ##return fin_rend
-    ##return apol,blk,sny
-
-def export_obj():
-    ##bpy.ops.wm.save_mainfile(filepath='./path_to_exp.blend')
-    ##bpy.ops.wm.save_as_mainfile(filepath='./exp_scene.blend')
-    ##bpy.ops.export_scene.obj(filepath='./exp_scene.obj', export_format='OBJ', export_materials='EXPORT')
-    bpy.ops.export_scene.gltf(filepath="./exp_scene.obj", export_format='OBJ', export_selected=False, export_apply=False, export_texcoords='TANGENT', export_normals='FACE', export_materials='EXPORT', export_colors='EXPORT', export_cameras='EXPORT', export_lights='EXPORT', export_displacement='NONE', export_animations=True, export_frame_range=True, export_frame_step=1, export_force_sampling=False, export_current_frame=False, export_extras=True)
-    ##bpy.ops.export_scene.gltf(filepath='./exp_scene.glb', export_format='GLB', check_existing=False, export_image_format='AUTO', export_materials='EXPORT', export_cameras=True, export_lights=True)
-    ##bpy.ops.export_scene.gltf(filepath='./exp_scene.gltf', export_selected=False)
-    ##cntxt = bpy.context
-    ##bpy.ops.export_scene.gltf(filepath='./exp_scene.glb', export_format='GLB')
-    ##bpy.ops.export_scene.gltf(cntxt, RenderSettings)
-    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(last_created_file)
-    ##return last_created_file
-    ##return fin_rend
-    return apol
+    return apol, sny, moog
 
 def export_glb():
-    ##bpy.ops.wm.save_mainfile(filepath='./path_to_exp.blend')
-    ##bpy.ops.wm.save_as_mainfile(filepath='./exp_scene.blend')
-    ##bpy.ops.export_scene.obj(filepath='./exp_scene.obj', export_format='OBJ', export_materials='EXPORT')
-    ##for ob in bpy.context.scene.objects:
-    ##    ob.select_set(False)
-    ##bpy.ops.export_scene.gltf(filepath='./exp_scene.glb', export_format='GLB', export_image_format='NONE', export_materials='NONE', export_cameras=True, export_lights=True)
     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)
-    ##export_settings = {
-    ##'filepath': './exp_scene.glb',
-    ##'export_format': 'GLB',
-    ##'gltf_current_frame': 1,
-    ##}
-
-    ##gltf2_blender_export.save(bpy.context, export_settings)
-    ##bpy.ops.export_scene.gltf(filepath="./exp_scene.glb", export_format='GLB', export_selected=False, export_apply=False, export_texcoords=True, export_normals=True, export_materials=True, export_colors=True, export_cameras=True, export_lights=True, export_displacement=False, export_animations=True, export_frame_range=True, export_frame_step=1, export_force_sampling=False, export_current_frame=False, export_extras=True)
-    ##bpy.ops.export_scene.gltf(filepath='./exp_scene.gltf', export_selected=False)
-    ##cntxt = bpy.context
-    ##bpy.ops.export_scene.gltf(filepath='./exp_scene.glb', export_format='GLB')
-    ##bpy.ops.export_scene.gltf(cntxt, RenderSettings)
-    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(last_created_file)
-    ##return last_created_file
-    ##return fin_rend
-    ##return apol
-    return last_created_file
-
-def export_gltf():
-    ##bpy.ops.wm.save_mainfile(filepath='./path_to_exp.blend')
-    ##bpy.ops.wm.save_as_mainfile(filepath='./exp_scene.blend')
-    ##bpy.ops.export_scene.obj(filepath='./exp_scene.obj', export_format='OBJ', export_materials='EXPORT')
-    ##bpy.ops.export_scene.gltf(filepath='./exp_scene.glb', export_format='GLB', check_existing=False, export_image_format='AUTO', export_materials='EXPORT', export_cameras=True, export_lights=True)
-    ##bpy.ops.export_scene.gltf(filepath='./exp_scene.gltf',export_format='GLTF_SEPARATE')
-    bpy.ops.export_scene.gltf(filepath='./exp_scene.gltf',export_format='GLB')
-    ##cntxt = bpy.context
-    ##bpy.ops.export_scene.gltf(filepath='./exp_scene.glb', export_format='GLB')
-    ##bpy.ops.export_scene.gltf(cntxt, RenderSettings)
     last_created_time = 0
     last_created_file = None
     apol=[]
@@ -356,14 +226,10 @@ def export_gltf():
                 last_created_time = created_time
                 last_created_file = file_path
 
-    ##imoge = load_image(last_created_file).convert('RGB').resize((512, 512))
     apol.append(last_created_file)
-    ##return last_created_file
-    ##return fin_rend
-    ##return apol
     return last_created_file
 
-def render_anim():
+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'
@@ -385,34 +251,21 @@ def render_anim():
                 last_created_time = created_time
                 last_created_file = file_path
 
-    ##imoge = load_image(last_created_file).convert('RGB').resize((512, 512))
     apol.append(last_created_file)
-    ##return last_created_file
-    ##return fin_rend
-    return apol[0]
-    
-    ##return fin_anim
-
-##export_glb()
-##export_gltf()
-##export_blend()
+    moog = gr.Model3D(value=sny)
+    return apol[0], sny, moog
 
 with gr.Blocks() as iface:
     with gr.Column():
-        # HTML block to view the exported scene using Three.js
-        
         moog = gr.Model3D()
-        ##zery = gr.Button("create Scene")
         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)
-        ##sey = gr.File(value=export_gltf(),interactive=False)
         siy = gr.File(value=export_blend(),interactive=False)
-        blk=gr.HTML(value="")
 
-    ony.click(render_scene, [blk,sny], [rdr,blk,sny,moog])
-    twy.click(render_anim, None, rani)
+    ony.click(render_scene, [sny], [rdr,sny,moog])
+    twy.click(render_anim, [sny], [rani,sny,moog])
 
 iface.launch()