Create app.py

app.py

@@ -0,0 +1,229 @@
+import gradio as gr
+import torch
+from transformers import pipeline, set_seed
+import re
+import random
+
+# Set up the text generation pipeline
+try:
+    # Try to use GPU if available, otherwise fallback to CPU
+    device = 0 if torch.cuda.is_available() else -1
+    generator = pipeline(
+        "text-generation",
+        model="gpt2-medium",
+        device=device,
+        pad_token_id=50256
+    )
+except Exception as e:
+    print(f"Error loading model: {e}")
+    # Fallback to CPU with smaller model
+    generator = pipeline(
+        "text-generation",
+        model="gpt2",
+        device=-1,
+        pad_token_id=50256
+    )
+
+def clean_caption(text):
+    """Clean and format the generated caption"""
+    # 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:'):
+            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):
+    """Generate an Instagram-style caption 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, 1000))
+    
+    # Create Instagram-style prompts
+    prompts = [
+        f"Photo: {description}\nInstagram Caption: Living my best life",
+        f"Photo: {description}\nInstagram Caption: When life gives you",
+        f"Photo: {description}\nInstagram Caption: Just another day in paradise",
+        f"Photo: {description}\nInstagram Caption: Feeling grateful for",
+        f"Photo: {description}\nInstagram Caption: Making memories with",
+        f"Photo: {description}\nInstagram Caption: Can't get enough of",
+        f"Photo: {description}\nInstagram Caption: Sunday vibes",
+        f"Photo: {description}\nInstagram Caption: Good times and tan lines",
+        f"Photo: {description}\nInstagram Caption: Chasing sunsets and dreams",
+        f"Photo: {description}\nInstagram Caption: Life is better when you're"
+    ]
+    
+    # Randomly select a prompt style
+    prompt = random.choice(prompts)
+    
+    try:
+        # Generate text
+        generated = generator(
+            prompt,
+            max_length=len(prompt.split()) + 25,  # Keep it short
+            num_return_sequences=1,
+            temperature=0.8,
+            do_sample=True,
+            top_p=0.9,
+            repetition_penalty=1.1,
+            pad_token_id=50256
+        )
+        
+        # Extract and clean the caption
+        full_text = generated[0]['generated_text']
+        caption = clean_caption(full_text.replace(prompt, "").strip())
+        
+        if not caption:
+            # Fallback captions if generation fails
+            fallback_captions = [
+                f"✨ {description} moments ✨",
+                f"Living for these {description} vibes 🌟",
+                f"Just me and my {description} mood 📸",
+                f"When {description} hits different 💫",
+                f"Grateful for {description} days like these 🙏",
+                f"Making memories with {description} energy ⭐",
+                f"Can't get enough of this {description} life 🌈",
+                f"Sunday vibes: {description} edition 🌅",
+                f"Chasing dreams and {description} scenes 🎨",
+                f"Life is better with {description} moments 💕"
+            ]
+            caption = random.choice(fallback_captions)
+        
+        # Add some popular hashtags occasionally
+        if random.random() < 0.3:  # 30% chance to add hashtags
+            hashtags = ["#vibes", "#mood", "#life", "#grateful", "#memories", "#blessed", "#goodtimes", "#livingmybestlife"]
+            selected_tags = random.sample(hashtags, min(2, len(hashtags)))
+            caption += " " + " ".join(selected_tags)
+        
+        return caption
+        
+    except Exception as e:
+        print(f"Generation error: {e}")
+        return f"✨ {description} vibes ✨ #mood #life"
+
+# 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 Instagram Caption",
+                lines=4,
+                max_lines=6,
+                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>🎨 Each generation creates a unique caption with different vibes</p>
+        </div>
+        """
+    )
+
+# Launch the app
+if __name__ == "__main__":
+    demo.launch()