Update app.py

app.py

@@ -206,7 +206,6 @@ def process_and_show_prompt(image1, image2, image3, prompt):
         logger.exception("처리 중 오류 발생:")
         return None, f"오류 발생: {str(e)}", prompt
 
-# Gradio 인터페이스를 하나의 Blocks()로 구성
 with gr.Blocks() as demo:
     gr.HTML(
         """
@@ -244,7 +243,6 @@ with gr.Blocks() as demo:
             )
             prompt_display = gr.Textbox(label="Used Prompt (English)", visible=True)
     
-    # 예제 실행 시 자동으로 image_input 및 prompt_input에 값 채워짐
     gr.Markdown("## Try these examples", elem_classes="gr-examples-header")
     
     examples = [
@@ -253,14 +251,18 @@ with gr.Blocks() as demo:
     ]
     
     gr.Examples(
+        fn=lambda img, pr: process_and_show_prompt(img, None, None, pr),
         examples=examples,
         inputs=[image_input, prompt_input],
+        outputs=[output_gallery, output_text, prompt_display],
+        run_on_click=True,
         elem_id="examples-grid"
     )
     
+    # submit_btn 클릭 시 image2, image3는 None으로 전달
     submit_btn.click(
-        fn=process_and_show_prompt,
-        inputs=[image_input, None, None, prompt_input],
+        fn=lambda img, pr, key: process_and_show_prompt(img, None, None, pr),
+        inputs=[image_input, prompt_input, gemini_api_key],
         outputs=[output_gallery, output_text, prompt_display],
     )