Update generator.py

generator.py

@@ -22,8 +22,18 @@ class Generator:
         
         # 3. Generate Prompt
         if not user_prompt.strip():
-            generated_caption = get_caption(processed_image)
-            final_prompt = f"{Config.STYLE_TRIGGER}, {generated_caption}"
+            # Only get caption if face analysis didn't fail (to avoid double loading)
+            if face_emb is not None:
+                generated_caption = get_caption(processed_image)
+                final_prompt = f"{Config.STYLE_TRIGGER}, {generated_caption}"
+            else:
+                # If face analysis failed, we might be here, so try captioning
+                try:
+                    generated_caption = get_caption(processed_image)
+                    final_prompt = f"{Config.STYLE_TRIGGER}, {generated_caption}"
+                except Exception as e:
+                    print(f"Captioning failed: {e}, using default prompt.")
+                    final_prompt = f"{Config.STYLE_TRIGGER}, a beautiful pixel art image"
         else:
             final_prompt = f"{Config.STYLE_TRIGGER}, {user_prompt}"
             
@@ -42,8 +52,11 @@ class Generator:
             self.mh.pipeline.set_ip_adapter_scale(0.9) # Set IP-Adapter (likeness) strength
         else:
             print("No face detected: Disabling InstantID.")
-            controlnet_conditioning_scale = [0.0, 0.4, 0.4] # Disable InstantID
-            control_guidance_end = [0.0, 0.8, 0.8]
+            controlnet_conditioning_scale = [0.0, 0.4, 0.4] # Disable InstantID weight
+            # --- START FIX ---
+            # Set end to 0.5 (matching the 'if' block) to avoid 0.0 >= 0.0 error
+            control_guidance_end = [0.5, 0.8, 0.8] 
+            # --- END FIX ---
             self.mh.pipeline.set_ip_adapter_scale(0.0)
 
         # 6. Run Inference