Update app.py

app.py

@@ -54,7 +54,6 @@ def generate_story(text, lang):
         Generate a short bedtime story for a 5 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.
-        Always finish your story with "The End".
         """
     )
     story = LLMChain(llm=llm, prompt=prompt)
@@ -111,11 +110,15 @@ with gr.Blocks(css=css) as demo:
                 Generate a bedtime story for a 5 years old audience who needs to get some sleep 😴  
                 </p>
             </div>""")
-        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")
-        submit_btn = gr.Button('Submit')
-        audio = gr.Audio(label="The story audio told")
-        story = gr.Textbox(label="The story text")
+        with gr.Group():
+            with gr.Column():
+                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")
+                submit_btn = gr.Button('Submit')
+        with gr.Group():
+            with gr.Column():
+                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])