Update app.py

app.py

@@ -57,13 +57,16 @@ def format_sampler_params(sampler_params):
     return f"scheduler: {scheduler}\nCFG: {cfg_scale}\nSteps: {steps}"
 
 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 để lưu
     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)
     
     if image_input is None:
         return None, gr.Info("Không thể lưu tệp: Không có ảnh nào được tải lên.", duration=2)
 
+    # Lấy tên ảnh để đặt tên cho file txt
+    image_filename = os.path.basename(image_input)
+    txt_filename = os.path.splitext(image_filename)[0] + ".txt"
+
     # Tạo nội dung file txt từ metadata
     file_content = (
         f"Prompt: {prompt}\n"
@@ -81,8 +84,8 @@ def save_metadata_to_file(image_input, prompt, neg_prompt, model, loras, seed, s
     txt_file.write(file_content.encode('utf-8'))
     txt_file.seek(0)
 
-    # Trả về file để tải xuống
-    return txt_file, gr.Info(f"Đang tải file metadata.txt...", duration=2)
+    return gr.File.update(value=txt_file, filename=txt_filename), gr.Info(f"Đang tải file {txt_filename}...", duration=2)
+
 
 
     
@@ -336,7 +339,7 @@ with gr.Blocks(js = js_func) as demo:
             model_output, lora_output, seed_output, 
             sampler_output, sampler_params_output, other_metadata_output
         ],
-        outputs=None  # Tải file trực tiếp về máy
+        outputs=[download_file]  # Tải file trực tiếp về máy
     )