Create new file

app.py

@@ -0,0 +1,141 @@
+import torch
+from diffusers import StableDiffusionPipeline
+from torch import autocast
+import numpy as np
+import gradio as gr
+import openai
+import os
+
+pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16, use_auth_token=True)  
+
+pipe = pipe.to("cuda")
+
+def generateStory(theme1, theme2):
+    openai.api_key = os.environ.get('openai-key')
+    prompt_text = "Write the first paragraph of a story integrates the themes \"{}\" and \"{}\" in a creative way.\n\nFirst paragraph of story:\n\n".format(theme1,theme2)
+    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
+    )
+    story = response["choices"][0]["text"]
+
+    illustration_response = openai.Completion.create(
+      model="text-davinci-002",
+      prompt="Transform the following story into a caption of an accompanying illustration. Start with 'Beautiful digital illustration of':\n\nStory:\n\nI stand at the edge of the Blue Mountains and gaze out at the vastness before me. It's a beautiful day, and the sun is shining. I can see for miles and miles, and it feels like I'm standing at the edge of the world. I'm here with the person I love, and we're about to embark on a great adventure. I can't wait to explore every inch of this place with them.\n\nIllustration caption:\n\nBeautiful digital illustration of two people standing by the edge of a mountain holding hands looking out\n\nStory:\n\n{}\n\nIllustration caption:".format(story),
+      temperature=0.7,
+      max_tokens=256,
+      top_p=1,
+      frequency_penalty=0,
+      presence_penalty=0
+    )
+
+    image_prompt = illustration_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":
+      prompt = image_prompt
+
+      with autocast("cuda"):
+        image = pipe(prompt).images[0]  # image here is in [PIL format](https://pillow.readthedocs.io/en/stable/)
+    else: 
+      story='Please generate again'
+      image = np.zeros([100,100,3],dtype=np.uint8)
+      image.fill(255) # or img[:] = 255
+
+    if story.startswith('\n\n'):
+      story = story[2:]
+    return story, image
+
+'''
+demo = gr.Interface(
+    fn=themes,
+
+    
+    inputs=["text", "text"],
+    outputs=["text", "image"],
+)
+
+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; max-width: 650px; margin: 0 auto;">
+              <div
+                style="
+                  display: inline-flex;
+                  align-items: center;
+                  gap: 0.8rem;
+                  font-size: 3rem;
+                 
+                  font-style: normal;
+                  font-weight: bold;
+                "
+              >
+                
+                <h1 style="font-weight: 900; margin-bottom: 7px;">
+                  Illustrated Narrative Device
+                </h1>
+              </div>
+              <p style="margin-bottom: 10px; font-size: 94%;">
+                Tickle your imagination* by writing two themes that AI will try to connect in a story and illustration. <br><br> *Use as an imperfect tool for imaginative thought through creative curiosity.
+                
+              </p>
+              
+            </div>
+        """)
+    with gr.Row():        
+      theme1 = gr.Textbox(label='Theme 1', elem_id = 'theme')
+      theme2 = gr.Textbox(label='Theme 2', elem_id = 'theme')
+    
+    b1 = gr.Button("Generate", elem_id="generate-btn")
+
+    story_output = gr.Textbox(label='Story')
+    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, illustration])
+    
+demo.launch()