Update app.py

app.py

@@ -28,12 +28,10 @@ TEMPLATES = {
     "mountain":   "In this picture, {caption}. The background looks like mountains, suggesting a hiking scene."
 }
 
-def generate_caption(filepaths, choice_index):
+def generate_caption(image_path):
     try:
-        # 1) 선택된 이미지 불러오기
-        idx = int(choice_index)
-        img_path = filepaths[idx]
-        img = Image.open(img_path).convert("RGB")
+        # 1) 이미지 불러오기
+        img = Image.open(image_path).convert("RGB")
 
         # 2) 원본 캡션 생성
         out = captioner(img)
@@ -50,25 +48,15 @@ def generate_caption(filepaths, choice_index):
         return template.format(caption=raw)
 
     except Exception as e:
-        # 문제가 있을 경우 오류 메시지 반환
         return f"🔴 Error: {e}"
 
 # 5) Gradio 인터페이스 정의
 with gr.Blocks() as demo:
-    gr.Markdown("## 📸 TOEIC Part 1: 상황별 사진 묘사")
-    img_inputs = gr.Files(
-        file_count="multiple",
-        type="filepath",
-        label="Upload up to 4 images"
-    )
-    choice = gr.Dropdown(
-        choices=[str(i) for i in range(4)],
-        value="0",
-        label="Which image to describe? (0–3)"
-    )
+    gr.Markdown("## 📸 TOEIC Part 1: 상황별 사진 묘사 (Single Image)")
+    img_in = gr.Image(type="filepath", label="Upload an image")
     btn = gr.Button("Describe")
     output = gr.Textbox(label="TOEIC Part 1 Response", lines=4)
-    btn.click(fn=generate_caption, inputs=[img_inputs, choice], outputs=output)
+    btn.click(fn=generate_caption, inputs=img_in, outputs=output)
 
 # 6) 앱 실행
 if __name__ == "__main__":