Rearrainge Spaces references

app.py

@@ -1,9 +1,10 @@
+import spaces
 import gradio as gr
-from PIL import Image, ImageDraw, ImageChops, ImageColor
+from PIL import Image
 from haishoku.haishoku import Haishoku
-import os
+#import os
 from tempfile import NamedTemporaryFile
-from pathlib import Path
+#from pathlib import Path
 import atexit
 import random
 # Import constants
@@ -15,7 +16,6 @@ IS_SHARED_SPACE = constants.IS_SHARED_SPACE
 from utils.file_utils import cleanup_temp_files
 
 from utils.color_utils import (
-    rgb_to_hex,
     hex_to_rgb,
     detect_color_format,
     update_color_opacity,

utils/ai_generator_diffusers_flux.py

@@ -1,8 +1,8 @@
 # utils/ai_generator_diffusers_flux.py
+import spaces
 import gradio as gr
 import os
 import utils.constants as constants
-import spaces
 import torch
 from diffusers import FluxPipeline,FluxImg2ImgPipeline,FluxControlPipeline
 import accelerate 
@@ -34,7 +34,8 @@ warnings.filterwarnings("ignore", message=".*Torch was not compiled with flash a
 
 PIPELINE_CLASSES = {
     "FluxPipeline": FluxPipeline,
-    "FluxImg2ImgPipeline": FluxImg2ImgPipeline
+    "FluxImg2ImgPipeline": FluxImg2ImgPipeline,
+    "FluxControlPipeline": FluxControlPipeline
 }
 @spaces.GPU(duration=140)
 def generate_image_from_text(