Update app.py

app.py

@@ -163,6 +163,67 @@ def generate_instagram_captions(description):
     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": ["Hot and about", "Sun kissed skin", "Salt water heals everything", "Beach vibes only", "Sandy toes salty kisses"],
+        "sea": ["Hot and about", "Ocean child wild heart", "Sea you later", "Mermaid vibes activated", "Waves for days"],
+        "ocean": ["Hot and about", "Deep blue dreams", "Ocean therapy session", "Salty but sweet", "Endless summer vibes"],
+        "sand": ["Hot and about", "Sandy hair dont care", "Life is better", "Footprints in sand", "Sun sand and"],
+        "sun": ["Hot and about", "Chasing golden hour", "Sunshine on my mind", "Solar powered and happy", "Vitamin D loading"],
+        "sunset": ["Hot and about", "Golden hour magic", "Painted sky perfection", "Sunset state of mind", "Colors of happiness"],
+        "swimming": ["Hot and about", "Pool day vibes", "Making waves today", "Dive into happiness", "Swim your worries away"],
+        "surfing": ["Hot and about", "Riding the waves", "Salt life chosen", "Surf vibes activated", "Ocean is calling"],
+        
+        # Food related
+        "food": ["Good food good mood", "Foodie life chosen", "Taste buds happy", "Eating my feelings", "Food coma incoming"],
+        "coffee": ["But first coffee", "Coffee and good vibes", "Espresso yourself freely", "Caffeine and confidence", "Morning fuel activated"],
+        "pizza": ["Pizza my heart", "Slice of heaven", "Cheesy and happy", "Pizza party vibes", "Life needs more pizza"],
+        "brunch": ["Brunch squad assembled", "Bottomless mimosa mood", "Weekend brunch vibes", "Eggs benedict bliss", "Brunch so hard"],
+        
+        # Fitness/Health
+        "gym": ["Sweat now shine later", "Strong not skinny", "Gym life chosen", "Gains and glory", "Fitness is lifestyle"],
+        "workout": ["No pain no gain", "Sweat is fat crying", "Beast mode activated", "Stronger than yesterday", "Fitness addiction real"],
+        "yoga": ["Peace love and yoga", "Zen mode activated", "Namaste all day", "Breathe deeply live fully", "Inner peace found"],
+        "running": ["Run like you stole", "Miles and smiles", "Runner high achieved", "Pavement pounding therapy", "One step closer"],
+        
+        # Travel/Adventure
+        "travel": ["Wanderlust and city dust", "Adventure awaits us", "Passport ready heart", "Collect moments not things", "Travel bug activated"],
+        "adventure": ["Adventure is out there", "Wild hearts cant", "Explore more worry less", "Adventure time activated", "Life is adventure"],
+        "hiking": ["Take only pictures", "Mountain therapy session", "Hike more worry less", "Trail blazing today", "Nature is medicine"],
+        "camping": ["Into the wild", "Campfire and chill", "Nature is home", "Unplugged and recharged", "Wild and free"],
+        
+        # Party/Celebration
+        "party": ["Party like its", "Good times roll", "Dance floor ready", "Party mode activated", "Celebration time come"],
+        "birthday": ["Another year wiser", "Birthday vibes activated", "Cake and celebration", "Getting better with age", "Birthday squad assembled"],
+        "celebration": ["Cheers to good times", "Celebration mode activated", "Party till sunrise", "Good vibes only", "Making memories tonight"],
+        
+        # Friends/Social
+        "friends": ["Squad goals right here", "Friends like these matter", "Good company great memories", "Squad deeper than ocean", "Friendship goals achieved"],
+        "squad": ["Squad up for success", "Squad goals achieved", "Squad tight bond stronger", "Team work makes", "Squad deeper than"],
+        "bestie": ["Partner in crime", "Bestie for life", "Ride or die", "Best friend goals", "Friendship never ends"],
+        
+        # Fashion/Style
+        "outfit": ["Outfit on point", "Style game strong", "Fashion forward thinking", "Dressed to impress", "Style is everything"],
+        "fashion": ["Fashion is art", "Style over everything", "Runway ready always", "Fashion game strong", "Dressed for success"],
+        "style": ["Style is eternal", "Personal style activated", "Fashion forward always", "Style game unmatched", "Dressed to kill"],
+        
+        # Love/Romance
+        "love": ["Love is everything", "Heart full of", "Love wins always", "Romance is alive", "Love story continues"],
+        "couple": ["Better together always", "Love story goals", "Partner in everything", "Two hearts one", "Couple goals achieved"],
+        "date": ["Date night vibes", "Romance is alive", "Perfect evening together", "Love is beautiful", "Hearts aligned perfectly"],
+        
+        # Nature/Outdoors
+        "nature": ["Nature is medicine", "Wild and free", "Back to nature", "Earth therapy session", "Natural beauty everywhere"],
+        "flowers": ["Bloom where planted", "Fresh as daisy", "Flower power activated", "Natural beauty blooms", "Petals and positivity"],
+        "garden": ["Garden therapy session", "Growing something beautiful", "Nature nurtures soul", "Bloom and grow", "Earth magic happening"],
+        
+        # Work/Success
+        "work": ["Work hard play harder", "Hustle and heart", "Success is earned", "Grind never stops", "Work mode activated"],
+        "success": ["Success tastes sweet", "Hard work pays", "Dream big achieve", "Success is journey", "Winning mindset activated"],
+        "business": ["Business mode activated", "Entrepreneur life chosen", "Success is planned", "Building empire daily", "Boss moves only"]
+    }
+    
     # Enhanced cool captions database with complete, meaningful phrases
     cool_captions = [
         "Squad goals right here",
@@ -217,6 +278,14 @@ def generate_instagram_captions(description):
         "Namaste all day long"
     ]
     
+    # 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)
+    
     # Generate AI captions if model is available
     ai_captions = []
     if generator is not None:
@@ -227,8 +296,12 @@ def generate_instagram_captions(description):
                 ai_captions.append(caption)
             attempts += 1
     
-    # Combine AI and predefined captions
-    all_captions = ai_captions + cool_captions
+    # Combine keyword-specific, AI and predefined captions
+    # Prioritize keyword-specific captions if found
+    if keyword_specific_captions:
+        all_captions = keyword_specific_captions + ai_captions + cool_captions
+    else:
+        all_captions = ai_captions + cool_captions
     
     # Remove duplicates while preserving order
     seen = set()
@@ -256,7 +329,7 @@ def generate_instagram_captions(description):
         if final_caption not in final_captions:
             final_captions.append(final_caption)
     
-    # Format as numbered list with emojis
+    # Format as simple numbered list (removed "Your Instagram Captions" text)
     result = ""
     emojis = ["🎯", "⚡", "🌟", "🔥", "💫"]
     
@@ -287,18 +360,18 @@ 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), var(--ig-tertiary), var(--ig-quaternary));
-    -webkit-background-clip: text;
-    -webkit-text-fill-color: transparent;
-    background-clip: text;
+    color: #000000 !important;
     font-size: 3em;
-    font-weight: 900;
+    font-weight: 400;
     margin-bottom: 0.5em;
-    text-shadow: 0 2px 4px rgba(0,0,0,0.1);
     padding: 1rem 0;
+    background: rgba(255, 255, 255, 0.9);
+    border-radius: 15px;
+    margin: 1rem;
+    box-shadow: 0 4px 20px rgba(0,0,0,0.1);
 }
 
 .subtitle {
@@ -464,7 +537,8 @@ with gr.Blocks(
     title="InstaCap - AI Caption Generator"
 ) as demo:
     
-    gr.HTML('<h1 class="main-header white-title"><strong>📸 InstaCap – AI Caption Generator</strong></h1>')
+    # Move title to the front and make it regular black
+    gr.HTML('<h1 class="main-header">InstaCap – AI Caption Generator</h1>')
     gr.HTML('<p class="subtitle">Create short, catchy Instagram captions in seconds! Perfect for your social media posts ✨</p>')
     
     with gr.Row():
@@ -493,25 +567,7 @@ with gr.Blocks(
                 elem_classes="output-box"
             )
     
-    # Add examples with diverse Instagram-style descriptions - REMOVED elem_classes
-    gr.Examples(
-        examples=[
-            ["group photo with best friends at sunset"],
-            ["coffee date with my bestie"],
-            ["hiking adventure with the squad"],
-            ["beach day with good vibes"],
-            ["cozy movie night with favorites"],
-            ["birthday celebration with everyone"],
-            ["weekend brunch with the crew"],
-            ["road trip memories being made"],
-            ["concert night with amazing energy"],
-            ["graduation day with family pride"]
-        ],
-        inputs=description_input,
-        outputs=caption_output,
-        fn=generate_instagram_captions,
-        cache_examples=False
-    )
+    # Examples are kept in code but not shown in UI
     
     # Connect the button to the function
     generate_btn.click(