Update app.py

app.py

@@ -212,32 +212,29 @@ def cancel_copy():
     )
 
 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)
+        return None  # Không có gì để lưu
     
     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)
+        return None  # Không có ảnh nào để lưu
 
-    # Tạo nội dung file txt từ metadata
-    file_content = (
-        f"Prompt: {prompt}\n"
-        f"Negative Prompt: {neg_prompt}\n"
-        f"Model: {model}\n"
-        f"Loras: {loras}\n"
-        f"Seed: {seed}\n"
-        f"Sampler: {sampler}\n"
-        f"Sampler Parameters: {sampler_params}\n"
-        f"Other Metadata:\n{metadata}\n"
-    )
+    # 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)
 
-    # Sử dụng io.BytesIO để tạo file tạm thời trong bộ nhớ
-    txt_file = io.BytesIO()
-    txt_file.write(file_content.encode('utf-8'))
-    txt_file.seek(0)
+    # Ghi thông tin metadata vào file txt
+    with open(txt_filepath, "w", encoding="utf-8") as f:
+        f.write(f"Prompt: {prompt}\n")
+        f.write(f"Negative Prompt: {neg_prompt}\n")
+        f.write(f"Model: {model}\n")
+        f.write(f"Loras: {loras}\n")
+        f.write(f"Seed: {seed}\n")
+        f.write(f"Sampler: {sampler}\n")
+        f.write(f"Sampler Parameters: {sampler_params}\n")
+        f.write(f"Other Metadata:\n{metadata}\n")
 
-    # Trả về file để tải xuống
-    return txt_file, gr.Info(f"Đang tải file metadata.txt...", duration=2)
+    return txt_filepath  # Trả về đường dẫn tệp tin đã tạo
 
 
 def check_image_size(image_input):
@@ -341,7 +338,7 @@ with gr.Blocks(js = js_func) as demo:
         sampler_output, sampler_params_output, other_metadata_output
     ],
     outputs=[download_file]  # Tải file trực tiếp về máy
-)
+    )