Update app.py
Browse files
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 |
-
|
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,
|