Update app.py

app.py

@@ -60,12 +60,14 @@ 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), gr.update(visible=False)
-
+        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 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)
+    txt_filepath = os.path.join("outputs", txt_filename)
 
     # Ghi thông tin metadata vào file txt
     with open(txt_filepath, "w", encoding="utf-8") as f:
@@ -78,8 +80,7 @@ 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.update(visible=True) 
-
+    return txt_filepath, gr.Info(f"Đã lưu thông tin vào file: {txt_filename}", duration=2)
 
 
     
@@ -223,27 +224,7 @@ 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):
-    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
-    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)
 
-    # 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  # Trả về đường dẫn file
 
 
 
@@ -348,7 +329,7 @@ with gr.Blocks(js = js_func) as demo:
         model_output, lora_output, seed_output, 
         sampler_output, sampler_params_output, other_metadata_output
     ],
-    outputs=[download_file, message_output]
+    outputs=[download_file, message_output]  # Tải file và hiển thị thông báo
 )