Update main app to class SDXLImageGenerator class

app.py

@@ -1,11 +1,7 @@
 import gradio as gr
 import tempfile
 import os
-
-class SDXLGenerator:
-    def generate_image(self, prompt):
-        # Placeholder for image generation (e.g., returning a sample image or placeholder text)
-        return "Placeholder for Generated Image"
+from SDXLImageGenerator import SDXLImageGenerator  # Import your existing class
 
 class ControlNetProcessor:
     def controlnet_image(self, image):
@@ -27,12 +23,12 @@ class VideoGenerator:
 
 class GradioApp:
     def __init__(self):
-        self.sdxl_generator = SDXLGenerator()
+        self.sdxl_generator = SDXLImageGenerator()  # Use your existing class
         self.controlnet_processor = ControlNetProcessor()
         self.video_generator = VideoGenerator()
 
     def full_pipeline(self, prompt):
-        initial_image = self.sdxl_generator.generate_image(prompt)
+        initial_image = self.sdxl_generator.generate_images([prompt])[0]
         controlled_image = self.controlnet_processor.controlnet_image(initial_image)
         video_path = self.video_generator.generate_3d_video(controlled_image)
         return initial_image, controlled_image, video_path
@@ -42,7 +38,7 @@ class GradioApp:
             fn=self.full_pipeline,
             inputs=gr.Textbox(label="Input Prompt"),
             outputs=[
-                gr.Textbox(label="Generated Image Placeholder"),
+                gr.Image(label="Generated Image"),
                 gr.Textbox(label="ControlNet Output Image Placeholder"),
                 gr.Video(label="3D Model Video")
             ],