Update app.py

app.py

@@ -1,7 +1,6 @@
 from fpdf import FPDF
 import gradio as gr
 from urllib.parse import quote
-import re
 import os
 import openai
 from gradio_client import Client
@@ -42,7 +41,7 @@ def create_markdown_table():
     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/")
 
@@ -61,7 +60,7 @@ def next_chapter(audience, keyword, protagonist):
     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
+    return current_story, current_image_url, create_markdown_table()
 
 def infer(image_input, audience, keyword, protagonist):
     global story_intro, current_story, current_image_url, chapter_num, chapters
@@ -81,7 +80,6 @@ def infer(image_input, audience, keyword, protagonist):
     current_story = clipi_result
     return next_chapter(audience, keyword, protagonist)
 
-# Gradio UI
 css = """
 #col-container {max-width: 910px; margin-left: auto; margin-right: auto;}
 a {text-decoration-line: underline; font-weight: 600;}
@@ -107,22 +105,10 @@ 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")  # 추가된 마크다운 테이블
+                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, 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()  # 마크다운 테이블 추가