Update app.py

app.py

@@ -26,221 +26,28 @@ except Exception as e:
 
 def clean_caption(text):
     """Clean and format the generated caption"""
-    # Remove the prompt part and get first line
+    # Remove the prompt part
     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:'):
-            # 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
+            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
     return ""
 
 def generate_instagram_caption(description):