Update app.py

app.py

@@ -9,7 +9,7 @@ import tempfile
 
 
 def enable_GPUS():
-    bpy.data.scenes[0].render.engine = "CYCLES"
+    bpy.data.scenes[0].render.engine = "CYCLES" #"CYCLES"
     # Set the device_type
     bpy.context.preferences.addons[
         "cycles"
@@ -28,11 +28,10 @@ def enable_GPUS():
         print(d["name"])
 
 
-enable_GPUS()
+# enable_GPUS()
 
 # bpy.ops.wm.read_factory_settings(use_empty=True)
 
-
 def generate(
     color1,
     color2,
@@ -48,6 +47,8 @@ def generate(
     rgb1 = tuple(v / 255.0 for v in rgb1)
     rgb2 = ImageColor.getcolor(color2, "RGBA")
     rgb2 = tuple(v / 255.0 for v in rgb2)
+    # bpy.ops.wm.read_homefile(use_empty=True)
+
 
     # Delete all mesh objects from the scene
     bpy.ops.object.select_all(action="DESELECT")
@@ -58,16 +59,17 @@ def generate(
     bpy.ops.mesh.primitive_torus_add(
         major_radius=1.5,
         minor_radius=0.75,
-        major_segments=48 * 4,
-        minor_segments=12 * 4,
+        major_segments=12,
+        minor_segments=12,
         align="WORLD",
         location=(0, 1, 1),
+        rotation=(torus_X,torus_Y,torus_Z)
+
     )
 
     # Assigning the torus to a variable
-    # torus = bpy.context.active_object
     torus = bpy.context.view_layer.objects.active
-    # print(torus)
+
     # Create a new material and assign it to the torus
     material = bpy.data.materials.new(name="RainbowGradient")
     torus.data.materials.append(material)
@@ -104,11 +106,8 @@ def generate(
     material.node_tree.links.new(bsdf.outputs["BSDF"], output.inputs["Surface"])
 
     # Rotate the gradient to apply it from left to right
-    torus.rotation_euler = (
-        torus_X,
-        torus_Y,
-        torus_Z,
-    )  # Rotate 90 degrees on the Z axis
+    torus = bpy.context.view_layer.objects.active
+    # torus.rotation_euler = 
 
     # Light
     light = bpy.data.objects["Light"]
@@ -120,26 +119,31 @@ def generate(
     camera.data.dof.use_dof = True
     camera.data.dof.focus_distance = 5
     camera.data.dof.aperture_fstop = 4
-
+  
     # Render
-    with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as f:
-        bpy.context.scene.render.resolution_y = 128
-        bpy.context.scene.render.resolution_x = 128
-        bpy.context.scene.render.image_settings.file_format = "PNG"
+    with tempfile.NamedTemporaryFile(suffix=".JPEG", delete=False) as f:
+        
+        bpy.context.scene.render.resolution_y = 256
+        bpy.context.scene.render.resolution_x = 256
+        bpy.context.scene.render.image_settings.file_format = "JPEG"
         bpy.context.scene.render.filepath = f.name
+        bpy.context.scene.frame_current = bpy.context.scene.frame_end
 
         with tqdm(total=bpy.context.scene.frame_end) as pbar:
 
-            def elapsed(dummy):
-                pbar.update()
+          def elapsed(dummy):
+            pbar.update()
 
-            bpy.app.handlers.render_stats.append(elapsed)
-            bpy.ops.render.render(animation=False, write_still=True)
-            bpy.data.images["Render Result"].save_render(
-                filepath=bpy.context.scene.render.filepath
-            )
-            bpy.app.handlers.render_stats.clear()
-            return f.name
+          bpy.app.handlers.render_stats.append(elapsed)
+          # bpy.ops.render.render(animation=False, write_still=True)
+          # bpy.ops.render.render(animation=False, write_still=True)
+          bpy.ops.render.render(animation=False, write_still=True)
+                                
+          bpy.data.images["Render Result"].save_render(
+              filepath=bpy.context.scene.render.filepath
+          )
+          bpy.app.handlers.render_stats.clear()
+          return f.name
 
 
 # generate("#ffffff", "#aaa", 1)
@@ -152,11 +156,11 @@ with gr.Blocks() as demo:
             camera_Y = gr.Slider(minimum=-100, maximum=100, value=-3, label="Camera Y")
             camera_Z = gr.Slider(minimum=-100, maximum=100, value=4, label="Camera Z")
             torus_X = gr.Slider(minimum=-pi, maximum=pi, value=0, label="Torus φ")
-            torus_Y = gr.Slider(minimum=-pi, maximum=pi, value=0, label="Torus θ")
-            torus_Z = gr.Slider(minimum=-pi, maximum=pi, value=pi / 2, label="Torus ψ")
+            torus_Y = gr.Slider(minimum=-pi, maximum=pi, value=-3, label="Torus θ")
+            torus_Z = gr.Slider(minimum=-pi, maximum=pi, value=1.5, label="Torus ψ")
 
             render_btn = gr.Button("Render")
-        with gr.Column():
+        with gr.Column(scale=3):
             image = gr.Image(type="filepath")
 
     render_btn.click(
@@ -175,4 +179,4 @@ with gr.Blocks() as demo:
     )
 
 demo.queue()
-demo.launch(debug=True, inline=True)
+demo.launch(debug=True, inline=True)