Update app.py

app.py

@@ -56,35 +56,29 @@ def format_sampler_params(sampler_params):
     steps = params.get("steps", "Không xác định")
     return f"scheduler: {scheduler}\nCFG: {cfg_scale}\nSteps: {steps}"
 
+download_file = gr.File(label="Tải file đã tạo", visible=False)
+
 def save_metadata_to_file(image_input, prompt, neg_prompt, model, loras, seed, sampler, sampler_params, metadata):
     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)
+        return None, gr.Info("Không có thông tin để lưu!", duration=2), gr.update(visible=False)
 
-    # Lấy tên ảnh để đặt tên cho file txt
+    # 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)
 
-    # 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"
-    )
-
-    # 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)
-
-    return gr.File.update(value=txt_file, filename=txt_filename), gr.Info(f"Đang tải file {txt_filename}...", duration=2)
+    # 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")
+    
+    return txt_filepath, gr.update(visible=True) 
 
 
 
@@ -354,7 +348,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  # Không cần `gr.File`, tự động tải file về
+    outputs=[download_file, message_output]
 )