Update app.py

app.py

@@ -200,7 +200,7 @@ def copy_to_clipboard(prompt, neg_prompt, seed, copy_prompt, copy_neg_prompt, co
             pyperclip.copy(copied_text)
             return gr.Info("Sao chép thành công!", duration=2)
         except pyperclip.PyperclipException:
-            return gr.Info("Hệ thống không hỗ trợ chức năng sao chép tự động. Vui lòng tự sao chép.", duration=2)
+            return gr.Info("Không hỗ trợ tính năng này trong bản demo. Vui lòng liên hệ tác giả để biết chi tiết !", duration=2)
     else:
         return gr.Info("Không có gì để sao chép!", duration=2)
 
@@ -225,17 +225,19 @@ def cancel_copy():
         )
 
 
+output_dir = "outputs"
+if not os.path.exists(output_dir):
+    os.mkdir(output_dir)
+
 def save_metadata_to_file(image_input, prompt, neg_prompt, model, loras, seed, sampler, sampler_params, metadata):
+    # Kiểm tra nếu không có dữ liệu thì không lưu
     if not any([prompt, neg_prompt, model, loras, seed, sampler, sampler_params, metadata]):
-        return None  # Không có gì để lưu
-    
-    if image_input is None:
-        return None  # Không có ảnh nào để lưu
+        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
     image_filename = os.path.basename(image_input)
     txt_filename = os.path.splitext(image_filename)[0] + ".txt"
-    txt_filepath = os.path.join("outputs", txt_filename)
+    txt_filepath = os.path.join(output_dir, txt_filename)
 
     # Ghi thông tin metadata vào file txt
     with open(txt_filepath, "w", encoding="utf-8") as f:
@@ -248,7 +250,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  # Trả về đường dẫn tệp tin đã tạo
+    # Trả về đường dẫn file để Gradio có thể cung cấp file cho người dùng tải xuống
+    return txt_filepath, gr.Info(f"Đã lưu thông tin vào file: {txt_filename}", duration=2)
 
 
 def check_image_size(image_input):
@@ -345,13 +348,14 @@ 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]
+        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
+    )
 )