Update app.py

app.py

@@ -30,11 +30,16 @@ def text2story(text):
     Returns:
         str: Generated story suitable for kids aged 3-10.
     """
-    # Prompt to ensure the story is happy and suitable for kids
-    prompt = f"Write a short, happy, and fun story for kids aged 3-10. The story should be about: {text}. Make it cheerful, imaginative, and suitable for young children. Avoid any scary or sad elements. The story should be under 95 words."
+    # Adjusted prompt to ensure the model generates a story
+    prompt = f"Write a short, happy, and fun story for kids aged 3-10 about: {text}. Make it cheerful, imaginative, and suitable for young children. Avoid any scary or sad elements. The story should be under 95 words."
     
     text_generator = pipeline("text2text-generation", model="t5-small")
-    story = text_generator(prompt, max_length=100)[0]['generated_text']
+    story = text_generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
+    
+    # Remove any repetition of the prompt in the output
+    if prompt in story:
+        story = story.replace(prompt, "").strip()
+    
     return story
 
 # Function to convert text to audio using gTTS