Update app.py

app.py

@@ -235,100 +235,61 @@ def run_example_2():
         result_img.save(output_path, "WEBP")
     return result_img, status, final_prompt
 
-# Gradio 인터페이스
+
+# 기존 Gradio 인터페이스 구성 (예제 적용 외 다른 코드는 그대로 유지)
 with gr.Blocks() as demo:
     gr.HTML(
         """
         <div style="text-align: center; margin-bottom: 1rem;">
-            <h1>간단한 이미지 생성기 (예제 포함)</h1>
-            <p>이미지를 업로드하거나 아래 예제 버튼을 클릭하면 처리된 결과가 down 폴더에 저장됩니다.</p>
+            <h1>Gemini for Image Editing</h1>
+            <p>Upload an image and enter a prompt to generate outputs.</p>
         </div>
         """
     )
     
     with gr.Row():
         with gr.Column():
-            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")
+            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)"
+            )
             prompt_input = gr.Textbox(
-                lines=3,
-                placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
-                label="프롬프트 (선택 사항)"
+                lines=2,
+                placeholder="Enter prompt here...",
+                label="Prompt"
             )
-            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")
+            submit_btn = gr.Button("Generate")
         with gr.Column():
-            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],
-    )
+            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."
+            )
     
-    with gr.Row():
-        example1_btn = gr.Button("예제 1 실행")
-        example2_btn = gr.Button("예제 2 실행")
+    gr.Markdown("## Try these examples", elem_classes="gr-examples-header")
     
-    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]
-    )
+    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.Markdown(
-        """
-        ### 사용 방법:
-        
-        1. **자동 합성**: 이미지를 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다.
-        2. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다.
-        3. **예제 실행**: 아래 "예제 1 실행" 또는 "예제 2 실행" 버튼을 누르면 down 폴더의 입력 이미지를 처리하여 결과가 저장됩니다.
-        4. **선택 옵션**: 위의 버튼을 클릭하면 프롬프트 입력란에 한국어 문구가 채워지고, 이후 Gemini‑2.0‑flash 모델을 통해 영어로 번역되어 실제 API 호출 프롬프트로 적용됩니다.
-        
-        > **팁**: 프롬프트를 직접 수정할 수도 있습니다.
-        """
+    gr.Examples(
+        examples=examples,
+        inputs=[image_input, prompt_input],
+        elem_id="examples-grid"
     )
 
-if __name__ == "__main__":
-    demo.launch(share=True)
+demo.queue(max_size=50).launch()