Update app.py

app.py

@@ -26,28 +26,221 @@ except Exception as e:
 
 def clean_caption(text):
     """Clean and format the generated caption"""
-    # Remove the prompt part
+    # Remove the prompt part and get first line
     lines = text.split('\n')
-    caption_lines = []
-    
     for line in lines:
         line = line.strip()
         if line and not line.startswith('Photo:') and not line.startswith('Caption:'):
-            caption_lines.append(line)
-    
-    if caption_lines:
-        caption = caption_lines[0]
-        # Remove extra quotes or incomplete sentences
-        caption = re.sub(r'^["\']|["\']$', '', caption)
-        # Ensure it doesn't end mid-sentence
-        if caption and caption[-1] not in '.!?':
-            # Find the last complete sentence
-            sentences = re.split(r'[.!?]', caption)
-            if len(sentences) > 1:
-                caption = '.'.join(sentences[:-1]) + '.'
-            else:
-                caption = caption + '.'
-        return caption
+            # Remove extra quotes
+            caption = re.sub(r'^["\']|["\']
+
+def generate_instagram_caption(description):
+    """Generate 5 clever Instagram-style captions based on the photo description"""
+    if not description.strip():
+        return "Please enter a description of your photo!"
+    
+    # Set a random seed for variety
+    set_seed(random.randint(1, 10000))
+    
+    # Create multiple clever Instagram-style prompts
+    prompt_templates = [
+        f"Photo: {description}\nWitty Instagram caption: Just",
+        f"Photo: {description}\nClever Instagram caption: When",
+        f"Photo: {description}\nFunny Instagram caption: Living",
+        f"Photo: {description}\nSassy Instagram caption: Feeling",
+        f"Photo: {description}\nCute Instagram caption: Making",
+        f"Photo: {description}\nInspiring Instagram caption: Chasing",
+        f"Photo: {description}\nCool Instagram caption: Caught",
+        f"Photo: {description}\nRelatable Instagram caption: That",
+        f"Photo: {description}\nTrendy Instagram caption: Current",
+        f"Photo: {description}\nPlayful Instagram caption: Plot"
+    ]
+    
+    captions = []
+    
+    # Generate 5 different captions
+    for i in range(5):
+        # Use different prompts for variety
+        prompt = random.choice(prompt_templates)
+        
+        try:
+            generated = generator(
+                prompt,
+                max_length=len(prompt.split()) + 15,  # Keep it 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 = clean_caption(full_text.replace(prompt, "").strip())
+            
+            if caption and len(caption) > 10:  # Make sure it's substantial
+                captions.append(caption)
+            
+        except Exception as e:
+            print(f"Generation error: {e}")
+            continue
+    
+    # Fill with clever fallback captions if we don't have enough
+    clever_fallbacks = [
+        f"✨ {description} kind of day ✨",
+        f"Plot twist: {description} was exactly what I needed",
+        f"Current mood: {description} energy",
+        f"Just me living my {description} dreams",
+        f"When life gives you {description}, make memories",
+        f"Serving {description} realness",
+        f"That {description} life chose me",
+        f"POV: You're living for {description} moments",
+        f"Main character energy: {description} edition",
+        f"No caption needed, just {description} vibes",
+        f"Caught in 4K living my {description} fantasy",
+        f"This is what {description} happiness looks like",
+        f"Manifesting more {description} energy",
+        f"When the {description} hits different",
+        f"Today's agenda: {description} and good vibes",
+        f"Reminder: {description} days are the best days",
+        f"Living proof that {description} is everything",
+        f"My {description} era is here to stay",
+        f"Just another {description} adventure",
+        f"Grateful for {description} moments like these"
+    ]
+    
+    # Add fallbacks if we need more captions
+    while len(captions) < 5:
+        fallback = random.choice(clever_fallbacks)
+        if fallback not in captions:
+            captions.append(fallback)
+    
+    # Return top 5 captions
+    final_captions = captions[:5]
+    
+    # Format as numbered list
+    result = ""
+    for i, caption in enumerate(final_captions, 1):
+        result += f"{i}. {caption}\n"
+    
+    return result.strip()
+
+# Create the Gradio interface
+with gr.Blocks(
+    theme=gr.themes.Soft(),
+    css="""
+    .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;
+    }
+    .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;
+    }
+    """
+) as demo:
+    
+    gr.HTML('<h1 class="main-header">📸 Instacap – AI Instagram Caption Generator</h1>')
+    gr.HTML('<p class="subtitle">Transform your photo descriptions into engaging Instagram captions!</p>')
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            description_input = gr.Textbox(
+                label="Describe your photo",
+                placeholder="E.g., sunset beach picnic, cozy vibe",
+                lines=3,
+                max_lines=5
+            )
+            
+            generate_btn = gr.Button(
+                "Generate Caption ✨",
+                variant="primary",
+                elem_classes="generate-btn"
+            )
+        
+        with gr.Column(scale=1):
+            caption_output = gr.Textbox(
+                label="Your 5 Instagram Caption Options",
+                lines=8,
+                max_lines=10,
+                interactive=False,
+                show_copy_button=True
+            )
+    
+    # Add some example inputs
+    gr.Examples(
+        examples=[
+            ["sunset beach picnic, cozy vibe"],
+            ["coffee shop laptop work, productive morning"],
+            ["hiking mountain trail, adventure time"],
+            ["cozy rainy day reading, warm blanket"],
+            ["friends dinner party, laughter and good food"],
+            ["morning yoga session, peaceful zen"],
+            ["city skyline night view, urban lights"],
+            ["homemade cooking, delicious comfort food"]
+        ],
+        inputs=description_input,
+        outputs=caption_output,
+        fn=generate_instagram_caption,
+        cache_examples=False
+    )
+    
+    # Connect the button to the function
+    generate_btn.click(
+        fn=generate_instagram_caption,
+        inputs=description_input,
+        outputs=caption_output
+    )
+    
+    # Also trigger on Enter key
+    description_input.submit(
+        fn=generate_instagram_caption,
+        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 with your descriptions for better captions!</p>
+            <p>🎨 Get 5 unique one-liner captions each time you generate</p>
+            <p>📋 Copy your favorite caption and paste it on Instagram!</p>
+        </div>
+        """
+    )
+
+# Launch the app
+if __name__ == "__main__":
+    demo.launch(), '', line)
+            # Keep it short - cut at first sentence if too long
+            if len(caption) > 80:
+                sentences = re.split(r'[.!?]', caption)
+                if sentences[0]:
+                    caption = sentences[0].strip()
+                    if caption and caption[-1] not in '.!?':
+                        caption += '.'
+            return caption
     return ""
 
 def generate_instagram_caption(description):