Update app.py

make it simple for speed

app.py

@@ -18,41 +18,32 @@ def infer(image_input, audience, keyword, protagonist):
     story_intro = f"""
     # Illustrated Tales
     ## Created by [Sigkawat Pengnoo](https://flowgpt.com/prompt/qzv2D3OvHkzkfSE4rQCqv) at FlowGPT
-
     Keyword: {keyword}
     Protagonist: {protagonist}
     한국어로 답변해줘.
-
     STORY : "{{ {clipi_result} }}"
-
     Let's begin with Chapter 1!
     """
 
     chapters = []
     images = []
     current_image_url = generate_image_url(clipi_result)
-    for chapter_num in range(1, 4):  # 3개의 장을 생성합니다.
+    for chapter_num in range(1, 2):  # 1개의 장을 생성합니다.
         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을 생성합니다.
 
     return {
         "chapter1_story": chapters[0],
         "chapter1_image": images[0],
-        "chapter2_story": chapters[1],
-        "chapter2_image": images[1],
-        "chapter3_story": chapters[2],
-        "chapter3_image": images[2],
     }, gr.Group.update(visible=True)
 
 css = """
 #col-container {max-width: 910px; margin-left: auto; margin-right: auto;}
 a {text-decoration-line: underline; font-weight: 600;}
-a {text-decoration-line: underline; font-weight: 600;}
 """
 
 with gr.Blocks(css=css) as demo:
@@ -73,11 +64,7 @@ with gr.Blocks(css=css) as demo:
             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")
-                chapter2_story = gr.Markdown(label="Chapter 2: 이야기", elem_id="chapter2_story")
-                chapter2_image = gr.Image(label="Chapter 2: 그림", elem_id="chapter2_image")
-                chapter3_story = gr.Markdown(label="Chapter 3: 이야기", elem_id="chapter3_story")
-                chapter3_image = gr.Image(label="Chapter 3: 그림", elem_id="chapter3_image")
 
     submit_btn.click(fn=infer, inputs=[image_in, audience, keyword_in, protagonist_in], 
-                     outputs=[chapter1_story, chapter1_image, chapter2_story, chapter2_image, chapter3_story, chapter3_image])
+                     outputs=[chapter1_story, chapter1_image])
     demo.queue(max_size=12).launch()