Update app.py

app.py

@@ -233,7 +233,7 @@ def save_metadata_to_file(image_input, prompt, neg_prompt, model, loras, seed, s
     if not any([prompt, neg_prompt, model, loras, seed, sampler, sampler_params, metadata]):
         return None, gr.Info("Không có thông tin để lưu!", duration=2)
 
-    # Lấy tên file ảnh và đặt tên cho file txt, chỉ thay đổi phần mở rộng
+    # Lấy tên file ảnh và đặt tên cho file txt
     image_filename = os.path.basename(image_input)
     txt_filename = os.path.splitext(image_filename)[0] + ".txt"
     txt_filepath = os.path.join(output_dir, txt_filename)
@@ -249,7 +249,8 @@ def save_metadata_to_file(image_input, prompt, neg_prompt, model, loras, seed, s
         f.write(f"Sampler Parameters: {sampler_params}\n")
         f.write(f"Other Metadata:\n{metadata}\n")
     
-    return txt_filepath, gr.Info(f"Đã lưu thông tin vào file: {txt_filename}", duration=2)
+    return txt_filepath  # Trả về đường dẫn file
+
 
 
 def check_image_size(image_input):
@@ -318,7 +319,7 @@ with gr.Blocks(js = js_func) as demo:
             sampler_output = gr.Textbox(label="Phương pháp lấy mẫu")
             sampler_params_output = gr.Textbox(label="Thông số lấy mẫu")
             other_metadata_output = gr.Textbox(label="Thông tin khác", lines=10)
-            download_file = gr.File(label="Tải file đã tạo", visible=True)
+            download_file = gr.File(label="Tải file đã tạo", visible=False)
             
     
     read_button.click(
@@ -347,14 +348,15 @@ with gr.Blocks(js = js_func) as demo:
         outputs=[copy_prompt, copy_neg_prompt, copy_seed, message_output]
     )
     download_button.click(
-        fn=save_metadata_to_file,
-        inputs=[
-            image_input, prompt_output, negative_prompt_output, 
-            model_output, lora_output, seed_output, 
-            sampler_output, sampler_params_output, other_metadata_output
-        ],
-        outputs=[download_file, message_output]  # Trả về tệp tin và thông báo trạng thái
-    )
+    fn=save_metadata_to_file,
+    inputs=[
+        image_input, prompt_output, negative_prompt_output, 
+        model_output, lora_output, seed_output, 
+        sampler_output, sampler_params_output, other_metadata_output
+    ],
+    outputs=None  # Không cần `gr.File`, tự động tải file về
+)
+