Update app.py

app.py

@@ -36,6 +36,13 @@ def save_as_pdf():
     pdf.output(pdf_path)
     return pdf_path
 
+def create_markdown_table():
+    global chapters
+    markdown_table = "| 챕터 | 이야기 | 그림 |\n|-------|-------|------|\n"
+    for i, chapter in enumerate(chapters, start=1):
+        markdown_table += f"| Chapter {i} | {chapter['story']} | ![Chapter {i} Image]({chapter['image']}) |\n"
+    return markdown_table
+    
 OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
 clipi_client = Client("https://fffiloni-clip-interrogator-2.hf.space/")
 
@@ -74,13 +81,12 @@ def infer(image_input, audience, keyword, protagonist):
     current_story = clipi_result
     return next_chapter(audience, keyword, protagonist)
 
-# CSS for layout
+# Gradio UI
 css = """
 #col-container {max-width: 910px; margin-left: auto; margin-right: auto;}
 a {text-decoration-line: underline; font-weight: 600;}
 """
 
-# Gradio UI
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(
@@ -101,9 +107,22 @@ with gr.Blocks(css=css) as demo:
             with gr.Column():
                 chapter_story = gr.Markdown(label="이야기", elem_id="chapter_story")
                 chapter_image = gr.Image(label="그림", elem_id="chapter_image")
+                table_markdown = gr.Markdown(label="이야기와 그림 정리", elem_id="table_markdown")  # 추가된 마크다운 테이블
 
-    submit_btn.click(fn=infer, inputs=[image_in, audience, keyword_in, protagonist_in], outputs=[chapter_story, chapter_image])
-    next_chapter_btn.click(fn=lambda: next_chapter(audience=audience.value, keyword=keyword_in.value, protagonist=protagonist_in.value), outputs=[chapter_story, chapter_image])
+    submit_btn.click(fn=infer, inputs=[image_in, audience, keyword_in, protagonist_in], outputs=[chapter_story, chapter_image, table_markdown])
+    next_chapter_btn.click(fn=lambda: next_chapter(audience=audience.value, keyword=keyword_in.value, protagonist=protagonist_in.value), outputs=[chapter_story, chapter_image, table_markdown])
     save_pdf_btn.click(fn=save_as_pdf, outputs=gr.File(label="PDF 다운로드"))
 
     demo.queue(max_size=12).launch()
+
+# 수정된 함수
+def next_chapter(audience, keyword, protagonist):
+    global chapter_num, current_story, current_image_url, chapters
+    current_image_url = generate_image_url(current_story)
+    gr.Info(f'Chapter {chapter_num}를 생성하고 있습니다...')
+    chapter_prompt = f"{story_intro}\n\nKeyword: {keyword}\nProtagonist: {protagonist}\n\n![Chapter {chapter_num} Image]({current_image_url})\n\nChapter {chapter_num} 내용을 만들어줘."
+    chat_completion = openai.ChatCompletion.create(model="gpt-3.5-turbo-16k", messages=[{"role": "user", "content": chapter_prompt}])
+    current_story = chat_completion.choices[0].message.content
+    chapters.append({"story": current_story, "image": current_image_url})
+    chapter_num += 1
+    return current_story, current_image_url, create_markdown_table()  # 마크다운 테이블 추가