Update app.py

app.py

@@ -95,7 +95,58 @@ with gr.Blocks(css='''
         </div>
     ''')
 
-    with gr.Column(elem_id="main_col") as main_col:
+        # the submit feedback btn needs to: send the feedback, 
+    # hide the elements and provide a message saying thanks for your feedback
+        
+    def show_feedback_text():
+        return {feedbackForm: gr.update(visible=True)}
+
+    def submitFeedback():
+        return {thankyou: gr.update(visible=True), feedbackForm: gr.update(visible=False)}
+
+    def generateStory(character):
+   
+      openai.api_key = os.environ['OPENAI_KEY']
+
+      messageHistory=[
+              {"role": "system", "content": "You are co-writing assistant that will help me generate an interactive storytelling game. You are going to generate one paragraph at a time but you are going to leave a blank somewhere in the last line of the paragraph. The user will have to fill this blank in order to generate the next paragraph. You have to write in the style of Julia Donaldson and Doctor Seuss. Make the story funny and exciting. For more context, the game works as follows. 1. You generate one, and only one paragraph, with a blank somewhere in the last line of the paragraph. 2. The user tells you how they want to fill the blank. 3. You generate the next paragraph based on the user’s response to continue the story. This next paragraph also has a blank somewhere in the last line of the paragraph. 4. The user tells you how they want to fill the blank. And so on. Please make sure that when you write a paragraph, there is only one blank and it is at the end of the last line."},
+              {"role": "user", "content": f"Please write the first paragraph of a story about {character}. The story needs to be set in a magic island. Make sure you only write one paragraph, that is not too long. Also, make sure is has a blank in the last line of the last paragraph for me to fill. There should only be one blank in the paragraph. The blank should be at the end of the last line of the paragraph. It is very important that there is only one blank and that it is at the end, and that you only generate one paragraph!"}
+          ]
+      response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages = messageHistory
+      )
+      
+      story = response["choices"][0]["message"]["content"]
+      messageHistory.append(response["choices"][0]["message"])
+    
+      return story, messageHistory
+
+    def finishStory(blank, messageHistory):
+
+      openai.api_key = os.environ['OPENAI_KEY']
+
+      messageHistory.append({"role": "user", "content": blank + "\n\nNow bring the story to a close. Write the necessary paragraphs to make it have a happy or funny ending. No need to leave a blank anymore."})
+
+
+      response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages = messageHistory
+      )
+      
+      story = response["choices"][0]["message"]["content"]
+      messageHistory.append(response["choices"][0]["message"])
+      print(messageHistory)
+      return story, messageHistory
+
+    def createNewStory():
+        return {character: gr.update(value=""), story_output: gr.update(value="")}
+
+    def hideContinueBtn():
+      return {continueStoryCol: gr.update(visible=False), newStoryCol: gr.update(visible=True), finishCol: gr.update(visible=False)}
+    
+
+    with gr.Column(elem_id="main_col"):
         
                
         character = gr.Textbox(label='Who will be the main character?', elem_id = 'theme', placeholder="i.e. Diego, an 11-year-old pirate")
@@ -134,69 +185,18 @@ with gr.Blocks(css='''
           messageHistoryTextBox = gr.Textbox(label='How will you fill the blank?', elem_id = 'blank', value=messageHistory)
           
     
-    # the submit feedback btn needs to: send the feedback, 
-    # hide the elements and provide a message saying thanks for your feedback
-        
-        def show_feedback_text():
-            return {feedbackForm: gr.update(visible=True)}
-    
-        def submitFeedback():
-            return {thankyou: gr.update(visible=True), feedbackForm: gr.update(visible=False)}
-    
-        def generateStory(character):
-       
-          openai.api_key = os.environ['OPENAI_KEY']
-    
-          messageHistory=[
-                  {"role": "system", "content": "You are co-writing assistant that will help me generate an interactive storytelling game. You are going to generate one paragraph at a time but you are going to leave a blank somewhere in the last line of the paragraph. The user will have to fill this blank in order to generate the next paragraph. You have to write in the style of Julia Donaldson and Doctor Seuss. Make the story funny and exciting. For more context, the game works as follows. 1. You generate one, and only one paragraph, with a blank somewhere in the last line of the paragraph. 2. The user tells you how they want to fill the blank. 3. You generate the next paragraph based on the user’s response to continue the story. This next paragraph also has a blank somewhere in the last line of the paragraph. 4. The user tells you how they want to fill the blank. And so on. Please make sure that when you write a paragraph, there is only one blank and it is at the end of the last line."},
-                  {"role": "user", "content": f"Please write the first paragraph of a story about {character}. The story needs to be set in a magic island. Make sure you only write one paragraph, that is not too long. Also, make sure is has a blank in the last line of the last paragraph for me to fill. There should only be one blank in the paragraph. The blank should be at the end of the last line of the paragraph. It is very important that there is only one blank and that it is at the end, and that you only generate one paragraph!"}
-              ]
-          response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages = messageHistory
-          )
-          
-          story = response["choices"][0]["message"]["content"]
-          messageHistory.append(response["choices"][0]["message"])
-        
-          return story, messageHistory
-    
-        def finishStory(blank, messageHistory):
-    
-          openai.api_key = os.environ['OPENAI_KEY']
-    
-          messageHistory.append({"role": "user", "content": blank + "\n\nNow bring the story to a close. Write the necessary paragraphs to make it have a happy or funny ending. No need to leave a blank anymore."})
-    
-    
-          response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages = messageHistory
-          )
-          
-          story = response["choices"][0]["message"]["content"]
-          messageHistory.append(response["choices"][0]["message"])
-          print(messageHistory)
-          return story, messageHistory
-    
-        def createNewStory():
-            return {character: gr.update(value=""), story_output: gr.update(value="")}
-    
-        def hideContinueBtn():
-          return {continueStoryCol: gr.update(visible=False), newStoryCol: gr.update(visible=True), finishCol: gr.update(visible=False)}
+    beginBtn.click(generateStory, inputs=[character], outputs=[story_output, messageHistory])
     
-    
-        beginBtn.click(generateStory, inputs=[character], outputs=[story_output, messageHistory])
-        
-        continueBtn.click(continueStory, inputs=[blank, messageHistory], outputs=[story_output, messageHistory])
-    
-        finishBtn.click(finishStory, inputs=[blank, messageHistory], outputs=[story_output, messageHistory])
-        finishBtn.click(hideContinueBtn, [], [continueStoryCol, newStoryCol, finishCol])
-    
-        provideFeedbackBtn.click(show_feedback_text, [], [feedbackForm])
-    
-        submitFeedbackBtn.click(submitFeedback, [], [thankyou, feedbackForm])
-    
-        newStoryBtn.click(createNewStory, [], [character, story_output])
+    continueBtn.click(continueStory, inputs=[blank, messageHistory], outputs=[story_output, messageHistory])
+
+    finishBtn.click(finishStory, inputs=[blank, messageHistory], outputs=[story_output, messageHistory])
+    finishBtn.click(hideContinueBtn, [], [continueStoryCol, newStoryCol, finishCol])
+
+    provideFeedbackBtn.click(show_feedback_text, [], [feedbackForm])
+
+    submitFeedbackBtn.click(submitFeedback, [], [thankyou, feedbackForm])
+
+    newStoryBtn.click(createNewStory, [], [character, story_output])
 
 
 demo.launch(debug=True, share=False)