Update app.py

app.py

@@ -1,95 +1,53 @@
 import numpy as np
+import os
 import gradio as gr
 import openai
-import os
-
-openai.api_key = os.environ['OPENAI_KEY']
-
-def generateStory(theme1, theme2):
-   
-    prompt_text = "Write a short children's story of less than 500 words, which will be read by a parent to their 4-year-old. Make the language witty and funny, and make the story have an interesting twist. The story has to creatively include the following themes: \"{}\" and \"{}\".".format(theme1,theme2)
-    response = openai.Completion.create(
-      engine="text-davinci-003",
-      prompt=prompt_text,
-      temperature=0.7,
-      max_tokens=600,
-      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 = "<|endoftext|>"+story+"\n--\nLabel:",
-          temperature=0,
-          max_tokens=1,
-          top_p=0,
-          logprobs=10
-        )
-    
-    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'
-
-    if story.startswith('\n\n'):
-      story = story[2:]
-    return story
 
 
-def illustratedStory(story, theme1, theme2):
 
-    if story != 'Please generate again':
-      #image_prompt = illustration_response["choices"][0]["text"]
-      image_prompt="A watercolour painting for a children's book of a {} and {}, artstation, award winning, stunning, trending on Artstation, volumetric lighting, ray tracing, 8k, masterpiece, fully adorable".format(theme1,theme2)
-      print(image_prompt)
-      response = openai.Image.create(
-          
-      prompt= image_prompt,
-      n=1,
-      size="1024x1024"
-      )
-      image_url = response['data'][0]['url']
+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)
 
-    else: 
-      image = np.zeros([100,100,3],dtype=np.uint8)
-      image.fill(255) # or img[:] = 255
+    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.'"
+    openai.api_key = os.environ['OPENAI_KEY']
 
-    return image_url
-
-def continueStory(inputStory):
-    prompt_text = inputStory
     response = openai.Completion.create(
-      engine="text-davinci-002",
-      prompt=prompt_text,
-      temperature=0.7,
-      max_tokens=250,
-      top_p=1,
-      frequency_penalty=0,
-      presence_penalty=0
+        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 = "<|endoftext|>"+story+"\n--\nLabel:",
-          temperature=0,
-          max_tokens=1,
-          top_p=0,
-          logprobs=10
-        )
-    
+        model="content-filter-alpha",
+        prompt="{}\n--\nLabel:".format(story),
+        temperature=0,
+        max_tokens=1,
+        top_p=0,
+        logprobs=10
+    )
+
     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
@@ -99,12 +57,7 @@ def continueStory(inputStory):
 
     if story.startswith('\n\n'):
       story = story[2:]
-    return inputStory + story
-
-def downloadStory(story):
-  with open("output.txt", "a") as f:
-    print("Hello stackoverflow!", file=f)
-    print("I have a question.", file=f)
+    return story
 
 '''
 demo = gr.Interface(
@@ -121,62 +74,47 @@ demo.launch()
 
 
 with gr.Blocks(css='''
-.h1 {
-    
-    font-family: HK Grotesk;
-    font-style: normal;
-    font-weight: bold;
-    font-size: 100px;
-    line-height: 105%;
-    margin: 0;
-  }
-  
-  ''') 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 {
+        font-family: HK Grotesk;
+        font-style: normal;
+        font-weight: bold;
+        font-size: 100px;
+        line-height: 105%;
+        margin: 0;
+    }
+''') 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
+                    Infinite Stories
                 </h1>
-              </div>
-              <p style="margin-bottom: 10px; font-size: 94%;">
-                
-              </p>
-              <br>
-              <p style="font-size: 70%;>Generate the beginning of a story by writing two themes, then edit, add to it, extend it and illustrate it! </p>
-              
             </div>
-        """)
-    with gr.Row():        
-      theme1 = gr.Textbox(label='Theme 1', elem_id = 'theme')
-      theme2 = gr.Textbox(label='Theme 2', elem_id = 'theme')
+            <p style="margin-bottom: 10px; font-size: 94%;">
+            </p>
+            <br>
+            
+        </div>
+    ''')
     
-    b1 = gr.Button("Create!", elem_id="generate-btn")
+    with gr.Row():
+        theme1 = gr.Textbox(label='Seed 1', elem_id='theme')
+        theme2 = gr.Textbox(label='Seed 2', elem_id='theme')
 
-    story_output = gr.Textbox(label='')
+    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")
 
-    with gr.Row():
-      b2 = gr.Button("Illustrate Story", elem_id="illustrated-btn")
-    
+    slider = gr.Slider(minimum=0, maximum=10, step=5, label="Approximate reading time", default=5)
 
-    with gr.Row():  
-      illustration = gr.Image(label='Illustration')
-      
-    
-    gr.HTML('<div style="text-align: center; max-width: 650px; margin: 0 auto;"><p style="margin-bottom: 10px; font-size: 94%;">Compute credits are expensive. Please help me keep this experiment running by buying me a coffee <a href="https://www.buymeacoffee.com/jrodolfoocG"> <u><b>here</u></b> :) </a></p></div><br>')
-    gr.HTML('<div style="text-align: center; max-width: 650px; margin: 0 auto;"><p style="margin-bottom: 10px; font-size: 70%;">Built with GPT-3, Stable Diffusion, the Diffusers library and Gradio, by <a href="https://research.rodolfoocampo.com"><u><b>Rodolfo Ocampo</u></b></a></p></div>')
-    
-    
-    b1.click(generateStory, inputs=[theme1,theme2], outputs=[story_output])
-    b2.click(illustratedStory, inputs=[story_output, theme1, theme2], outputs=[illustration])
+    b1 = gr.Button("Create!", elem_id="generate-btn")
 
+    story_output = gr.Textbox(label='Story')
 
-    
-demo.launch(debug=True)
+    b1.click(generateStory, inputs=[theme1, theme2, theme_dropdown, slider], outputs=[story_output])
+
+demo.launch(debug=True, share=True)