Update app.py

app.py

@@ -4,17 +4,21 @@ from transformers import pipeline
 # Load the text generation pipeline
 generator = pipeline("text-generation")
 
-def generate_bedtime_story(name, interests, friends):
-    story_prompt = f"Once upon a time, there was a child named {name}. {name} loved {', '.join(interests)} and had many friends including {', '.join(friends)}."
-    story = generator(story_prompt, max_length=100)[0]['generated_text']
+def generate_bedtime_story(name, interests, friends, age):
+    story_prompt = f"I want to tell a bed time story to a child named {name}. I want the story to revolve around some of {name}'s main interests, which are {interests}. I also want the story to include some of {name}'s friends, who are called: {friends}. Make sure the story is suitable for a child of the age {age}." 
+    
+    
+    Once upon a time, there was a child named {name}. {name} loved {', '.join(interests)} and had many friends including {', '.join(friends)}."
+    story = generator(story_prompt, max_length=1000)[0]['generated_text']
     return story
 
 def main():
     st.title("Bedtime Story Generator")
 
     name = st.text_input("Enter the name of the person you're reading to:")
-    interests = st.multiselect("Enter the interests of the person:", ["Adventure", "Magic", "Animals", "Science"])
-    friends = st.text_input("Enter the names of some friends:")
+    age = st.text_input("Enter the age of the person you're reading to:")
+    interests = st.text_input("Enter some of things the person is interested in:")
+    friends = st.text_input("Enter the names of some of their friends:")
 
     if st.button("Generate Story"):
         if name and interests and friends: