Siyun He commited on
Commit
2821745
·
1 Parent(s): c95bd7e

update app download function

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -1,10 +1,13 @@
1
  from ASCII_functions import generate_ascii_art
2
  import gradio as gr
 
3
 
4
  def save_ascii_art(ascii_art):
5
- with open("ascii_art.txt", "w") as file:
6
- file.write(ascii_art)
7
- return "ASCII art saved to ascii_art.txt"
 
 
8
 
9
  css = """
10
  #ascii_art_box textarea {
@@ -28,9 +31,10 @@ with gr.Blocks(css=css) as demo:
28
  ascii_art = gr.Textbox(label="ASCII Art", placeholder=None, interactive=True, lines=50, scale=2, elem_id="ascii_art_box")
29
  with gr.Row():
30
  save_button = gr.Button("Save ASCII Art")
 
31
 
32
  generate_button.click(fn=generate_ascii_art, inputs=[input_image], outputs=ascii_art)
33
- save_button.click(fn=save_ascii_art, inputs=ascii_art, outputs=gr.Textbox(label="Save Status"))
34
 
35
  # Adding examples
36
  gr.Examples(examples=examples, inputs=input_image)
 
1
  from ASCII_functions import generate_ascii_art
2
  import gradio as gr
3
+ import tempfile
4
 
5
  def save_ascii_art(ascii_art):
6
+ # Create a temporary file to save the ASCII art
7
+ with tempfile.NamedTemporaryFile(delete=False, suffix=".txt") as tmp_file:
8
+ tmp_file.write(ascii_art.encode('utf-8'))
9
+ tmp_file_path = tmp_file.name
10
+ return tmp_file_path
11
 
12
  css = """
13
  #ascii_art_box textarea {
 
31
  ascii_art = gr.Textbox(label="ASCII Art", placeholder=None, interactive=True, lines=50, scale=2, elem_id="ascii_art_box")
32
  with gr.Row():
33
  save_button = gr.Button("Save ASCII Art")
34
+ download_link = gr.File(label="Download ASCII Art")
35
 
36
  generate_button.click(fn=generate_ascii_art, inputs=[input_image], outputs=ascii_art)
37
+ save_button.click(fn=save_ascii_art, inputs=ascii_art, outputs=download_link)
38
 
39
  # Adding examples
40
  gr.Examples(examples=examples, inputs=input_image)