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 5 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 5 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,7 +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], 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
@@ -107,27 +108,27 @@ 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):
-                    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, story4, illustration4]
+    outputs = [story1, illustration1, story2,illustration2]
 
     Generate_btn.click(storyGPT, inputs=inputs, outputs=outputs)