[FIX] 保存機能を追加

app.py

@@ -13,8 +13,11 @@ model = MLModel()
 
 
 def save(image_path: str):
-    _ = model.save(image_path)
-    return
+    try:
+        id = model.save(image_path)
+    except Exception as e:
+        return str(e), None
+    return f"保存しました。ID: {id}", None
 
 
 def search(prompt: str):
@@ -29,7 +32,8 @@ with gr.Blocks() as app:
     btn = gr.Button("検索")
     btn.click(fn=search, inputs=input, outputs=output)
 
-    save_image = gr.Image()
+    save_image = gr.Image(type="filepath")
+    result = gr.Text(label="upload結果")
     save_btn = gr.Button("保存")
-    save_btn.click(fn=save, inputs=save_image, outputs=None)
+    save_btn.click(fn=save, inputs=save_image, outputs=[result, save_image])
 app.launch()