Create app.py

app.py

@@ -0,0 +1,45 @@
+import gradio as gr
+import openai
+
+openai.api_key = os.environ.get('openai-api')
+
+def generate_story(character1, character2, character3, character4, localizer, agefor, language):
+    agefor = min(max(agefor, 0), 20)  # Ensure agefor is between 0 and 20
+    
+    # Define the prompt
+    prompt = f'A story about {character1}, {character2}, {character3}, and {character4}, set in {localizer}.' \
+             f'The story is intended for a child aged {agefor}.'
+    
+    # Define the request to the OpenAI API
+    response = openai.Completion.create(
+        engine="text-davinci-002",
+        prompt=prompt,
+        temperature=0.9,
+        max_tokens=1000
+    )
+    
+    # Return the story
+    return response.choices[0].text.strip()
+
+with gr.Blocks() as demo:
+    gr.Markdown("Enter the details below and then click **Make me a story** to generate your story.")
+    
+    with gr.Row():
+        character1 = gr.Textbox(label="Character 1")
+        character2 = gr.Textbox(label="Character 2")
+        character3 = gr.Textbox(label="Character 3")
+        character4 = gr.Textbox(label="Character 4")
+    with gr.Row():
+        content = gr.Textbox(label="What you want to hear about in this story", placeholder= "a giant elephant eating pink banana, a broken chair, ...")
+    with gr.Row():
+        localizer = gr.Textbox(placeholder="Location")
+        agefor = gr.Slider(minimum=0, maximum=20, step=1, default=10, label="Age of the child")
+        language = gr.Dropdown(choices=["Mandarin", "Cantonese", "French", "English", "Korean", "Italian", "German", "Russian"], label="Language of the story")
+    with gr.Row():
+        out = gr.Textbox()
+    
+    btn = gr.Button("Make me a bed time story")
+    btn.click(fn=generate_story, inputs=[character1, character2, character3, character4, localizer, agefor, language], outputs=out)
+
+demo.launch()
+