Update app.py

app.py

@@ -3,12 +3,13 @@ import requests
 from PIL import Image, ImageDraw, ImageFont
 import os
 from transformers import BlipProcessor, BlipForConditionalGeneration
+from random import choice
 
 # Load the BLIP model for creative caption generation
 processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
 model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
 
-# Function to generate a creative caption
+# Function to generate a detailed, creative caption
 def generate_caption(image):
     # Prepare image for the model
     inputs = processor(images=image, return_tensors="pt")
@@ -19,13 +20,30 @@ def generate_caption(image):
     # Decode the output to get a readable caption
     caption = processor.decode(out[0], skip_special_tokens=True)
     
-    # Return the creative caption
-    return caption
+    # Enhance the caption with more details (assuming the image context is about nature, people, etc.)
+    detailed_caption = enhance_caption(caption)
+    
+    return detailed_caption
+
+# Enhance caption with surrounding context and social media flavor
+def enhance_caption(caption):
+    # Possible enhancements for social media flavor
+    emojis = ["🌅", "🌊", "🌻", "🌴", "🏖️", "🦋", "🌞", "🌇"]
+    hashtags = ["#NatureLovers", "#TravelVibes", "#Sunset", "#BeachLife", "#AdventureAwaits"]
+    
+    # Select random emojis and hashtags to add a fun social media touch
+    selected_emoji = choice(emojis)
+    selected_hashtags = " ".join(choice(hashtags) for _ in range(2))  # Select two random hashtags
+    
+    # Creating a more detailed, fun caption for social media
+    enhanced_caption = f"{caption} {selected_emoji} {selected_hashtags}"
+    
+    return enhanced_caption
 
 # Streamlit app
 def main():
     st.title("Creative Image Caption Generator")
-    st.write("Upload an image, and let the AI generate a creative and descriptive caption for it.")
+    st.write("Upload an image, and let the AI generate a creative and descriptive caption for it, ready for social media sharing!")
     
     # Upload image
     uploaded_file = st.file_uploader("Upload an Image", type=["jpg", "png", "jpeg"])