Spaces:
Runtime error
Runtime error
File size: 3,525 Bytes
8127c6a a5e32db 8127c6a a5e32db 8127c6a a5e32db 2839e95 a5e32db d179eb2 a5e32db 2839e95 a5e32db 2839e95 a5e32db 56c012f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
import gradio as gr
import openai
def generate_script(api_key, name1, name2, situation):
# Initialize OpenAI API with the provided key
openai.api_key = api_key
# Define the example script to set the context
example_script = (
"Setting: A cafe\n\n"
"Scene:\n\n"
"(Lights come up on a cozy little coffee shop. LOUIS and GIRL are sitting at a small table, with two untouched coffees between them. "
"It's clear from their expressions that the awkwardness from the previous scene hasn't worn off.)\n\n"
"LOUIS. (Attempting to break the ice) So, um... do you come to this coffee shop often?\n"
"GIRL. (Coldly) No, not really.\n"
"LOUIS. Oh. Well, they have the best almond croissants here. You should try one, sometime.\n"
"GIRL. (Slightly warming up) I'm more of a bagel person, actually.\n"
"LOUIS. Really? Me too! What's your favorite kind?\n"
"GIRL. Plain. With cream cheese.\n"
"LOUIS. Nice choice. I like mine with a bit of jam.\n"
"(A brief pause as they both sip their coffee, seemingly more at ease.)\n\n"
"LOUIS. (Hesitant) Look, I'm sorry about earlier... I don't know why I was acting like such a jerk.\n"
"GIRL. (Softening) It's alright. People can act strange when they meet someone new.\n"
"LOUIS. Yeah. It's just... I've been feeling kind of... out of place lately. And I guess I was trying too hard to impress you.\n"
"GIRL. You don't have to impress me, Louis. Just be yourself.\n"
"(LOUIS smiles shyly and looks down at his coffee. The GIRL does too, and for a moment, there's a comfortable silence between them.)\n\n"
"LOUIS. So... do you have any plans for the weekend?\n"
"GIRL. (Smiling) Not much. Just taking my dog to the park. Maybe catch a movie later. How about you?\n"
"LOUIS. Same here. Minus the dog, though.\n"
"GIRL. (Laughs) Well, maybe you can join me. And who knows, you might just enjoy the simple joy of a walk in the park.\n"
"LOUIS. (Smiling) I'd like that.\n\n"
"(As they continue chatting, the lights dim, suggesting the budding of a new connection between LOUIS and GIRL.)\n\n"
"(Blackout)\n"
)
# Define the prompt to transform the user's inputs into a stage play script
prompt_text = (
f"{example_script}\n\n"
f"Generate a new stage play script based on the following details:\n"
f"Location: Cafe\n"
f"Character 1: {name1}\n"
f"Character 2: {name2}\n"
f"Situation: {situation}\n"
)
# Use OpenAI's Completion endpoint to generate the stage play script
response = openai.Completion.create(engine="davinci", prompt=prompt_text, max_tokens=500)
script = response.choices[0].text.strip()
return script
# Define Gradio Interface
iface = gr.Interface(
fn=generate_script,
inputs=[
gr.components.Textbox(label="OpenAI API Key", type="password"),
gr.components.Textbox(label="Name 1", type="text"),
gr.components.Textbox(label="Name 2", type="text"),
gr.components.Textbox(label="Situation", type="text"),
],
outputs=gr.components.Textbox(label="Generated Stage Play Script", type="text"),
live=True,
title="Stage Play Script Generator",
description="Generate a stage play script set in a cafe based on given characters and situation using OpenAI!",
progress="Generating stage play script...",
)
# Launch the Gradio app
iface.launch()
|