Update app.py

app.py

@@ -31,11 +31,11 @@ def generate_content(image):
 
     # Create prompt for story
     prompt = (
-        f"Write a funny, interesting children's story that precisely centered on this scene {caption}\nStory:"
-        f"mention the exact place, location or venue within {caption}"
-        f"Avoid numbers,random letter combinations\n,and single-letter combinations"
+        f"Write a funny, interesting children's story that is precisely centered on this scene {caption}\nStory:"
+        f" mention the exact place, location or venue within {caption}"
+        f" in third-person narrative, that describes this scene exactly: {caption} "
+        f" Avoid numbers, random letter combinations, and single-letter combinations."
     )
-    
     # Generate raw story
     raw = storyer(
         prompt,
@@ -46,10 +46,10 @@ def generate_content(image):
         return_full_text=False
     )[0]["generated_text"].strip()
 
-    # Define allowed characters to keep (removes symbols like * and ~)
-    allowed_chars = string.ascii_letters + string.digits + " .,!?\"'-"
+    # Define allowed characters to keep (removes digits and symbols like * and ~)
+    allowed_chars = string.ascii_letters + " .,!?\"'-"
     
-    # Clean the raw story by keeping only allowed characters
+    # Clean the raw story by keeping only allowed characters (this filters out any digits)
     clean_raw = ''.join(c for c in raw if c in allowed_chars)
     
     # Split into words and trim to 100 words