Update app.py

app.py

@@ -28,15 +28,15 @@ 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, in a world inspired by the image described as '{caption}', a delightful fairy tale unfolded. The story, suitable for children aged 3-10, goes like this:\n\n"
+    prompt = f"Once upon a time, in a world inspired by the image of {caption}, a magical story took place. The fairy tale, suitable for children aged 3-10, unfolds as follows:\n\nIntroduction: Set the scene and introduce the main characters.\n\nConflict: Present a problem or challenge the characters face.\n\nResolution: Describe how the characters solve the problem or overcome the challenge.\n\nMoral: Conclude with a positive message or lesson learned.\n\nThe story should be well-structured, engaging, and easy to understand for young children. Let's begin:\n\n"
     story = story_generator(prompt, max_length=500, num_return_sequences=1)[0]["generated_text"]
     
     # Extract the story text from the generated output
     story_parts = story.split("\n\n")
-    if len(story_parts) > 1:
-        story = "\n\n".join(story_parts[1:]).strip()
+    if len(story_parts) > 5:
+        story = "\n\n".join(story_parts[5:]).strip()
     else:
-        story = story_parts[0].strip()
+        story = "\n\n".join(story_parts).strip()
     
     # Post-process the story (example: remove inappropriate words)
     inappropriate_words = ["violence", "horror", "scary"]