Update app.py

app.py

@@ -44,7 +44,7 @@ def generate_caption(image):
 def generate_story(caption):
     # Generate the story based on the caption using the GPT-2 model
     prompt = f"Once upon a time, based on the image described as '{caption}', here is a short, simple, and engaging story for children aged 3-10. The story should be easy to understand, use age-appropriate language, and convey a positive message. Focus on the main elements in the image and create a story that sparks their imagination:\n\n"
-    story = story_generator(prompt, max_length=100, num_return_sequences=1)[0]["generated_text"]
+    story = story_generator(prompt, max_length=500, num_return_sequences=1)[0]["generated_text"]
     
     # Extract the story text from the generated output
     story = story.split("\n\n")[1].strip()
@@ -53,8 +53,14 @@ def generate_story(caption):
     inappropriate_words = ["violence", "horror", "scary"]
     for word in inappropriate_words:
         story = story.replace(word, "")
+
+    # Limit the story to approximately 100 words
+    words = story.split()
+    if len(words) > 100:
+        story = " ".join(words[:100]) + "..."
     
     return story
+    
 
 def convert_to_audio(story):
     # Convert the story to audio using gTTS