Update app.py

app.py

@@ -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], illustration_parts[0], story_parts[1], illustration_parts[1], story_parts[2], illustration_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], illustration_parts[0], story_parts[1], illustration_parts[1], story_parts[2], illustration_parts[2])
 
 
 # EXAMPLE
@@ -108,14 +108,14 @@ with gr.Blocks(title="StoryGPT", css="#button{background-color:#4CAF50} #title{t
                     illustration2 = gr.Image(
                         show_label=False, brush_radius=0, lines=9).style(height=200)
 
-            # with gr.Row(variant="compact").style(equal_height=True):
-            #     with gr.Column(scale=3):
-            #         story3 = gr.Textbox(show_label=False, lines=9).style(
-            #             container=False)
+            with gr.Row(variant="compact").style(equal_height=True):
+                with gr.Column(scale=3):
+                    story3 = gr.Textbox(show_label=False, lines=9).style(
+                        container=False)
 
-            #     with gr.Column(scale=1):
-            #         illustration3 = gr.Image(
-            #             show_label=False, brush_radius=0).style(height=200)
+                with gr.Column(scale=1):
+                    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):
@@ -128,7 +128,7 @@ with gr.Blocks(title="StoryGPT", css="#button{background-color:#4CAF50} #title{t
 
     inputs = [key, name, situation, direction]
     # outputs = [story1, illustration1, story2,illustration2, story3, illustration3, story4, illustration4]
-    outputs = [story1, illustration1, story2,illustration2]
+    outputs = [story1, illustration1, story2,illustration2, story3, illustration3]
 
     Generate_btn.click(storyGPT, inputs=inputs, outputs=outputs)