a

app_v3.py

@@ -70,7 +70,7 @@ def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_
             control_guidance_start=0.0,
             control_guidance_end=guidance_end,
         ).images[0]
-        print("Type: " + str(type(image)))
+        # print("Type: " + str(type(image)))
     return image
 
 def combine_caption_focus(caption, focus):
@@ -108,7 +108,6 @@ def process_image(control_image, user_prompt, system_prompt, scale, steps,
                 guidance_end, temperature, top_p, max_new_tokens, log_prompt):
     # Initialize with empty caption
     final_prompt = user_prompt.strip()
-    print("User generated prompt: " + final_prompt)
     # If no user prompt provided, generate a caption first
     if not final_prompt:
         # Generate a detailed caption

debug.py

@@ -33,13 +33,16 @@ def save_image(image_id, image_path: Image.Image) -> None:
     print("Image ID: " + image_id)
     print("Image ID Type: " + str(type(image_id)))
     save_image_path = os.path.join(IMAGE_DIR, f"{image_id}.png")
+    print("Save image path: " + save_image_path)
     with scheduler.lock:
         try:
             # print("Image path: " + image_path)
             print("Type: " + str(type(image_path)))
-            commit_image = Image.open(image_path)
-            commit_image = commit_image.convert("P", palette=Image.ADAPTIVE, colors=256)
-            commit_image.save(save_image_path)
+            with Image.open(save_image_path) as img:
+                img = Image.open(image_path)
+                img = img.convert("P", palette=Image.ADAPTIVE, colors=256)
+                img.save(save_image_path)
+                
             print(f"Saved image: {save_image_path}")
         except Exception as e:
             print(f"Error saving image: {str(e)}")