Update app.py

app.py

@@ -22,6 +22,10 @@ import requests
 from google import genai
 from google.genai import types
 
+import numpy as np
+
+
+
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
 def maybe_translate_to_english(text: str) -> str:
@@ -107,11 +111,10 @@ def generate_by_google_genai(text, file_name, model="gemini-2.0-flash-exp"):
     return image_path, text_response
 
 def change_text_in_image_two_times(original_image, instruction):
-    """
-    Gemini API를 두 번 호출하여 텍스트 수정된 이미지 2개(버전 A, B)를 반환.
-    """
-    if original_image is None:
-        raise gr.Error("처리할 이미지가 없습니다. 먼저 이미지를 생성해주세요.")
+    # 만약 이미지가 numpy.ndarray 타입이면 PIL Image로 변환
+    if isinstance(original_image, np.ndarray):
+        original_image = Image.fromarray(original_image)
+    
     results = []
     for version_tag in ["(A)", "(B)"]:
         mod_instruction = f"{instruction} {version_tag}"
@@ -123,7 +126,7 @@ def change_text_in_image_two_times(original_image, instruction):
                     print(f"[DEBUG] Saved image to temporary file: {original_path}")
                 else:
                     raise gr.Error(f"예상된 PIL Image가 아닌 {type(original_image)} 타입이 제공되었습니다.")
-            print(f"[DEBUG] Google Gemini API에 보내는 지시사항: {mod_instruction}")
+            # 이후 Gemini API 호출 로직 유지
             image_path, text_response = generate_by_google_genai(
                 text=mod_instruction,
                 file_name=original_path
@@ -145,6 +148,7 @@ def change_text_in_image_two_times(original_image, instruction):
             results.append(original_image)
     return results
 
+
 def gemini_text_rendering(image, rendering_text):
     """
     주어진 이미지에 대해 Gemini API를 사용해 텍스트 렌더링을 적용.