fix: Improve video realism by using Forest HDRI envmap (same as UNLOCKED)

app.py

@@ -202,7 +202,7 @@ def image_to_3d(
             
             # --- Render Video Preview (8 seconds at 15fps = 120 frames) ---
             render_result = render_utils.render_video(mesh, num_frames=120, resolution=720, envmap=envmap, verbose=False)
-            video = render_result['shaded']
+            video = render_result.get('shaded_forest', render_result.get('shaded'))
             
             download_video = os.path.join(user_dir, 'sample.mp4')
             imageio.mimsave(download_video, video, fps=15, pixelformat='yuv420p')
@@ -417,10 +417,21 @@ if __name__ == "__main__":
     pipeline.cuda()
     
     # Load HDRI envmap for beautiful lighting
-    envmap = EnvMap(torch.tensor(
-        cv2.cvtColor(cv2.imread('assets/hdri/forest.exr', cv2.IMREAD_UNCHANGED), cv2.COLOR_BGR2RGB),
-        dtype=torch.float32, device='cuda'
-    ))
+    # Load HDRI envmap for beautiful lighting
+    envmap = {
+        'forest': EnvMap(torch.tensor(
+            cv2.cvtColor(cv2.imread('assets/hdri/forest.exr', cv2.IMREAD_UNCHANGED), cv2.COLOR_BGR2RGB),
+            dtype=torch.float32, device='cuda'
+        )),
+        'sunset': EnvMap(torch.tensor(
+            cv2.cvtColor(cv2.imread('assets/hdri/sunset.exr', cv2.IMREAD_UNCHANGED), cv2.COLOR_BGR2RGB),
+            dtype=torch.float32, device='cuda'
+        )),
+        'courtyard': EnvMap(torch.tensor(
+            cv2.cvtColor(cv2.imread('assets/hdri/courtyard.exr', cv2.IMREAD_UNCHANGED), cv2.COLOR_BGR2RGB),
+            dtype=torch.float32, device='cuda'
+        )),
+    }
     
     demo.launch()