Update app.py

app.py

@@ -7,7 +7,7 @@ openai.api_key = os.environ.get('openai-api')
 
 #Functions
 def generate_story(character1, character2, character3, character4, localizer, agefor, language, wordstouse):
-    agefor = min(max(agefor, 2), 20)  # Ensure agefor is between 0 and 20
+    agefor = min(max(agefor, 2), 12)  # Ensure agefor is between 0 and 20
     
     # Define the prompt
     prompt = f'A story about {character1}, {character2}, {character3}, and {character4}, set in {localizer}.' \
@@ -30,7 +30,7 @@ def generate_story(character1, character2, character3, character4, localizer, ag
     )
     
     # Return the story
-    return response.choices[0].text.strip()
+    return response['choices'][0]['message']['content']
 
 with gr.Blocks() as demo:
     gr.Markdown("Enter the details below and then click **Make me a story** to generate your story.")
@@ -46,7 +46,7 @@ with gr.Blocks() as demo:
         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=6, maximum=20, step=1, value=10, label="Age of the child")
+        agefor = gr.Slider(minimum=2, maximum=12, step=1, value=6, label="Age of the child")
         language = gr.Dropdown(choices=["Mandarin", "Cantonese", "French", "English", "Korean", "Italian", "German", "Russian"], label="Language of the story", value="English")
     with gr.Row():
         btn = gr.Button("Make me a bed time story")