laichaoyi commited on
Commit
48a7e1f
·
verified ·
1 Parent(s): 7958529

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -230,11 +230,10 @@ if not os.path.exists(output_dir):
230
  os.mkdir(output_dir)
231
 
232
  def save_metadata_to_file(image_input, prompt, neg_prompt, model, loras, seed, sampler, sampler_params, metadata):
233
- # Kiểm tra nếu không có dữ liệu thì không lưu
234
  if not any([prompt, neg_prompt, model, loras, seed, sampler, sampler_params, metadata]):
235
  return None, gr.Info("Không có thông tin để lưu!", duration=2)
236
 
237
- # Lấy tên file ảnh và đặt tên cho file txt
238
  image_filename = os.path.basename(image_input)
239
  txt_filename = os.path.splitext(image_filename)[0] + ".txt"
240
  txt_filepath = os.path.join(output_dir, txt_filename)
@@ -249,8 +248,7 @@ def save_metadata_to_file(image_input, prompt, neg_prompt, model, loras, seed, s
249
  f.write(f"Sampler: {sampler}\n")
250
  f.write(f"Sampler Parameters: {sampler_params}\n")
251
  f.write(f"Other Metadata:\n{metadata}\n")
252
-
253
- # Trả về đường dẫn file để Gradio có thể cung cấp file cho người dùng tải xuống
254
  return txt_filepath, gr.Info(f"Đã lưu thông tin vào file: {txt_filename}", duration=2)
255
 
256
 
@@ -320,8 +318,9 @@ with gr.Blocks(js = js_func) as demo:
320
  sampler_output = gr.Textbox(label="Phương pháp lấy mẫu")
321
  sampler_params_output = gr.Textbox(label="Thông số lấy mẫu")
322
  other_metadata_output = gr.Textbox(label="Thông tin khác", lines=10)
 
323
 
324
- download_file = gr.File(label="Tải file đã tạo", visible=False)
325
  read_button.click(
326
  fn=check_image_size,
327
  inputs=image_input,
 
230
  os.mkdir(output_dir)
231
 
232
  def save_metadata_to_file(image_input, prompt, neg_prompt, model, loras, seed, sampler, sampler_params, metadata):
 
233
  if not any([prompt, neg_prompt, model, loras, seed, sampler, sampler_params, metadata]):
234
  return None, gr.Info("Không có thông tin để lưu!", duration=2)
235
 
236
+ # Lấy tên file ảnh và đặt tên cho file txt, chỉ thay đổi phần mở rộng
237
  image_filename = os.path.basename(image_input)
238
  txt_filename = os.path.splitext(image_filename)[0] + ".txt"
239
  txt_filepath = os.path.join(output_dir, txt_filename)
 
248
  f.write(f"Sampler: {sampler}\n")
249
  f.write(f"Sampler Parameters: {sampler_params}\n")
250
  f.write(f"Other Metadata:\n{metadata}\n")
251
+
 
252
  return txt_filepath, gr.Info(f"Đã lưu thông tin vào file: {txt_filename}", duration=2)
253
 
254
 
 
318
  sampler_output = gr.Textbox(label="Phương pháp lấy mẫu")
319
  sampler_params_output = gr.Textbox(label="Thông số lấy mẫu")
320
  other_metadata_output = gr.Textbox(label="Thông tin khác", lines=10)
321
+ download_file = gr.File(label="Tải file đã tạo", visible=True)
322
 
323
+
324
  read_button.click(
325
  fn=check_image_size,
326
  inputs=image_input,