Update app.py

app.py

@@ -24,271 +24,349 @@ except Exception as e:
         pad_token_id=50256
     )
 
-def create_short_caption(text, max_words=5):
-    """Ensure caption is max 5 words"""
-    words = text.split()
-    if len(words) <= max_words:
-        return text
-    return ' '.join(words[:max_words])
+def generate_short_caption(description):
+    """Generate a short, catchy caption (max 5 words)"""
+    if not description.strip():
+        return ""
+    
+    # Set random seed for variety
+    set_seed(random.randint(1, 10000))
+    
+    # Short prompt templates for 5-word captions
+    prompts = [
+        f"Photo: {description}\nShort caption: Squad",
+        f"Photo: {description}\nCool caption: Good",
+        f"Photo: {description}\nCatchy caption: Living",
+        f"Photo: {description}\nFun caption: Making",
+        f"Photo: {description}\nTrendy caption: Caught"
+    ]
+    
+    prompt = random.choice(prompts)
+    
+    try:
+        generated = generator(
+            prompt,
+            max_length=len(prompt.split()) + 8,  # Very short
+            num_return_sequences=1,
+            temperature=0.9,
+            do_sample=True,
+            top_p=0.95,
+            repetition_penalty=1.2,
+            pad_token_id=50256
+        )
+        
+        full_text = generated[0]['generated_text']
+        caption = full_text.replace(prompt, "").strip()
+        
+        # Clean and limit to 5 words
+        if caption:
+            words = caption.split()[:5]
+            if len(words) >= 2:
+                return " ".join(words)
+    
+    except Exception as e:
+        print(f"Generation error: {e}")
+    
+    return ""
 
-def generate_instagram_caption(description):
+def generate_instagram_captions(description):
     """Generate 5 short, catchy Instagram captions (max 5 words each)"""
     if not description.strip():
         return "Please enter a description of your photo!"
     
-    # Set a random seed for variety
-    set_seed(random.randint(1, 10000))
-    
-    # Cool and catchy caption templates (will be trimmed to 5 words)
-    cool_templates = [
+    # Cool short captions for different vibes
+    cool_captions = [
         "Squad goals right here",
+        "Good vibes only today",
         "Living our best life",
-        "Good vibes only always",
         "Making memories with favorites",
-        "This crew hits different",
-        "Unstoppable energy with besties",
-        "Main character energy activated",
-        "Plot twist we're legendary",
-        "Caught being absolutely iconic",
-        "Serving looks and laughs",
-        "That golden hour magic",
-        "Friendship level absolutely maximum",
-        "Current mood pure happiness",
-        "Creating chaos looking cute",
-        "Weekend warriors in action",
-        "Living proof dreams work",
-        "Grateful for these humans",
-        "When life gets good",
-        "Positive vibes attract everything",
-        "Making ordinary moments special"
+        "Caught in the moment",
+        "Friends like these though",
+        "Energy levels unmatched today",
+        "Vibes on point always",
+        "Good times roll forever",
+        "Squad up for success",
+        "Living for these moments",
+        "Making magic happen daily",
+        "Catching feelings and sunsets",
+        "Good company great memories",
+        "Squad deeper than ocean",
+        "Vibes immaculate as always",
+        "Energy infectious spread it",
+        "Good times never ending",
+        "Making waves together always",
+        "Caught smiling too much",
+        "Living life in color",
+        "Squad goals achieved today",
+        "Good vibes attract everything",
+        "Making moments matter most",
+        "Catching dreams with crew",
+        "Living loud loving harder",
+        "Squad tight bond stronger",
+        "Good energy good people",
+        "Making memories worth keeping",
+        "Caught being absolutely legendary"
     ]
     
-    # Generate captions based on description context
-    captions = []
-    
-    # Try to generate some with AI
+    # Generate some AI captions
+    ai_captions = []
     for i in range(3):
-        try:
-            prompt = f"Cool Instagram caption for {description}:"
-            generated = generator(
-                prompt,
-                max_length=len(prompt.split()) + 8,
-                num_return_sequences=1,
-                temperature=0.9,
-                do_sample=True,
-                top_p=0.95,
-                repetition_penalty=1.2,
-                pad_token_id=50256
-            )
-            
-            full_text = generated[0]['generated_text']
-            caption_text = full_text.replace(prompt, "").strip()
-            
-            # Clean and shorten
-            caption_text = caption_text.strip('"\'').strip()
-            if caption_text and len(caption_text) > 3:
-                short_caption = create_short_caption(caption_text, 5)
-                if short_caption not in captions:
-                    captions.append(short_caption)
-                    
-        except Exception as e:
-            print(f"Generation error: {e}")
-            continue
+        caption = generate_short_caption(description)
+        if caption:
+            ai_captions.append(caption)
     
-    # Add cool predefined captions
-    random.shuffle(cool_templates)
-    for template in cool_templates:
-        if len(captions) >= 5:
-            break
-        short_caption = create_short_caption(template, 5)
-        if short_caption not in captions:
-            captions.append(short_caption)
+    # Combine AI and predefined captions
+    all_captions = ai_captions + cool_captions
     
-    # Context-specific captions based on description
-    context_captions = []
-    desc_lower = description.lower()
+    # Select 5 random captions
+    selected_captions = random.sample(all_captions, min(5, len(all_captions)))
     
-    if any(word in desc_lower for word in ['friend', 'group', 'squad', 'crew', 'team', 'gang']):
-        context_captions = [
-            "Squad goals achieved today",
-            "Friends who slay together",
-            "Main character energy activated",
-            "Chaos creators looking cute",
-            "Weekend warriors in action"
-        ]
-    elif any(word in desc_lower for word in ['sunset', 'beach', 'ocean', 'water']):
-        context_captions = [
-            "Chasing sunsets and dreams",
-            "Salt air good vibes",
-            "Beach therapy session complete",
-            "Ocean child energy activated",
-            "Waves and good times"
-        ]
-    elif any(word in desc_lower for word in ['coffee', 'morning', 'work', 'laptop']):
-        context_captions = [
-            "Morning hustle mode on",
-            "Coffee powered productivity session",
-            "Grinding with good vibes",
-            "Main character work energy",
-            "Caffeine and determination daily"
-        ]
-    elif any(word in desc_lower for word in ['food', 'dinner', 'eating', 'restaurant']):
-        context_captions = [
-            "Good food better company",
-            "Eating our way happy",
-            "Dinner dates hit different",
-            "Food coma incoming soon",
-            "Taste buds officially satisfied"
-        ]
-    elif any(word in desc_lower for word in ['adventure', 'hiking', 'mountain', 'nature']):
-        context_captions = [
-            "Adventure mode fully activated",
-            "Mountain therapy session complete",
-            "Nature calls we answer",
-            "Hiking boots happy feet",
-            "Views and good vibes"
-        ]
+    # Ensure each caption is max 5 words
+    final_captions = []
+    for caption in selected_captions:
+        words = caption.split()[:5]
+        if len(words) >= 2:
+            final_captions.append(" ".join(words))
     
-    # Add context-specific captions
-    for context_cap in context_captions:
-        if len(captions) >= 5:
-            break
-        short_caption = create_short_caption(context_cap, 5)
-        if short_caption not in captions:
-            captions.append(short_caption)
-    
-    # Ensure we have exactly 5 captions
-    universal_captions = [
-        "Living our best life",
-        "Good vibes only always",
-        "Making memories count today",
-        "Plot twist we're iconic",
-        "Current mood pure happiness",
-        "Grateful for these moments",
-        "Creating magic happening now",
-        "Weekend warrior energy activated",
-        "Positive vibes attract everything",
-        "This crew hits different"
-    ]
-    
-    for universal_cap in universal_captions:
-        if len(captions) >= 5:
-            break
-        short_caption = create_short_caption(universal_cap, 5)
-        if short_caption not in captions:
-            captions.append(short_caption)
-    
-    # Return exactly 5 captions
-    final_captions = captions[:5]
+    # Fill with defaults if needed
+    while len(final_captions) < 5:
+        fallback = random.choice(cool_captions)
+        words = fallback.split()[:5]
+        final_caption = " ".join(words)
+        if final_caption not in final_captions:
+            final_captions.append(final_caption)
     
     # Format as numbered list
     result = ""
-    for i, caption in enumerate(final_captions, 1):
+    for i, caption in enumerate(final_captions[:5], 1):
         result += f"{i}. {caption}\n"
     
     return result.strip()
 
-# Create the Gradio interface
-with gr.Blocks(
-    theme=gr.themes.Soft(),
-    css="""
+# Instagram-inspired theme with custom CSS
+instagram_css = """
+/* Instagram color scheme */
+:root {
+    --ig-primary: #E4405F;
+    --ig-secondary: #833AB4;
+    --ig-tertiary: #F77737;
+    --ig-quaternary: #FCAF45;
+    --ig-blue: #405DE6;
+    --ig-bg: #FAFAFA;
+    --ig-text: #262626;
+    --ig-border: #DBDBDB;
+}
+
+/* Main container styling */
+.gradio-container {
+    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
+    min-height: 100vh;
+}
+
+/* Header styling */
+.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;
+    font-size: 3em;
+    font-weight: 900;
+    margin-bottom: 0.5em;
+    text-shadow: 0 2px 4px rgba(0,0,0,0.1);
+}
+
+.subtitle {
+    text-align: center;
+    color: white;
+    font-size: 1.2em;
+    margin-bottom: 2em;
+    text-shadow: 0 1px 2px rgba(0,0,0,0.3);
+}
+
+/* Card styling */
+.card {
+    background: rgba(255, 255, 255, 0.95) !important;
+    border-radius: 20px !important;
+    box-shadow: 0 8px 32px rgba(0,0,0,0.1) !important;
+    backdrop-filter: blur(10px) !important;
+    border: 1px solid rgba(255,255,255,0.2) !important;
+    padding: 2rem !important;
+    margin: 1rem !important;
+}
+
+/* Input styling */
+.gr-textbox {
+    border-radius: 15px !important;
+    border: 2px solid var(--ig-border) !important;
+    transition: all 0.3s ease !important;
+}
+
+.gr-textbox:focus {
+    border-color: var(--ig-primary) !important;
+    box-shadow: 0 0 0 3px rgba(228, 64, 95, 0.1) !important;
+}
+
+/* Button styling */
+.generate-btn {
+    background: linear-gradient(45deg, var(--ig-primary), var(--ig-secondary)) !important;
+    border: none !important;
+    color: white !important;
+    font-weight: bold !important;
+    font-size: 1.2em !important;
+    padding: 15px 30px !important;
+    border-radius: 25px !important;
+    transition: all 0.3s ease !important;
+    box-shadow: 0 4px 15px rgba(228, 64, 95, 0.3) !important;
+}
+
+.generate-btn:hover {
+    transform: translateY(-3px) !important;
+    box-shadow: 0 8px 25px rgba(228, 64, 95, 0.4) !important;
+    background: linear-gradient(45deg, var(--ig-secondary), var(--ig-primary)) !important;
+}
+
+/* Output styling */
+.output-box {
+    background: rgba(255, 255, 255, 0.9) !important;
+    border-radius: 15px !important;
+    border: 2px solid var(--ig-border) !important;
+    font-family: 'Instagram Sans', -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif !important;
+    font-size: 1.1em !important;
+    line-height: 1.6 !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;
+}
+
+.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;
+    border-radius: 15px !important;
+    padding: 1.5rem !important;
+    margin-top: 2rem !important;
+    text-align: center !important;
+    box-shadow: 0 4px 20px rgba(0,0,0,0.1) !important;
+}
+
+.tips-section p {
+    color: var(--ig-text) !important;
+    font-size: 1em !important;
+    margin: 0.5rem 0 !important;
+}
+
+/* Mobile responsive */
+@media (max-width: 768px) {
     .main-header {
-        text-align: center;
-        background: linear-gradient(45deg, #ff6b6b, #4ecdc4, #45b7d1);
-        -webkit-background-clip: text;
-        -webkit-text-fill-color: transparent;
-        font-size: 2.5em;
-        font-weight: bold;
-        margin-bottom: 0.5em;
-    }
-    .subtitle {
-        text-align: center;
-        color: #666;
-        font-size: 1.1em;
-        margin-bottom: 2em;
+        font-size: 2.2em !important;
     }
-    .generate-btn {
-        background: linear-gradient(45deg, #ff6b6b, #4ecdc4) !important;
-        border: none !important;
-        color: white !important;
-        font-weight: bold !important;
-        font-size: 1.1em !important;
-        padding: 12px 24px !important;
-        border-radius: 25px !important;
-        transition: all 0.3s ease !important;
-    }
-    .generate-btn:hover {
-        transform: translateY(-2px) !important;
-        box-shadow: 0 5px 15px rgba(0,0,0,0.2) !important;
+    
+    .card {
+        margin: 0.5rem !important;
+        padding: 1.5rem !important;
     }
-    """
+}
+"""
+
+# Create the Gradio interface
+with gr.Blocks(
+    theme=gr.themes.Soft(
+        primary_hue="red",
+        secondary_hue="purple",
+        neutral_hue="gray"
+    ),
+    css=instagram_css
 ) as demo:
     
-    gr.HTML('<h1 class="main-header">📸 Instacap – AI Instagram Caption Generator</h1>')
-    gr.HTML('<p class="subtitle">Transform your photo descriptions into short, catchy Instagram captions!</p>')
+    gr.HTML('<h1 class="main-header">📸 InstaCap – AI Caption Generator</h1>')
+    gr.HTML('<p class="subtitle">Create short, catchy Instagram captions in seconds! ✨</p>')
     
     with gr.Row():
-        with gr.Column(scale=1):
+        with gr.Column(scale=1, elem_classes="card"):
             description_input = gr.Textbox(
-                label="Describe your photo",
-                placeholder="E.g., group photo with friends, sunset beach picnic, coffee shop vibes",
+                label="📝 Describe your photo",
+                placeholder="E.g., group photo with friends, sunset vibes, coffee date...",
                 lines=3,
-                max_lines=5
+                max_lines=5,
+                elem_classes="gr-textbox"
             )
             
             generate_btn = gr.Button(
-                "Generate Caption ✨",
+                "✨ Generate Captions",
                 variant="primary",
                 elem_classes="generate-btn"
             )
         
-        with gr.Column(scale=1):
+        with gr.Column(scale=1, elem_classes="card"):
             caption_output = gr.Textbox(
-                label="Your 5 Short & Catchy Captions (Max 5 words each)",
+                label="🎯 Your 5 Short Captions (Max 5 Words Each)",
                 lines=8,
                 max_lines=10,
                 interactive=False,
-                show_copy_button=True
+                show_copy_button=True,
+                elem_classes="output-box"
             )
     
-    # Add some example inputs
+    # Add examples with Instagram-style descriptions
     gr.Examples(
         examples=[
-            ["group photo with friends laughing"],
-            ["sunset beach picnic, cozy vibe"],
-            ["coffee shop laptop work, productive morning"],
-            ["hiking mountain trail, adventure time"],
-            ["friends dinner party, good food"],
-            ["morning yoga session, peaceful zen"],
-            ["city skyline night view, urban lights"],
-            ["weekend brunch with squad, happy vibes"]
+            ["group photo with best friends"],
+            ["sunset beach vibes together"],
+            ["coffee date good times"],
+            ["squad night out fun"],
+            ["adventure hiking with crew"],
+            ["cozy indoor friend hangout"],
+            ["party celebration with everyone"],
+            ["weekend brunch with favorites"]
         ],
         inputs=description_input,
         outputs=caption_output,
-        fn=generate_instagram_caption,
-        cache_examples=False
+        fn=generate_instagram_captions,
+        cache_examples=False,
+        elem_classes="gr-examples"
     )
     
     # Connect the button to the function
     generate_btn.click(
-        fn=generate_instagram_caption,
+        fn=generate_instagram_captions,
         inputs=description_input,
         outputs=caption_output
     )
     
     # Also trigger on Enter key
     description_input.submit(
-        fn=generate_instagram_caption,
+        fn=generate_instagram_captions,
         inputs=description_input,
         outputs=caption_output
     )
     
     gr.HTML(
         """
-        <div style="text-align: center; margin-top: 2em; color: #888; font-size: 0.9em;">
-            <p>💡 Tip: Be specific for better context-matched captions!</p>
-            <p>🎨 Get 5 short, catchy captions (max 5 words each)</p>
-            <p>📋 Perfect for group photos, adventures, and good vibes!</p>
+        <div class="tips-section">
+            <p>💡 <strong>Pro Tips:</strong></p>
+            <p>🎯 Each caption is max 5 words for maximum impact</p>
+            <p>⚡ Perfect for group photos, adventures, and good vibes</p>
+            <p>📱 Copy and paste directly to Instagram</p>
+            <p>🔄 Generate again for new options</p>
         </div>
         """
     )