Update app.py

app.py

@@ -33,18 +33,18 @@ with gr.Blocks(title="대치동 포켓몬 도감 생성기 🐙") as block:
         gr.Markdown("# 대치동 포켓몬 도감 생성기 🐙")
     with gr.Row():
         generation = gr.Dropdown(
-                    [f"{k}세대" for k in range(1, 10)] + ["모든 세대"], value="1세대", label="포켓몬 세대", info="원하는 포켓몬 세대를 선택하세요."
+                    [f"{k}세대" for k in range(1, 10)] + ["모든 세대"], value="1세대", label="🌈포켓몬 세대", info="원하는 포켓몬 세대를 선택하세요."
                 )
         poke_types = gr.Dropdown(
-                    pokemons_types, value="모든 타입", label="포켓몬 타입", info="원하는 포켓몬 타입을 선택하세요."
+                    pokemons_types, value="모든 타입", label="🩻포켓몬 타입", info="원하는 포켓몬 타입을 선택하세요."
                 )
-        button = gr.Button(label="🎩 도감 생성")
+        button = gr.Button(value="🎩 도감 생성")
     with gr.Row():
         with gr.Column():
-            download = gr.File(label="파란 글씨를 눌러서 다운로드 받으세요.")
-            df_view = gr.DataFrame(label="포켓몬 리스트", wrap=True, row_count=10)
+            download = gr.File(label="🔑파란 글씨를 눌러서 다운로드 받으세요.")
+            df_view = gr.DataFrame(label="📝포켓몬 리스트", wrap=True, row_count=10)
         with gr.Column():
-            gallery = gr.Gallery(columns=3, label="포켓몬 갤러리")
+            gallery = gr.Gallery(columns=3, label="👩🏻‍🎨포켓몬 갤러리")
             report = gr.Markdown("## 타입별 포켓몬 분석")            
 
     def write_pdf(gen, poke_type):
@@ -103,8 +103,9 @@ with gr.Blocks(title="대치동 포켓몬 도감 생성기 🐙") as block:
 
         doc.build(story)
         df = pd.DataFrame(data_dict)
-        return df[['번호', '이름', '타입']], filename, images
+        analysis = df[['이름', '타입']].groupby(by='타입').count().to_markdown().replace('이름', '개수')
+        return df[['번호', '이름', '타입']], filename, images, analysis
 
-    button.click(write_pdf, [generation, poke_types], [df_view, download, gallery])
+    button.click(write_pdf, [generation, poke_types], [df_view, download, gallery, report])
 block.queue(concurrency_count=3)
 block.launch()