Update app.py

app.py

@@ -217,6 +217,62 @@ def fill_image(prompt, image, model_selection):
 def clear_result():
     return gr.update(value=None)
 
+def process_inpainting(image, mask_input, prompt):
+    """이미지 인페인팅 처리 함수"""
+    try:
+        if image is None or mask_input is None or not prompt:
+            raise gr.Error("Please provide image, mask, and prompt")
+            
+        # 프롬프트 번역 (한글인 경우)
+        translated_prompt = translate_if_korean(prompt)
+        
+        # 마스크 처리
+        source = image
+        if isinstance(mask_input, dict):
+            mask = mask_input["layers"][0]
+            alpha_channel = mask.split()[3]
+            binary_mask = alpha_channel.point(lambda p: p > 0 and 255)
+        else:
+            raise gr.Error("Invalid mask input")
+
+        # 인페인팅을 위한 이미지 준비
+        cnet_image = source.copy()
+        cnet_image.paste(0, (0, 0), binary_mask)
+
+        # 프롬프트 임베딩
+        (
+            prompt_embeds,
+            negative_prompt_embeds,
+            pooled_prompt_embeds,
+            negative_pooled_prompt_embeds,
+        ) = pipe.encode_prompt(translated_prompt, "cuda", True)
+
+        # 인페인팅 실행
+        result = None
+        for image in pipe(
+            prompt_embeds=prompt_embeds,
+            negative_prompt_embeds=negative_prompt_embeds,
+            pooled_prompt_embeds=pooled_prompt_embeds,
+            negative_pooled_prompt_embeds=negative_pooled_prompt_embeds,
+            image=cnet_image,
+        ):
+            result = image
+
+        if result is None:
+            raise gr.Error("Inpainting failed")
+
+        # 결과 이미지 처리
+        result = result.convert("RGBA")
+        cnet_image.paste(result, (0, 0), binary_mask)
+        
+        return cnet_image
+
+    except Exception as e:
+        print(f"Inpainting error: {str(e)}")
+        raise gr.Error(f"Inpainting failed: {str(e)}")
+    finally:
+        clear_memory()
+        
 #--------------- 이미지 인페인팅 끝 ----------------    
 
 class timer: