Update app.py

app.py

@@ -6,17 +6,20 @@ import os as os
 openai.api_key = os.environ.get('openai-api')
 
 #Functions
-def generate_story(character1, character2, character3, character4, localizer, agefor, language):
-    agefor = min(max(agefor, 0), 20)  # Ensure agefor is between 0 and 20
+def generate_story(character1, character2, character3, character4, localizer, agefor, language, wordstouse):
+    agefor = min(max(agefor, 2), 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}.'
+             f'The story is intended for a child aged {agefor}.' \
+             f'As much as possible use words that are in this list: {wordstouse}.'
+    
+    inject = [{"role":"user", "content":f"prompt"}]
     
     # Define the request to the OpenAI API
-    response = openai.Completion.create(
+    response = openai.ChatCompletion.create(
         engine="gpt-3.5-turbo",
-        prompt=prompt,
+        prompt=inject,
         temperature=0.9,
         max_tokens=1000
     )
@@ -35,14 +38,17 @@ with gr.Blocks() as demo:
     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")
+        wordstouse = gr.Textbox(label="Words for reading", info="Put here the words you would like the child to train on, not compulstory", placeholder= "the then car if this ...")
+    with gr.Row():
+        localizer = gr.Textbox(label="Location", value="Hong Kong", 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")
+        language = gr.Dropdown(choices=["Mandarin", "Cantonese", "French", "English", "Korean", "Italian", "German", "Russian"], label="Language of the story", value="English")
     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)
+        btn = gr.Button("Make me a bed time story")
+    with gr.Row():
+        out = gr.Textbox(label="Here's my story")
+        
+    btn.click(fn=generate_story, inputs=[character1, character2, character3, character4, localizer, agefor, language, wordstouse], outputs=out)
 
 demo.launch()