Update app.py

app.py

@@ -78,7 +78,7 @@ def generate_by_google_genai(text, file_name, model="gemini-2.0-flash-exp"):
     - file_name: 원본 이미지(예: .png) 경로
     - model: 사용할 gemini 모델 이름
     """
-    # GAPI_TOKEN 환경변수에서 키를 가져옴 (필수)
+    # (1) 환경 변수에서 API 키 가져오기 (필수)
     api_key = os.getenv("GAPI_TOKEN", None)
     if not api_key:
         raise ValueError(
@@ -86,12 +86,13 @@ def generate_by_google_genai(text, file_name, model="gemini-2.0-flash-exp"):
             "Google GenAI API를 사용하기 위해서는 GAPI_TOKEN이 필요합니다."
         )
 
+    # (2) Google Client 초기화
     client = genai.Client(api_key=api_key)
     
-    # 이미지 업로드
+    # (3) 이미지 업로드
     files = [client.files.upload(file=file_name)]
     
-    # gemini에 전달할 Content 준비
+    # (4) gemini에 전달할 Content 준비 (이미지 + 프롬프트)
     contents = [
         types.Content(
             role="user",
@@ -105,6 +106,7 @@ def generate_by_google_genai(text, file_name, model="gemini-2.0-flash-exp"):
         ),
     ]
 
+    # (5) 생성/변환 설정
     generate_content_config = types.GenerateContentConfig(
         temperature=1,
         top_p=0.95,
@@ -117,9 +119,10 @@ def generate_by_google_genai(text, file_name, model="gemini-2.0-flash-exp"):
     text_response = ""
     image_path = None
 
-    # 임시 파일에 이미지 응답을 저장할 준비
+    # 임시 파일로 이미지 받을 준비
     with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
         temp_path = tmp.name
+        # 응답 스트림을 받으면서 이미지/텍스트 구분 처리
         for chunk in client.models.generate_content_stream(
             model=model,
             contents=contents,
@@ -129,32 +132,39 @@ def generate_by_google_genai(text, file_name, model="gemini-2.0-flash-exp"):
                 continue
             candidate = chunk.candidates[0].content.parts[0]
 
-            # inline_data(이미지) 응답인 경우
+            # inline_data가 있으면 이미지 응답
             if candidate.inline_data:
                 save_binary_file(temp_path, candidate.inline_data.data)
                 print(f"File of mime type {candidate.inline_data.mime_type} saved to: {temp_path}")
                 image_path = temp_path
                 break
             else:
+                # 이미지 없이 텍스트만 반환되는 경우
                 text_response += chunk.text + "\n"
     
+    # 업로드한 File 객체 제거
     del files
+
     return image_path, text_response
 
 #######################################
-# 3. Gradio 함수
+# 3. Gradio 함수 
+#    (1) FLUX로 이미지 생성 -> (2) Google GenAI로 텍스트 교체
 #######################################
 
 def generate_initial_image(prompt, text, height, width, steps, scale, seed):
     """
-    FLUX를 이용해 텍스트가 포함된 이미지를 생성
-    - prompt 내에 <text>라는 특수 구분자가 있으면, 거기에 text가 치환됨.
-    - 그렇지 않은 경우, 기존처럼 prompt 뒤에 “with clear readable text that says ...”를 추가.
+    FLUX 파이프라인을 사용해 '텍스트가 포함된 이미지를' 먼저 생성.
+    - prompt 내 <text>를 text로 치환
+    - <text>가 없다면 "with clear readable text that says '<text>'"를 자동 붙임
     """
     if "<text>" in prompt:
         combined_prompt = prompt.replace("<text>", text)
     else:
         combined_prompt = f"{prompt} with clear readable text that says '{text}'"
+    
+    # 디버그용: 최종 들어가는 프롬프트를 확인
+    print(f"[DEBUG] Final combined_prompt: {combined_prompt}")
 
     with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16), timer("inference"):
         result = pipe(
@@ -171,11 +181,10 @@ def generate_initial_image(prompt, text, height, width, steps, scale, seed):
 
 def change_text_in_image(original_image, new_text):
     """
-    Gemini 모델을 통해,
+    Google GenAI의 gemini 모델을 통해,
     업로드된 이미지 내부의 문구를 `new_text`로 변경해주는 함수.
     """
     try:
-        # 임시 파일에 먼저 저장
         with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
             original_path = tmp.name
             original_image.save(original_path)
@@ -187,35 +196,34 @@ def change_text_in_image(original_image, new_text):
         )
 
         if image_path:
-            # Gradio 구버전에는 decode_base64_to_image가 없으므로 PIL을 직접 사용
+            # Gradio 구버전에는 decode_base64_to_image가 없으므로 PIL로 처리
             with open(image_path, "rb") as f:
                 image_data = f.read()
             modified_img = Image.open(io.BytesIO(image_data))
             return modified_img, ""
         else:
+            # 이미지가 없이 텍스트만 반환된 경우
             return None, text_response
 
     except Exception as e:
         raise gr.Error(f"Error: {e}")
 
 #######################################
-# 4. Gradio 인터페이스
+# 4. Gradio 인터페이스 구성
 #######################################
 
 with gr.Blocks(title="Flux + Google GenAI Text Replacement") as demo:
     gr.Markdown(
         """
-        # Flux Image Generation + Google GenAI Text Replacement
-
-        **Usage Instructions (in English)**  
-        1. Write a prompt that may contain the special placeholder `<text>`.  
-           - Example: `A white cat says <text> in a cartoon style`.  
-        2. Enter the actual text in the "Text to Include in the Image" field.  
-           - Example: `안녕`  
-        3. Click the "Generate Base Image" button.  
-           - The prompt will be transformed so that `<text>` is replaced with your actual text.  
-           - If `<text>` is **not** found, the text will be appended automatically as `with clear readable text that says ...`.  
-        4. (Optional) If you want to change the text again, use the "Change Text in Image" button.
+        # Flux 기반 이미지 생성 + Google GenAI를 통한 텍스트 변환
+
+        **Usage**:
+        - You can include `<text>` in the prompt. For example:  
+          `white cat with speech bubble says <text>`  
+        - Then, type the actual text in "Text to Include in the Image" (ex: "Hello" or "안녕").  
+        - If `<text>` is not found in your prompt, the text will be automatically appended as:  
+          `with clear readable text that says '<text>'`.  
+        - Finally, you can optionally change the text again via Gemini.
 
         ---
         """
@@ -223,32 +231,32 @@ with gr.Blocks(title="Flux + Google GenAI Text Replacement") as demo:
 
     with gr.Row():
         with gr.Column():
-            gr.Markdown("## 1) Generate the Base Image (FLUX)")
+            gr.Markdown("## 1) Step 1: FLUX로 텍스트 포함 이미지 생성")
             prompt_input = gr.Textbox(
                 lines=3,
-                label="Prompt (with optional `<text>` placeholder)",
-                placeholder="e.g. A white cat says <text> in a cartoon style"
+                label="이미지 장면/배경 Prompt (use `<text>` placeholder if you like)",
+                placeholder="e.g. A white cat with speech bubble says <text>"
             )
             text_input = gr.Textbox(
                 lines=1,
-                label="Text to Include in the Image",
-                placeholder="e.g. 안녕"
+                label="이미지 안에 들어갈 텍스트",
+                placeholder="e.g. Hello or 안녕"
             )
-            with gr.Accordion("Advanced Settings", open=False):
+            with gr.Accordion("고급 설정 (확장)", open=False):
                 height = gr.Slider(label="Height", minimum=256, maximum=1152, step=64, value=512)
                 width = gr.Slider(label="Width", minimum=256, maximum=1152, step=64, value=512)
                 steps = gr.Slider(label="Inference Steps", minimum=6, maximum=25, step=1, value=8)
-                scale = gr.Slider(label="Guidance Scale", minimum=0.0, maximum=5.0, step=0.1, value=3.5)
+                scale = gr.Slider(label="Guidance Scale", minimum=0.0, maximum=10.0, step=0.5, value=3.5)
                 seed = gr.Number(label="Seed (reproducibility)", value=1234, precision=0)
 
             generate_btn = gr.Button("Generate Base Image", variant="primary")
-            generated_image = gr.Image(label="Generated Image", type="pil")
+            generated_image = gr.Image(label="Generated Image (with text)", type="pil")
 
         with gr.Column():
-            gr.Markdown("## 2) (Optional) Change Text in the Generated Image (Gemini)")
+            gr.Markdown("## 2) Step 2: 생성된 이미지 내 텍스트 수정")
             new_text_input = gr.Textbox(
-                label="New Text to Insert",
-                placeholder="e.g. Hello"
+                label="새로 바꿀 텍스트",
+                placeholder="예) Hello world"
             )
             modify_btn = gr.Button("Change Text in Image via Gemini", variant="secondary")
             output_img = gr.Image(label="Modified Image", type="pil")