Added age settings

app.py

@@ -46,19 +46,19 @@ def join_wav_files(input_files, output_file):
                 with wave.open(input_file, 'rb') as input:
                     output.writeframes(input.readframes(input.getnframes()))
 
-def generate_story(text, lang):
+def generate_story(text, lang, age):
     """Generate a story using the langchain library and OpenAI's GPT-3 model."""
     prompt = PromptTemplate(
-        input_variables=["text", "lang"],
+        input_variables=["text", "lang", "age"],
         template=""" 
         You are a fun and seasoned storyteller. 
-        Generate a short bedtime story for a 5 years old audience about {text}.
+        Generate a short bedtime story for a {age} years old audience about {text}.
         Your story must be written in {lang}.
         Use short sentences. The story is not too long, but not too short either.
         """
     )
     story = LLMChain(llm=llm, prompt=prompt)
-    story_result = story.run(text=text, lang=lang)
+    story_result = story.run(text=text, lang=lang, age=age)
     print(story_result)
     print("""
     —
@@ -92,8 +92,8 @@ def generate_story(text, lang):
     
     return story_result, 'output.mp3'
 
-def app(text, lang):
-    story = generate_story(text, lang)
+def app(text, lang, age):
+    story = generate_story(text, lang, age)
     return story
 
 css = """
@@ -122,13 +122,14 @@ with gr.Blocks(css=css) as demo:
         
         with gr.Group():           
             text = gr.Textbox(label="Subject: what the story should be about ?", info="Will generate and tell a story about {your text input}")
-            lang = gr.Dropdown(label="Pick a language", choices=["English", "French", "German", "Hindi", "Italian", "Polish", "Portuguese", "Spanish"], value="English")
-        
+            with gr.Row():
+                lang = gr.Dropdown(label="Pick a language", choices=["English", "French", "German", "Hindi", "Italian", "Polish", "Portuguese", "Spanish"], value="English")
+                age = gr.Dropdown(label="Age target", choices=["3","4","5","6","7"], value="5")
         submit_btn = gr.Button('Submit')
         
         audio = gr.Audio(label="The story audio told")
         story = gr.Textbox(label="The story text")
 
-    submit_btn.click(fn=app, inputs=[text, lang], outputs=[story, audio])
+    submit_btn.click(fn=app, inputs=[text, lang, age], outputs=[story, audio])
         
 demo.launch()