Update app.py

app.py

@@ -159,122 +159,264 @@ def generate_short_caption(description, max_retries=5):
     return ""
 
 def generate_instagram_captions(description):
-    """Generate 5 short, catchy Instagram captions (max 5 words each)"""
+    """Generate 5 short, catchy Instagram captions (max 5 words each) using cool captions"""
     if not description.strip():
         return "Please enter a description of your photo! 📸"
     
-    # Keyword-based caption training for specific contexts
-    keyword_captions = {
-        # Beach/Sea related
-        "beach": ["Beach vibes only"],
-        "sea": ["Sea you later"],
-        "ocean": ["Ocean child wild heart"],
-        "sand": ["Sandy toes salty kisses"],
-        "sun": ["Chasing golden hour"],
-        "sunset": ["Golden hour magic"],
-        "swimming": ["Making waves today"],
-        "surfing": ["Riding the waves"],
-        
-        # Food related
-        "food": ["Good food good mood"],
-        "coffee": ["But first coffee"],
-        "pizza": ["Pizza my heart"],
-        "brunch": ["Brunch squad assembled"],
-        
-        # Fitness/Health
-        "gym": ["Sweat now shine later"],
-        "workout": ["No pain no gain"],
-        "yoga": ["Peace love and yoga"],
-        "running": ["Run like you stole"],
-        
-        # Travel/Adventure
-        "travel": ["Wanderlust and city dust"],
-        "adventure": ["Adventure is out there"],
-        "hiking": ["Take only pictures"],
-        "camping": ["Into the wild"],
-        
-        # Party/Celebration
-        "party": ["Party like its"],
-        "birthday": ["Another year wiser"],
-        "celebration": ["Cheers to good times"],
-        
-        # Friends/Social
-        "friends": ["Squad goals right here"],
-        "squad": ["Squad goals achieved"],
-        "bestie": ["Partner in crime"],
-        
-        # Fashion/Style
-        "outfit": ["Outfit on point"],
-        "fashion": ["Fashion is art"],
-        "style": ["Style is eternal"],
-        
-        # Love/Romance
-        "love": ["Love is everything"],
-        "couple": ["Better together always"],
-        "date": ["Date night vibes"],
-        
-        # Nature/Outdoors
-        "nature": ["Nature is medicine"],
-        "flowers": ["Bloom where planted"],
-        "garden": ["Garden therapy session"],
-        
-        # Work/Success
-        "work": ["Work hard play harder"],
-        "success": ["Success tastes sweet"],
-        "business": ["Business mode activated"]
+    # Massive database of cool modern captions organized by vibe
+    cool_captions_db = {
+        "general_vibes": [
+            "Good vibes only today",
+            "Living our best life",
+            "Squad goals right here",
+            "Making memories with favorites",
+            "Caught in the moment",
+            "Energy levels unmatched today",
+            "Vibes on point always",
+            "Good times roll forever",
+            "Living for these moments",
+            "Making magic happen daily",
+            "Good company great memories",
+            "Vibes immaculate as always",
+            "Energy infectious spread everywhere",
+            "Making waves together always",
+            "Living life in color",
+            "Good vibes attract everything",
+            "Caught being absolutely legendary",
+            "Creating our own magic",
+            "Blessed with the best",
+            "Dreams do come true",
+            "Forever young at heart",
+            "Happiness looks like this"
+        ],
+        "friendship_squad": [
+            "Squad deeper than ocean",
+            "Friends like these matter",
+            "Squad tight bond stronger",
+            "Good energy good people",
+            "Squad up for success",
+            "Squad goals achieved today",
+            "Team work makes dreams",
+            "Squad deeper than family",
+            "Friends forever and always",
+            "Squad goals right here",
+            "Squad assembled ready always",
+            "Squad till the end",
+            "Squad stronger than ever",
+            "Squad vibes unmatched today",
+            "Squad goals achieved finally",
+            "Squad energy off charts",
+            "Squad love runs deep",
+            "Squad goals never ending",
+            "Squad till we die",
+            "Squad brings out best"
+        ],
+        "confident_boss": [
+            "Boss moves only today",
+            "Confidence level maximum reached",
+            "Main character energy activated",
+            "That girl energy today",
+            "Boss babe vibes only",
+            "Confidence is my outfit",
+            "Serving looks and attitude",
+            "Boss energy never fades",
+            "Confidence on ten always",
+            "Boss mode permanently on",
+            "Confidence through the roof",
+            "Boss vibes all day",
+            "Confidence is my superpower",
+            "Boss lady energy strong",
+            "Confidence level expert mode",
+            "Boss moves make history",
+            "Confidence speaks for itself",
+            "Boss energy infectious spreading",
+            "Confidence is my weapon",
+            "Boss vibes never off"
+        ],
+        "aesthetic_mood": [
+            "Aesthetic vibes only today",
+            "Mood captured perfectly here",
+            "Aesthetic game too strong",
+            "Vibes aesthetic perfection achieved",
+            "Mood board inspiration daily",
+            "Aesthetic energy flowing freely",
+            "Mood ring says happy",
+            "Aesthetic goals achieved today",
+            "Mood boost successfully delivered",
+            "Aesthetic vibes never ending",
+            "Mood elevated to maximum",
+            "Aesthetic perfection in progress",
+            "Mood lighting hits different",
+            "Aesthetic energy through roof",
+            "Mood captured in pixels",
+            "Aesthetic vibes permanently on",
+            "Mood board worthy moment",
+            "Aesthetic goals finally achieved",
+            "Mood enhanced by beauty",
+            "Aesthetic vibes multiply daily"
+        ],
+        "fun_playful": [
+            "Chaos and we prosper",
+            "Weekend warriors in action",
+            "Sunshine mixed with chaos",
+            "Adventure awaits us always",
+            "Mischief managed successfully today",
+            "Chaos coordinator reporting duty",
+            "Fun police cant stop",
+            "Chaos with a purpose",
+            "Mischief level expert achieved",
+            "Chaos brings us together",
+            "Fun mode permanently activated",
+            "Chaos creates best memories",
+            "Mischief makes life interesting",
+            "Chaos with good intentions",
+            "Fun times multiply exponentially",
+            "Chaos theory in action",
+            "Mischief planned and executed",
+            "Chaos brings out creativity",
+            "Fun vibes spreading everywhere",
+            "Chaos with positive energy"
+        ],
+        "motivational_success": [
+            "Grinding until dreams reality",
+            "Success tastes so sweet",
+            "Hustle hard dream bigger",
+            "Grinding never stops here",
+            "Success is only option",
+            "Hustle mode permanently on",
+            "Grinding toward bigger goals",
+            "Success follows consistent effort",
+            "Hustle beats talent consistently",
+            "Grinding creates diamond pressure",
+            "Success is earned daily",
+            "Hustle culture lifestyle choice",
+            "Grinding through every obstacle",
+            "Success mindset never rests",
+            "Hustle harder than yesterday",
+            "Grinding until goals achieved",
+            "Success requires relentless pursuit",
+            "Hustle creates unstoppable momentum",
+            "Grinding builds character strength",
+            "Success is inevitable outcome"
+        ],
+        "chill_relaxed": [
+            "Zen mode activated today",
+            "Peace love and positivity",
+            "Chill vibes only please",
+            "Zen energy flowing freely",
+            "Peace found in simplicity",
+            "Chill mode permanently on",
+            "Zen master in training",
+            "Peace within creates harmony",
+            "Chill vibes multiply naturally",
+            "Zen state of mind",
+            "Peace radiates from within",
+            "Chill energy infectious spreading",
+            "Zen garden of thoughts",
+            "Peace is my superpower",
+            "Chill vibes heal everything",
+            "Zen moments create clarity",
+            "Peace found in present",
+            "Chill mode healing process",
+            "Zen energy restores balance",
+            "Peace multiplies when shared"
+        ],
+        "trendy_gen_z": [
+            "No cap this slaps",
+            "Periodt end of discussion",
+            "This hits different though",
+            "Absolutely serving looks today",
+            "This is it chief",
+            "We understood the assignment",
+            "This absolutely sends me",
+            "Living rent free here",
+            "This is the vibe",
+            "Absolutely no skips today",
+            "This is so valid",
+            "We really did that",
+            "This is the moment",
+            "Absolutely understood assignment perfectly",
+            "This hits every time",
+            "We said what said",
+            "This is peak content",
+            "Absolutely no notes needed",
+            "This is the energy",
+            "We chose violence today"
+        ],
+        "night_out": [
+            "Night out squad assembled",
+            "Party mode fully activated",
+            "Night vibes hit different",
+            "Party till sunrise today",
+            "Night out energy unmatched",
+            "Party like there tomorrow",
+            "Night out goals achieved",
+            "Party mode never ends",
+            "Night vibes absolutely immaculate",
+            "Party energy through roof",
+            "Night out memories made",
+            "Party until legs hurt",
+            "Night out squad goals",
+            "Party mode is lifestyle",
+            "Night vibes multiply exponentially",
+            "Party energy infectious spreading",
+            "Night out therapy session",
+            "Party mode permanently activated",
+            "Night vibes heal everything",
+            "Party like world watching"
+        ],
+        "self_love": [
+            "Self love Sunday activated",
+            "Main character energy strong",
+            "Self care Sunday vibes",
+            "Self love looks good",
+            "Main character moment captured",
+            "Self care never optional",
+            "Self love is superpower",
+            "Main character energy activated",
+            "Self care Sunday ritual",
+            "Self love journey continues",
+            "Main character vibes only",
+            "Self care is healthcare",
+            "Self love wins always",
+            "Main character development complete",
+            "Self care Sunday success",
+            "Self love radiates outward",
+            "Main character energy infectious",
+            "Self care heals everything",
+            "Self love multiplies daily",
+            "Main character story continues"
+        ]
     }
     
-    # Find keyword-specific captions
-    description_lower = description.lower()
-    keyword_specific_captions = []
-    
-    for keyword, captions in keyword_captions.items():
-        if keyword in description_lower:
-            keyword_specific_captions.extend(captions)
+    # Flatten all cool captions into one big list
+    all_cool_captions = []
+    for category, captions in cool_captions_db.items():
+        all_cool_captions.extend(captions)
     
     # Generate AI captions if model is available
     ai_captions = []
     if generator is not None:
         attempts = 0
-        while len(ai_captions) < 5 and attempts < 15:  # Try to get 5 AI captions
+        while len(ai_captions) < 2 and attempts < 10:  # Try to get at least 2 good AI captions
             caption = generate_short_caption(description)
             if caption and caption not in ai_captions and is_valid_caption(caption):
                 ai_captions.append(caption)
             attempts += 1
     
-    # Combine keyword-specific and AI captions
-    # Prioritize keyword-specific captions if found
-    if keyword_specific_captions:
-        all_captions = keyword_specific_captions + ai_captions
-    else:
-        all_captions = ai_captions
+    # Combine AI and predefined cool captions
+    combined_captions = ai_captions + all_cool_captions
     
     # Remove duplicates while preserving order
     seen = set()
     unique_captions = []
-    for caption in all_captions:
+    for caption in combined_captions:
         if caption not in seen:
             seen.add(caption)
             unique_captions.append(caption)
     
-    # If we don't have enough captions, create basic fallbacks
-    if len(unique_captions) < 5:
-        basic_fallbacks = [
-            "Picture perfect moment",
-            "Living my best life",
-            "Good vibes only",
-            "Making memories today",
-            "Life is beautiful"
-        ]
-        
-        for fallback in basic_fallbacks:
-            if fallback not in unique_captions:
-                unique_captions.append(fallback)
-                if len(unique_captions) >= 5:
-                    break
-    
-    # Select 5 captions
-    selected_captions = unique_captions[:5]
+    # Select 5 random captions
+    selected_captions = random.sample(unique_captions, min(5, len(unique_captions)))
     
     # Ensure each caption is max 5 words and properly formatted
     final_captions = []
@@ -283,6 +425,14 @@ def generate_instagram_captions(description):
         if len(words) >= 2:
             final_captions.append(" ".join(words))
     
+    # Fill with defaults if needed
+    while len(final_captions) < 5:
+        fallback = random.choice(all_cool_captions)
+        words = fallback.split()[:5]
+        final_caption = " ".join(words)
+        if final_caption not in final_captions:
+            final_captions.append(final_caption)
+    
     # Format as simple numbered list
     result = ""
     emojis = ["🎯", "⚡", "🌟", "🔥", "💫"]
@@ -292,7 +442,7 @@ def generate_instagram_captions(description):
     
     return result.strip()
 
-# Simplified Instagram-inspired theme with custom CSS
+# Enhanced Instagram-inspired theme with custom CSS
 instagram_css = """
 /* Instagram color scheme and modern design */
 :root {
@@ -314,19 +464,15 @@ instagram_css = """
     font-family: 'Instagram Sans', -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif !important;
 }
 
-/* Header styling */
+/* Header styling - Modified to be regular black text */
 .main-header {
     text-align: center;
-    background: linear-gradient(45deg, var(--ig-primary), var(--ig-secondary));
-    -webkit-background-clip: text;
-    -webkit-text-fill-color: transparent;
-    background-clip: text;
-    color: transparent;
+    color: #000000 !important;
     font-size: 3em;
     font-weight: 400;
     margin-bottom: 0.5em;
     padding: 1rem 0;
-    background-color: rgba(255, 255, 255, 0.9);
+    background: rgba(255, 255, 255, 0.9);
     border-radius: 15px;
     margin: 1rem;
     box-shadow: 0 4px 20px rgba(0,0,0,0.1);
@@ -407,6 +553,29 @@ instagram_css = """
     padding: 1rem !important;
 }
 
+/* Examples styling */
+.gr-examples {
+    background: rgba(255, 255, 255, 0.8) !important;
+    border-radius: 15px !important;
+    padding: 1rem !important;
+    margin-top: 1rem !important;
+}
+
+.gr-examples .gr-button {
+    background: linear-gradient(45deg, var(--ig-tertiary), var(--ig-quaternary)) !important;
+    border: none !important;
+    border-radius: 10px !important;
+    color: white !important;
+    margin: 0.25rem !important;
+    transition: all 0.3s ease !important;
+    font-weight: 600 !important;
+}
+
+.gr-examples .gr-button:hover {
+    transform: translateY(-2px) !important;
+    box-shadow: 0 4px 12px rgba(0,0,0,0.2) !important;
+}
+
 /* Tips section */
 .tips-section {
     background: rgba(255, 255, 255, 0.9) !important;
@@ -449,6 +618,16 @@ instagram_css = """
         padding: 12px 25px !important;
     }
 }
+
+/* Loading animation */
+@keyframes pulse {
+    0%, 100% { opacity: 1; }
+    50% { opacity: 0.5; }
+}
+
+.loading {
+    animation: pulse 2s infinite;
+}
 """
 
 # Create the Gradio interface
@@ -511,10 +690,10 @@ with gr.Blocks(
         <div class="tips-section">
             <p>💡 <strong>Pro Tips for Better Captions:</strong></p>
             <p>🎯 Each caption is maximum 5 words for maximum impact</p>
-            <p>⚡ Be specific in your description for better AI-generated captions</p>
+            <p>⚡ Get fresh trendy captions every time you generate</p>
             <p>📱 Copy and paste directly to Instagram, TikTok, or any social platform</p>
-            <p>🔄 Click generate again for fresh new options</p>
-            <p>✨ Mix AI-generated captions with trending phrases</p>
+            <p>🔄 Click generate again for completely new cool options</p>
+            <p>✨ Mix of AI-generated and trending cool captions</p>
         </div>
         """
     )