Update app.py

app.py

@@ -349,8 +349,7 @@ def on_change_prompt(img: Image.Image | None, prompt: str | None, bg_prompt: str
 def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = None, 
                   aspect_ratio: str = "1:1", position: str = "bottom-center", 
                   scale_percent: float = 100) -> tuple[Image.Image, Image.Image]:
-    try:        
-
+    try:
         if img is None or prompt.strip() == "":
             raise gr.Error("Please provide both image and prompt")
         
@@ -365,7 +364,6 @@ def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = None,
         
         results, _ = _process(img, prompt, bg_prompt, aspect_ratio)
         
-        
         if bg_prompt:
             try:
                 print(f"Using position: {position}")  # 디버깅용
@@ -383,14 +381,17 @@ def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = None,
                     position=position,
                     scale_percent=scale_percent
                 )
-
                 return combined, results[2]
             except Exception as e:
                 print(f"Combination error: {str(e)}")
                 return results[1], results[2]
         
-        # 이 부분이 누락되었습니다
         return results[1], results[2]  # 기본 반환 추가
+    except Exception as e:
+        print(f"Error in process_prompt: {str(e)}")
+        raise gr.Error(str(e))
+    finally:
+        clear_memory()
 
     
 def process_bbox(img: Image.Image, box_input: str) -> tuple[Image.Image, Image.Image]: