Update app.py

app.py

@@ -5,116 +5,136 @@ import openai
 
 
 
-def generateStory(theme1, theme2, teaching, read_time):
-    if read_time > 0:
-      word_counts = read_time*200
-    else:
-      read_time=1
-      word_counts = read_time*200
-    prompt_text = ""
-    if teaching == "No teaching":
-        prompt_text = "Write a long children's story that combines the following two topics: {} and {}. Your story should be appropriate, understandable and fun for children aged around 10 years old. Use descriptive language, dialogue and intricate details and vivid imagery to bring your the story life.".format(theme1, theme2)
-    elif teaching == "Open teaching":
-        prompt_text = "Write a long children's story that combines the following two topics: {} and {}. Your story should be appropriate, understandable and fun for children aged around 10 years old and should include a positive message or moral. Use descriptive language, dialogue and intricate details and vivid imagery to bring your the story life. ".format(theme1, theme2)
-    else:
-        prompt_text = "Write a long children's story that combines the following two topics: {} and {}. Your story should be appropriate, understandable and fun for children aged around 10 years old and should teach children about {}. Use descriptive language, dialogue and intricate details and vivid imagery to bring your the story life. ".format(theme1, theme2, teaching)
-
-    prompt_text += f" The story should be approximately {word_counts} words long, corresponding to approximately a {read_time} minute read."
-    prompt_text += "The story should have a tone similar to this: 'Once there was an evil wizard who made a mirror with his dark magic. If anything good or beautiful was put in front of the mirror, the reflection that showed back was only rotten and gray. The wizard laughed. He wanted to show his evil mirror to the whole world! He took it and flew up high into the sky.'"
-    
+def generateStory(character):
+   
     openai.api_key = os.environ['OPENAI_KEY']
-    response = openai.Completion.create(
-        engine="text-davinci-003",
-        prompt=prompt_text,
-        temperature=0.7,
-        max_tokens=2300,
-        top_p=1,
-        frequency_penalty=0,
-        presence_penalty=0
-    )
-    story = response["choices"][0]["text"]
-
-    content_to_classify = "Your content here"
 
-    response = openai.Completion.create(
-        model="content-filter-alpha",
-        prompt="{}\n--\nLabel:".format(story),
-        temperature=0,
-        max_tokens=1,
-        top_p=0,
-        logprobs=10
+    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."},
+            {"role": "user", "content": f"Please write the first paragraph of a story about {character}. The story needs to be set in a magic island."}
+        ]
+    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
 
-    output_label = response["choices"][0]["text"]
-
-
-    # This is the probability at which we evaluate that a "2" is likely real
-    # vs. should be discarded as a false positive
-    toxic_threshold = -0.355
 
-    if output_label == "2":
-      story='Please generate again'
+def continueStory(blank, messageHistory):
 
-    if story.startswith('\n\n'):
-      story = story[2:]
-    return story
+    openai.api_key = os.environ['OPENAI_KEY']
 
-'''
-demo = gr.Interface(
-    fn=themes,
+    messageHistory.append({"role": "user", "content": blank + "\n\nGenerate the next paragraph with a blank in the last line for me to fill."})
 
+    response = openai.ChatCompletion.create(
+      model="gpt-3.5-turbo",
+      messages = messageHistory
+    )
     
-    inputs=["text", "text"],
-    outputs=["text", "image"],
-)
-
-demo.launch()
-'''
-
+    story = response["choices"][0]["message"]["content"]
+    messageHistory.append(response["choices"][0]["message"])
+    print(messageHistory)
+    return story, messageHistory
 
 
 with gr.Blocks(css='''
-    .h1 {
-        font-family: HK Grotesk;
-        font-style: normal;
-        font-weight: bold;
-        font-size: 100px;
-        line-height: 105%;
-        margin: 0;
-    }
+  .gradio-container {
+    display: flex;
+    flex-direction: column;
+    justify-content: center;
+    align-items: center;
+    background-image: url('file=https://cdn.discordapp.com/attachments/941582479117127680/1080730421425344542/rodotcom_colorful_abstract_illustration_for_the_background_of_c_f1b331d7-6493-4a33-9063-345d31a66ddb.png');
+    padding: 1rem;
+  }
+
+  h1 {
+    font-size: 3rem;
+    font-weight: 700;
+    margin-bottom: 1rem;
+    color: white;
+  }
+
+  p {
+    font-size: 1.25rem;
+    margin-bottom: 2rem;
+    color: white;
+  }
+
+  .gradio-row {
+    width: 100%;
+    margin-bottom: 1.5rem;
+  }
+
+  label {
+    font-size: 1.25rem;
+    font-weight: 500;
+    margin-bottom: 0.5rem;
+  }
+
+  input[type="text"], textarea {
+    font-size: 1.25rem;
+    padding: 0.5rem;
+    border: 2px solid #ccc;
+    border-radius: 5px;
+    width: 100%;
+  }
+
+  .gradio-button {
+    background-color: #2196f3;
+    color: #fff;
+    font-size: 1.25rem;
+    font-weight: 500;
+    padding: 0.75rem 1.5rem;
+    border-radius: 5px;
+    cursor: pointer;
+    transition: all 0.2s ease-in-out;
+  }
+
+  .gradio-button:hover {
+    background-color: #0c7cd5;
+  }
+
+  .gradio-textbox {
+    font-size: 1.25rem;
+    padding: 0.5rem;
+    border: 2px solid #ccc;
+    border-radius: 5px;
+    width: 100%;
+    height: 200px;
+    resize: none;
+  }
+
 ''') as demo:
     title = gr.HTML('''
-        <div style="text-align: center; margin: 0;">
-            <div style="
-                display: inline-flex;
-                align-items: center;
-                gap: 0.8rem;
-                font-size: 1.75rem;
-            ">
-                <h1 style="font-weight: 900; margin-bottom: 7px;">
-                    Infinite Stories
-                </h1>
-            </div>
-            <p style="margin-bottom: 10px; font-size: 94%;">
-            </p>
-            <br>
-            
+        <div style="text-align: center;">
+            <h1 style="color: white; font-size: 3rem; font-weight: 700; margin-bottom: 1rem; color: white;">Infinite Stories</h1>
+            <p style="color: white; font-size: 1.25rem; margin-bottom: 2rem; color: white;">Enter a character name and let the AI generate a story for you!</p>
         </div>
     ''')
-    
-    with gr.Row():
-        theme1 = gr.Textbox(label='Seed 1', elem_id='theme')
-        theme2 = gr.Textbox(label='Seed 2', elem_id='theme')
-
-    theme_options = ["No teaching", "Open teaching", "Gratitude", "The value of friendship", "Caring for the planet", "Believing in yourself", "Being brave", "Choosing to do the right thing", "Accepting people different than us", "Being compassionate"]
-    theme_dropdown = gr.Dropdown(choices=theme_options, label="Moral")
-
-    slider = gr.Slider(minimum=0, maximum=10, step=5, label="Approximate reading time", default=5)
+    with gr.Row():        
+      character = gr.Textbox(label='Character', elem_id = 'theme')
+      
+    messageHistory = gr.State()
 
-    b1 = gr.Button("Create!", elem_id="generate-btn")
+    b1 = gr.Button("Generate!", elem_id="generate-btn")
 
     story_output = gr.Textbox(label='Story')
 
-    b1.click(generateStory, inputs=[theme1, theme2, theme_dropdown, slider], outputs=[story_output])
+    blank = gr.Textbox(label='How will you fill the blank?', elem_id = 'blank')
+
+    b3 = gr.Button("Continue Story", elem_id="continue-btn") 
+     
+    
+    
+    
+    b1.click(generateStory, inputs=[character], outputs=[story_output, messageHistory])
+    
 
+    b3.click(continueStory, inputs=[blank, messageHistory], outputs=[story_output, messageHistory])
+    
+    
 demo.launch(debug=True, share=False)