Update inpainting_module.py

inpainting_module.py

@@ -893,13 +893,28 @@ class InpaintingModule:
                 self._current_conditioning_type
             )
 
-            # Enhance prompt
-            if progress_callback:
-                progress_callback("Enhancing prompt...", 25)
+            # Conditional prompt enhancement based on template
+            # Check if we should enhance the prompt or use it directly
+            template_key = kwargs.get('template_key', None)
+            should_enhance = True  # Default: enhance
 
-            enhanced_prompt, negative_prompt = self.enhance_prompt_for_inpainting(
-                prompt, image, processed_mask
-            )
+            if template_key and hasattr(self, 'template_manager'):
+                template = self.template_manager.get_template(template_key)
+                if template and hasattr(template, 'enhance_prompt'):
+                    should_enhance = template.enhance_prompt
+
+            if should_enhance:
+                if progress_callback:
+                    progress_callback("Enhancing prompt...", 25)
+                enhanced_prompt, negative_prompt = self.enhance_prompt_for_inpainting(
+                    prompt, image, processed_mask
+                )
+                logger.info(f"Prompt enhanced with OpenCLIP context")
+            else:
+                # Use prompt directly without enhancement
+                enhanced_prompt = prompt
+                negative_prompt = self._get_inpainting_negative_prompt()
+                logger.info("Prompt enhancement disabled for this template")
 
             # Setup generator for reproducibility
             if seed is None: