Update app.py

app.py

@@ -8,10 +8,24 @@ from gradio_client import Client
 OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
 clipi_client = Client("https://fffiloni-clip-interrogator-2.hf.space/")
 
+chapter_num = 0
+chapters = []
+images = []
+
 def generate_image_url(keywords):
     return f"https://image.pollinations.ai/prompt/{quote(keywords)}"
 
+def next_chapter():
+    global chapter_num
+    if chapter_num < len(chapters):
+        result = chapters[chapter_num], images[chapter_num]
+        chapter_num += 1
+        return result
+    else:
+        return "이미 모든 장을 읽었습니다.", ""
+
 def infer(image_input, audience, keyword, protagonist):
+    global chapters, images
     gr.Info('Calling CLIP Interrogator, 이미지를 해석하고 있습니다...')
     clipi_result = clipi_client.predict(image_input, "best", 4, api_name="/clipi2")[0]
 
@@ -25,22 +39,19 @@ def infer(image_input, audience, keyword, protagonist):
     Let's begin with Chapter 1!
     """
 
-    chapters = []
-    images = []
+    chapters.clear()
+    images.clear()
     current_image_url = generate_image_url(clipi_result)
-    for chapter_num in range(1, 2):  # 1개의 장을 생성합니다.
+    for chapter_num in range(1, 4):  # 3개의 장을 생성합니다.
         gr.Info(f'Chapter {chapter_num}를 생성하고 있습니다...')
         chapter_prompt = f"{story_intro}\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}])
         chapter_story = chat_completion.choices[0].message.content
         chapters.append(chapter_story)
         images.append(current_image_url)
-        
-    return chapters[0], images[0]
-    # return {
-    #     "chapter1_story": chapters[0],
-    #     "chapter1_image": images[0],
-    # }, gr.Group.update(visible=True)
+        current_image_url = generate_image_url(chapter_story)  # 다음 장의 이미지 URL을 생성합니다.
+
+    return next_chapter()
 
 css = """
 #col-container {max-width: 910px; margin-left: auto; margin-right: auto;}
@@ -62,9 +73,15 @@ with gr.Blocks(css=css) as demo:
                 keyword_in = gr.Textbox(label="핵심 키워드")
                 protagonist_in = gr.Textbox(label="주인공")
                 submit_btn = gr.Button('이야기와 그림을 만들어 주세요')
+                next_chapter_btn = gr.Button('다음 이야기')
             with gr.Column():
                 chapter1_story = gr.Markdown(label="Chapter 1: 이야기", elem_id="chapter1_story")
                 chapter1_image = gr.Image(label="Chapter 1: 그림", elem_id="chapter1_image")
+
     submit_btn.click(fn=infer, inputs=[image_in, audience, keyword_in, protagonist_in], 
                      outputs=[chapter1_story, chapter1_image])
+
+    next_chapter_btn.click(fn=next_chapter, 
+                           outputs=[chapter1_story, chapter1_image])
+
     demo.queue(max_size=12).launch()