Update app.py

app.py

@@ -15,12 +15,11 @@ current_image_url = ""
 
 def generate_image_url(keywords):
     return f"https://image.pollinations.ai/prompt/{quote(keywords)}"
-
-def next_chapter():
+def next_chapter(audience, keyword, protagonist):
     global chapter_num, current_story, current_image_url
     current_image_url = generate_image_url(current_story)  # 다음 장의 이미지 URL을 생성합니다.
     gr.Info(f'Chapter {chapter_num}를 생성하고 있습니다...')
-    chapter_prompt = f"{story_intro}\n\n![Chapter {chapter_num} Image]({current_image_url})\n\nChapter {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
     chapter_num += 1
@@ -72,8 +71,9 @@ with gr.Blocks(css=css) as demo:
 
     submit_btn.click(fn=infer, inputs=[image_in, audience, keyword_in, protagonist_in], 
                      outputs=[chapter_story, chapter_image])
-
-    next_chapter_btn.click(fn=next_chapter, 
-                           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])
+    # next_chapter_btn.click(fn=next_chapter, 
+    #                        outputs=[chapter_story, chapter_image])
 
     demo.queue(max_size=12).launch()