Update app.py

app.py

@@ -19,15 +19,7 @@ def load_pipelines():
 
 captioner, storyer, tts = load_pipelines()
 
-# Function to extract keywords from caption
-def extract_keywords(caption):
-    # Simple keyword extraction: split words and filter out common stop words
-    stop_words = {'the', 'a', 'an', 'in', 'on', 'at', 'of', 'to', 'is', 'are', 'with', 'and'}
-    words = caption.lower().split()
-    # Keep words longer than 2 characters and not in stop words
-    keywords = [word.strip(".,!?\"'") for word in words if word not in stop_words and len(word) > 2]
-    return keywords
-
+# Function part
 # Function to generate content from an image
 def generate_content(image):
     pil_image = Image.open(image)
@@ -37,16 +29,11 @@ def generate_content(image):
     st.write("**🌟 What's in the picture: 🌟**")
     st.write(caption)
 
-    # Extract keywords from the caption
-    keywords = extract_keywords(caption)
-    keywords_str = ", ".join(keywords)
-
-    # Create prompt for story, ensuring keywords are included
+    # Create prompt for story
     prompt = (
         f"Write a funny, warm children's story for ages 3-10, 50–100 words, "
-        f"in third-person narrative, that describes this scene exactly: {caption}. "
-        f"Explicitly include these keywords from the caption in the story: {keywords_str}. "
-        f"Mention the exact place, location, or venue within {caption}, such as a park, pool, or gym."
+        f"in third-person narrative, that describes this scene exactly: {caption} "
+        f"mention the exact place, location or venue within {caption},location example like park,pool,gym"
     )
     
     # Generate raw story
@@ -84,17 +71,6 @@ def generate_content(image):
     return caption, story, temp_file_path
 
 # Streamlit UI
-st.markdown(
-    """
-    <style>
-    .stApp {
-        background: radial-gradient(circle, #e6f3ff, #e6fff2);
-    }
-    </style>
-    """,
-    unsafe_allow_html=True
-)
-
 st.title("✨ Magic Story Maker ✨")
 st.markdown("Upload a picture to make a funny story and hear it too! 📸")