Update app.py

app.py

@@ -25,14 +25,14 @@ def storyGPT(key, name, situation, direction):
         system_message = story["choices"][0]["message"]
         messages.append(system_message)
 
-    # for i in ["Third", "Final"]:
-    #     new_story_second_half = {
-    #         "role": "user", "content": f"Write the {i} part of the story among 3 continuing from the previous part"+"in the "+direction+". Include a bold heading above the top starting with '{i} part:'. Limit your answer to 4 sentences or less"}
-    #     messages.append(new_story_second_half)
-    #     story = openai.ChatCompletion.create(
-    #         model="gpt-3.5-turbo", messages=messages)
-    #     system_message = story["choices"][0]["message"]
-    #     messages.append(system_message)
+    for i in ["Third", "Final"]:
+        new_story_second_half = {
+            "role": "user", "content": f"Write the {i} part of the story among 3 continuing from the previous part"+"in the "+direction+". Include a bold heading above the top starting with '{i} part:'. Limit your answer to 4 sentences or less"}
+        messages.append(new_story_second_half)
+        story = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo", messages=messages)
+        system_message = story["choices"][0]["message"]
+        messages.append(system_message)
 
     story_parts = []
     for message in messages:
@@ -52,8 +52,8 @@ def storyGPT(key, name, situation, direction):
     for i in range(len(story_parts)):
         story_parts[i] = story_parts[i].replace(". ", ".\n\n")
 
-    # return (story_parts[0], illustration_parts[0], story_parts[1], illustration_parts[1], story_parts[2], illustration_parts[2], story_parts[3], illustration_parts[3])
-    return (story_parts[0], story_parts[1], story_parts[2])
+    return (story_parts[0], illustration_parts[0], story_parts[1], illustration_parts[1], story_parts[2], illustration_parts[2], story_parts[3], illustration_parts[3])
+    # return (story_parts[0], story_parts[1], story_parts[2])
 
 
 # EXAMPLE
@@ -117,18 +117,18 @@ with gr.Blocks(title="StoryGPT", css="#button{background-color:#4CAF50} #title{t
                     illustration3 = gr.Image(
                         show_label=False, brush_radius=0).style(height=200)
 
-            # with gr.Row(variant="compact").style(equal_height=True):
-            #     with gr.Column(scale=3):
-            #         story4 = gr.Textbox(show_label=False, lines=9).style(
-            #             container=False)
+            with gr.Row(variant="compact").style(equal_height=True):
+                with gr.Column(scale=3):
+                    story4 = gr.Textbox(show_label=False, lines=9).style(
+                        container=False)
 
-            #     with gr.Column(scale=1):
-            #         illustration4 = gr.Image(
-            #             show_label=False, brush_radius=0).style(height=200)
+                with gr.Column(scale=1):
+                    illustration4 = gr.Image(
+                        show_label=False, brush_radius=0).style(height=200)
 
     inputs = [key, name, situation, direction]
-    # outputs = [story1, illustration1, story2,illustration2, story3, illustration3, story4, illustration4]
-    outputs = [story1, illustration1, story2,illustration2, story3, illustration3]
+    outputs = [story1, illustration1, story2,illustration2, story3, illustration3, story4, illustration4]
+    # outputs = [story1, illustration1, story2,illustration2, story3, illustration3]
 
     Generate_btn.click(storyGPT, inputs=inputs, outputs=outputs)