yield 삭제

app.py

@@ -30,7 +30,7 @@ generation = gr.Dropdown(
 download = gr.File(label="Download a file")
 text = gr.DataFrame()
 
-def write_docx(gen):
+def write_docx(gen, progress=gr.Progress()):
     filename = f'포켓몬{gen}.docx'
 
     document = Document()
@@ -50,7 +50,7 @@ def write_docx(gen):
 
     data_dict = []
     start, end = GEN_RANGE[gen]
-    for k in range(start, end+1):
+    for k in progress.tqdm(range(start, end+1)):
         name = pokemons[k-1]['name']
         number = pokemons[k-1]['number']
         types = pokemons[k-1]['types']
@@ -91,13 +91,9 @@ def write_docx(gen):
         hdr_cells[0].paragraphs[0].alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.CENTER
 
         document.add_page_break()
-        yield df[['No', '이름', '타입']], None
 
-    print("Saving docx file...")
-    print(f"{os.listdir()}")
     document.save(filename)
-    print(f"{os.listdir()}")
-    return df, filename
+    return df[['No', '이름', '타입']], filename
 
 demo = gr.Interface(write_docx, generation, [text, download], title="대치동 포켓몬 도감 생성기",
                     description="원하는 포켓몬 세대를 선택하고, 다운로드를 눌러주세요.")