Update app.py

app.py

@@ -8,24 +8,27 @@ 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 = []
+chapter_num = 1
+story_intro = ""
+current_story = ""
+current_image_url = ""
 
 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 "이미 모든 장을 읽었습니다.", ""
+    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} 내용을 만들어줘."
+    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
+    return current_story, current_image_url
 
 def infer(image_input, audience, keyword, protagonist):
-    global chapters, images
+    global story_intro, current_story, current_image_url, chapter_num
+    chapter_num = 1
     gr.Info('Calling CLIP Interrogator, 이미지를 해석하고 있습니다...')
     clipi_result = clipi_client.predict(image_input, "best", 4, api_name="/clipi2")[0]
 
@@ -39,18 +42,7 @@ def infer(image_input, audience, keyword, protagonist):
     Let's begin with Chapter 1!
     """
 
-    chapters.clear()
-    images.clear()
-    current_image_url = generate_image_url(clipi_result)
-    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)
-        current_image_url = generate_image_url(chapter_story)  # 다음 장의 이미지 URL을 생성합니다.
-
+    current_story = clipi_result
     return next_chapter()
 
 css = """
@@ -75,13 +67,13 @@ with gr.Blocks(css=css) as demo:
                 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")
+                chapter_story = gr.Markdown(label="이야기", elem_id="chapter_story")
+                chapter_image = gr.Image(label="그림", elem_id="chapter_image")
 
     submit_btn.click(fn=infer, inputs=[image_in, audience, keyword_in, protagonist_in], 
-                     outputs=[chapter1_story, chapter1_image])
+                     outputs=[chapter_story, chapter_image])
 
     next_chapter_btn.click(fn=next_chapter, 
-                           outputs=[chapter1_story, chapter1_image])
+                           outputs=[chapter_story, chapter_image])
 
     demo.queue(max_size=12).launch()