Update app.py

app.py

@@ -235,90 +235,100 @@ def run_example_2():
         result_img.save(output_path, "WEBP")
     return result_img, status, final_prompt
 
-# -------------------------------------------------------------------
-# 예제 적용 부분 (참조 코드 적용)
-# -------------------------------------------------------------------
-# Set up the interaction with two outputs.
-submit_btn.click(
-    fn=process_image_and_prompt,
-    inputs=[image_input, prompt_input, gemini_api_key],
-    outputs=[output_gallery, output_text],
-)
-
-gr.Markdown("## Try these examples", elem_classes="gr-examples-header")
-
-examples = [
-    ["data/1.webp", 'change text to "AMEER"', ""],
-    ["data/2.webp", "remove the spoon from hand only", ""],
-    ["data/3.webp", 'change text to "Make it "', ""],
-    ["data/1.jpg", "add joker style only on face", ""],
-    ["data/1777043.jpg", "add joker style only on face", ""],
-    ["data/2807615.jpg", "add lipstick on lip only", ""],
-    ["data/76860.jpg", "add lipstick on lip only", ""],
-    ["data/2807615.jpg", "make it happy looking face only", ""],
-]
-
-gr.Examples(
-    examples=examples,
-    inputs=[image_input, prompt_input],
-    elem_id="examples-grid"
-)
-# -------------------------------------------------------------------
-
-# 기존 Gradio 인터페이스 구성 (예제 적용 외 다른 코드는 그대로 유지)
+# Gradio 인터페이스
 with gr.Blocks() as demo:
     gr.HTML(
         """
         <div style="text-align: center; margin-bottom: 1rem;">
-            <h1>Gemini for Image Editing</h1>
-            <p>Upload an image and enter a prompt to generate outputs.</p>
+            <h1>간단한 이미지 생성기 (예제 포함)</h1>
+            <p>이미지를 업로드하거나 아래 예제 버튼을 클릭하면 처리된 결과가 down 폴더에 저장됩니다.</p>
         </div>
         """
     )
     
     with gr.Row():
         with gr.Column():
-            image_input = gr.Image(
-                type="pil",
-                label="Upload Image",
-                image_mode="RGBA",
-                elem_id="image-input"
-            )
-            gemini_api_key = gr.Textbox(
-                lines=1,
-                placeholder="Enter Gemini API Key (optional)",
-                label="Gemini API Key (optional)"
-            )
+            with gr.Row():
+                image1_input = gr.Image(type="pil", label="#1", image_mode="RGB")
+                image2_input = gr.Image(type="pil", label="#2", image_mode="RGB")
+                image3_input = gr.Image(type="pil", label="#3", image_mode="RGB")
             prompt_input = gr.Textbox(
-                lines=2,
-                placeholder="Enter prompt here...",
-                label="Prompt"
+                lines=3,
+                placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
+                label="프롬프트 (선택 사항)"
             )
-            submit_btn = gr.Button("Generate")
+            with gr.Row():
+                image_change_btn = gr.Button("이미지 변경")
+                text_remove_btn = gr.Button("글자지우기")
+                clothes_change_btn = gr.Button("옷바꾸기")
+                background_change_btn = gr.Button("배경바꾸기")
+                composite_product_btn = gr.Button("이미지 합성(상품포함)")
+            submit_btn = gr.Button("이미지 생성", variant="primary")
         with gr.Column():
-            output_gallery = gr.Gallery(label="Generated Outputs")
-            output_text = gr.Textbox(
-                label="Gemini Output",
-                placeholder="Text response will appear here if no image is generated."
-            )
+            output_image = gr.Image(label="생성된 이미지")
+            output_text = gr.Textbox(label="상태 메시지")
+            prompt_display = gr.Textbox(label="사용된 프롬프트 (영어)", visible=True)
+    
+    # 선택 옵션 버튼 클릭 시 프롬프트 입력란 업데이트 (한국어 문구)
+    image_change_btn.click(
+        fn=lambda: "#1 이미지의 [다른 모습]으로 변경하라.",
+        inputs=[],
+        outputs=prompt_input
+    )
+    text_remove_btn.click(
+        fn=lambda: "#1 이미지의 [중국어를 모두]를 제거하라.",
+        inputs=[],
+        outputs=prompt_input
+    )
+    clothes_change_btn.click(
+        fn=lambda: "#1 이미지의 인물 의상을 #2 이미지의 의상으로 변경하라.",
+        inputs=[],
+        outputs=prompt_input
+    )
+    background_change_btn.click(
+        fn=lambda: "#1 이미지의 배경을 #2 이미지의 배경으로 변경하라.",
+        inputs=[],
+        outputs=prompt_input
+    )
+    composite_product_btn.click(
+        fn=lambda: "#1 이미지와 #2 이미지(또는 #3 이미지)를 자연스럽게 합성하라.",
+        inputs=[],
+        outputs=prompt_input
+    )
+    
+    submit_btn.click(
+        fn=process_and_show_prompt,
+        inputs=[image1_input, image2_input, image3_input, prompt_input],
+        outputs=[output_image, output_text, prompt_display],
+    )
     
-    gr.Markdown("## Try these examples", elem_classes="gr-examples-header")
+    with gr.Row():
+        example1_btn = gr.Button("예제 1 실행")
+        example2_btn = gr.Button("예제 2 실행")
     
-    examples = [
-        ["data/1.webp", 'change text to "AMEER"', ""],
-        ["data/2.webp", "remove the spoon from hand only", ""],
-        ["data/3.webp", 'change text to "Make it "', ""],
-        ["data/1.jpg", "add joker style only on face", ""],
-        ["data/1777043.jpg", "add joker style only on face", ""],
-        ["data/2807615.jpg", "add lipstick on lip only", ""],
-        ["data/76860.jpg", "add lipstick on lip only", ""],
-        ["data/2807615.jpg", "make it happy looking face only", ""],
-    ]
+    example1_btn.click(
+        fn=run_example_1,
+        inputs=[],
+        outputs=[output_image, output_text, prompt_display]
+    )
+    example2_btn.click(
+        fn=run_example_2,
+        inputs=[],
+        outputs=[output_image, output_text, prompt_display]
+    )
     
-    gr.Examples(
-        examples=examples,
-        inputs=[image_input, prompt_input],
-        elem_id="examples-grid"
+    gr.Markdown(
+        """
+        ### 사용 방법:
+        
+        1. **자동 합성**: 이미지를 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다.
+        2. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다.
+        3. **예제 실행**: 아래 "예제 1 실행" 또는 "예제 2 실행" 버튼을 누르면 down 폴더의 입력 이미지를 처리하여 결과가 저장됩니다.
+        4. **선택 옵션**: 위의 버튼을 클릭하면 프롬프트 입력란에 한국어 문구가 채워지고, 이후 Gemini‑2.0‑flash 모델을 통해 영어로 번역되어 실제 API 호출 프롬프트로 적용됩니다.
+        
+        > **팁**: 프롬프트를 직접 수정할 수도 있습니다.
+        """
     )
 
-demo.queue(max_size=50).launch()
+if __name__ == "__main__":
+    demo.launch(share=True)