Fix render issue

Image3DProcessor.py

@@ -101,13 +101,16 @@ class Image3DProcessor:
             rendered_image = t_to_512(rendered_image)
             loop_renders.append(torch.clamp(rendered_image * 255, 0.0, 255.0).detach().permute(1, 2, 0).cpu().numpy().astype(np.uint8))
 
-        # Save the mesh and video in memory and return them as byte arrays
+        # Save the mesh to an in-memory buffer
         mesh_buffer = BytesIO()
-        export_to_obj(mesh_buffer, reconstruction_unactivated)  # Function that saves mesh to an in-memory buffer
+        export_to_obj(mesh_buffer, reconstruction)  # Ensure `export_to_obj()` can write to `BytesIO`
+        mesh_buffer.seek(0)  # Reset buffer position for reading
         mesh_data = mesh_buffer.getvalue()
 
+        # Save video to an in-memory buffer
         video_buffer = BytesIO()
         imageio.mimwrite(video_buffer, loop_renders, format='mp4', fps=30)
+        video_buffer.seek(0)  # Reset buffer position for reading
         video_data = video_buffer.getvalue()
 
-        return mesh_data, video_data
+        return mesh_data, video_data

app.py

@@ -6,7 +6,6 @@ from SDXLImageGenerator import SDXLImageGenerator  # Import your existing class
 import sys
 from Image3DProcessor import Image3DProcessor  # Import your 3D processing class
 
-
 class ControlNetProcessor:
     def controlnet_image(self, image):
         # Placeholder for ControlNet processing (e.g., returning a processed image or placeholder text)
@@ -40,12 +39,11 @@ class GradioApp:
         # controlled_image = self.controlnet_processor.controlnet_image(initial_image)
         mesh_data, video_data = self.video_generator.generate_3d_video(initial_image)
         
-        # Create a temporary file to display the mesh content
+        # Create temporary files to display mesh and video content
         with tempfile.NamedTemporaryFile(delete=False, suffix=".obj") as mesh_file:
             mesh_file.write(mesh_data)
             mesh_path = mesh_file.name
         
-        # Create a temporary file to display the video content
         with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as video_file:
             video_file.write(video_data)
             video_path = video_file.name