Update app.py

app.py

@@ -4,14 +4,14 @@ import os
 import requests
 import random
 
-# Initialize HF client
 def get_client():
+    """Get HF client"""
     token = os.environ.get("HF_TOKEN", "")
     if not token:
         return None, "❌ Add HF_TOKEN to Space Settings → Repository Secrets"
     return InferenceClient(token=token), None
 
-# Meme templates mapping
+# Meme templates
 MEME_TEMPLATES = {
     "Drake": "181913649",
     "Distracted Boyfriend": "112126428",
@@ -23,6 +23,9 @@ MEME_TEMPLATES = {
     "Bad Luck Brian": "61585",
     "Philosoraptor": "61516",
     "One Does Not Simply": "61579",
+    "Batman Slapping Robin": "438680",
+    "Ancient Aliens": "101470",
+    "X, X Everywhere": "347390",
 }
 
 def generate_meme_text(idea: str, model: str):
@@ -32,47 +35,74 @@ def generate_meme_text(idea: str, model: str):
         return None, None, error
     
     try:
-        prompt = f"""Generate meme text for: "{idea}"
+        prompt = f"""Generate funny meme text for: "{idea}"
 
-Return ONLY in this exact format:
-TOP: [text for top of meme]
-BOTTOM: [text for bottom of meme]
+Format your response exactly like this:
+TOP: [setup/first part - max 10 words]
+BOTTOM: [punchline/second part - max 10 words]
 
-Make it funny, relatable, and max 10 words each line."""
+Example for "when code works first try":
+TOP: When your code works
+BOTTOM: On the first try
 
-        response = ""
-        for message in client.chat_completion(
-            messages=[{"role": "user", "content": prompt}],
+Now create for: "{idea}"
+TOP:"""
+
+        # Use text_generation instead of chat_completion
+        response = client.text_generation(
+            prompt,
             model=model,
-            max_tokens=200,
-            stream=True,
-        ):
-            if message.choices[0].delta.content:
-                response += message.choices[0].delta.content
+            max_new_tokens=150,
+            temperature=0.7,
+            do_sample=True,
+            return_full_text=False
+        )
         
         # Parse response
-        lines = response.strip().split('\n')
+        full_response = "TOP:" + response
+        lines = full_response.strip().split('\n')
+        
         top_text = ""
         bottom_text = ""
         
         for line in lines:
+            line = line.strip()
             if line.startswith("TOP:"):
                 top_text = line.replace("TOP:", "").strip()
             elif line.startswith("BOTTOM:"):
                 bottom_text = line.replace("BOTTOM:", "").strip()
         
+        # Clean up text
+        top_text = top_text.split('\n')[0].strip('[]"\'')
+        bottom_text = bottom_text.split('\n')[0].strip('[]"\'')
+        
         if not top_text or not bottom_text:
-            return None, None, "❌ Could not parse AI response. Try again."
+            # Fallback to simple split
+            words = idea.split()
+            mid = len(words) // 2
+            top_text = ' '.join(words[:mid])
+            bottom_text = ' '.join(words[mid:])
         
         return top_text, bottom_text, None
         
     except Exception as e:
-        return None, None, f"❌ Error: {str(e)}"
+        error_msg = str(e)
+        
+        if "404" in error_msg:
+            return None, None, f"❌ Model not available. Try a different model.\n\nError: {error_msg}"
+        elif "503" in error_msg:
+            return None, None, "❌ Model is loading. Wait 30 seconds and try again."
+        elif "429" in error_msg:
+            return None, None, "❌ Rate limit. Wait 60 seconds."
+        else:
+            return None, None, f"❌ Error: {error_msg[:200]}"
 
 def create_meme(idea: str, template: str, model: str):
     """Generate complete meme"""
-    if not idea:
-        return None, "❌ Enter a meme idea!"
+    if not idea or len(idea.strip()) < 3:
+        return None, "❌ Enter a meme idea (minimum 3 characters)!"
+    
+    status_msg = f"🔄 Generating meme text with AI...\n\nIdea: {idea}\nTemplate: {template}\nModel: {model}"
     
     # Get AI-generated text
     top, bottom, error = generate_meme_text(idea, model)
@@ -94,114 +124,216 @@ def create_meme(idea: str, template: str, model: str):
     }
     
     try:
-        response = requests.post(url, data=payload, timeout=10)
+        response = requests.post(url, data=payload, timeout=15)
         data = response.json()
         
         if data['success']:
             meme_url = data['data']['url']
             
             # Download image
-            img_response = requests.get(meme_url, timeout=10)
+            img_response = requests.get(meme_url, timeout=15)
             if img_response.status_code == 200:
                 temp_path = f"/tmp/meme_{random.randint(1000,9999)}.jpg"
                 with open(temp_path, 'wb') as f:
                     f.write(img_response.content)
                 
-                return temp_path, f"✅ Success!\n\nTop: {top}\nBottom: {bottom}\n\nURL: {meme_url}"
+                return temp_path, f"""✅ **Meme Generated Successfully!**
+
+📝 **Top Text:** {top}
+📝 **Bottom Text:** {bottom}
+
+🤖 **AI Model:** {model}
+🖼️ **Template:** {template}
+
+🔗 **Direct URL:** {meme_url}
+
+💡 Right-click image to save or share!"""
             else:
-                return None, f"✅ Meme created!\n\nURL: {meme_url}"
+                return None, f"✅ Meme created!\n\n🔗 **URL:** {meme_url}\n\nTop: {top}\nBottom: {bottom}"
         else:
-            return None, f"❌ ImgFlip API error: {data.get('error_message', 'Unknown')}"
+            error_msg = data.get('error_message', 'Unknown error')
+            return None, f"❌ ImgFlip API Error: {error_msg}\n\nTry a different template or try again."
             
+    except requests.exceptions.Timeout:
+        return None, "❌ Request timeout. Try again."
     except Exception as e:
-        return None, f"❌ Error creating meme: {str(e)}"
+        return None, f"❌ Error creating meme: {str(e)[:200]}"
 
-# Models
+# Models (using correct model IDs)
 MODELS = {
-    "Qwen 2.5 72B": "Qwen/Qwen2.5-72B-Instruct",
-    "Llama 3.2 3B": "meta-llama/Llama-3.2-3B-Instruct",
-    "Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3",
+    "Llama 3.2 3B (Recommended)": "meta-llama/Llama-3.2-3B-Instruct",
+    "Qwen 2.5 7B": "Qwen/Qwen2.5-7B-Instruct",
+    "Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.2",
+    "Phi 3 Mini": "microsoft/Phi-3-mini-4k-instruct",
+    "Gemma 2B": "google/gemma-2b-it",
 }
 
 # Examples
 examples = [
-    ["When you finally fix the bug at 3 AM", "Success Kid", "Llama 3.2 3B"],
-    ["Trying to explain code to non-developers", "Distracted Boyfriend", "Qwen 2.5 72B"],
-    ["My code vs production", "Drake", "Llama 3.2 3B"],
+    ["When you finally fix the bug at 3 AM", "Success Kid", "Llama 3.2 3B (Recommended)"],
+    ["Junior dev vs Senior dev looking at same error", "Drake", "Llama 3.2 3B (Recommended)"],
+    ["My code in development vs in production", "Distracted Boyfriend", "Qwen 2.5 7B"],
+    ["Trying to explain code to non-developers", "Ancient Aliens", "Llama 3.2 3B (Recommended)"],
+    ["Me before coffee vs after coffee", "Two Buttons", "Llama 3.2 3B (Recommended)"],
+    ["When the code works on first try", "Disaster Girl", "Mistral 7B"],
 ]
 
+# Custom CSS
+custom_css = """
+.gradio-container {max-width: 1200px !important;}
+.header-gradient {
+    text-align: center;
+    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+    padding: 30px;
+    border-radius: 15px;
+    color: white;
+    margin-bottom: 20px;
+}
+"""
+
 # UI
-with gr.Blocks(theme=gr.themes.Soft(), title="AI Meme Generator") as demo:
+with gr.Blocks(css=custom_css, theme=gr.themes.Soft(), title="🥸 AI Meme Generator") as demo:
     
     gr.HTML("""
-    <div style='text-align: center; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); 
-                padding: 30px; border-radius: 15px; color: white; margin-bottom: 20px;'>
+    <div class="header-gradient">
         <h1>🥸 AI Meme Generator</h1>
-        <h3>Powered by HuggingFace + ImgFlip API</h3>
+        <h3>Powered by HuggingFace AI + ImgFlip API 🤗</h3>
+        <p>AI creates funny meme text, then generates your meme instantly!</p>
     </div>
     """)
     
-    gr.Markdown("⚠️ **Setup:** Add `HF_TOKEN` to Space Settings → Repository Secrets ([Get Token](https://huggingface.co/settings/tokens))")
+    gr.Markdown("""
+    ⚠️ **First Time Setup:** Add `HF_TOKEN` to Space Settings → Repository Secrets 
+    ([Get Free Token](https://huggingface.co/settings/tokens))
+    """)
     
     with gr.Row():
         with gr.Column(scale=2):
             idea = gr.Textbox(
-                label="🎨 Meme Idea",
-                placeholder="When you find a bug in production...",
-                lines=2
+                label="🎨 Your Meme Idea",
+                placeholder="Example: When you find a bug in production on Friday...",
+                lines=2,
+                info="Describe your meme idea - AI will create funny text for it"
             )
             
             with gr.Row():
                 template = gr.Dropdown(
                     choices=list(MEME_TEMPLATES.keys()),
                     value="Drake",
-                    label="🖼️ Template"
+                    label="🖼️ Meme Template",
+                    info="Choose which meme format to use"
                 )
                 
                 model = gr.Dropdown(
                     choices=list(MODELS.keys()),
-                    value="Llama 3.2 3B",
-                    label="🤖 AI Model"
+                    value="Llama 3.2 3B (Recommended)",
+                    label="🤖 AI Model",
+                    info="Llama 3.2 3B is fastest and most reliable"
                 )
             
-            btn = gr.Button("🚀 Generate Meme", variant="primary", size="lg")
+            with gr.Row():
+                btn = gr.Button("🚀 Generate Meme", variant="primary", size="lg")
+                clear_btn = gr.ClearButton(components=[idea], value="🗑️ Clear")
         
         with gr.Column(scale=1):
             gr.Markdown("""
-            ### �� Setup
+            ### 📖 Quick Setup
             
-            **1. Get Token:**
-            - [HF Tokens](https://huggingface.co/settings/tokens)
-            - Create "Read" token
-            - Copy it
+            **1. Get HuggingFace Token:**
+            - Visit [HF Tokens](https://huggingface.co/settings/tokens)
+            - Click "New token"
+            - Select "Read" permission
+            - Copy token (starts with `hf_`)
             
             **2. Add to Space:**
-            - Settings → Secrets
-            - Name: `HF_TOKEN`
-            - Value: (paste)
-            - Restart
+            - Click Settings (top right)
+            - Go to "Repository Secrets"
+            - Add new secret:
+              - Name: `HF_TOKEN`
+              - Value: (paste token)
+            - Save and restart Space
             
-            **3. Generate!**
-            - Pick template
-            - Enter idea
-            - Click generate
+            **3. Create Memes:**
+            - Enter your idea
+            - Pick a template
+            - Click "Generate Meme"
+            - Download and share!
             
-            ⚡ **Instant results!**
+            **💡 Tips:**
+            - ✅ Be specific and descriptive
+            - ✅ Include context/scenario
+            - ✅ Use "Llama 3.2 3B" for best results
+            - ✅ Generation takes 2-5 seconds
             """)
     
-    output_img = gr.Image(label="Generated Meme", type="filepath")
-    output_status = gr.Textbox(label="Status", lines=4)
+    with gr.Row():
+        output_img = gr.Image(
+            label="🖼️ Your Generated Meme", 
+            type="filepath",
+            height=400,
+            show_download_button=True
+        )
+    
+    output_status = gr.Textbox(
+        label="📊 Status & Details", 
+        lines=8,
+        show_copy_button=True
+    )
+    
+    gr.Markdown("### 💡 Example Ideas (Click to Try!)")
     
     gr.Examples(
         examples=examples,
-        inputs=[idea, template, model]
+        inputs=[idea, template, model],
+        label="Click any example to auto-fill and try it out"
     )
     
+    gr.Markdown("""
+    ---
+    
+    ### 🔧 Troubleshooting
+    
+    | Problem | Solution |
+    |---------|----------|
+    | ❌ Token not found | Add `HF_TOKEN` in Space Settings → Repository Secrets |
+    | ❌ 404 Model error | Model may not be available - try "Llama 3.2 3B" |
+    | ❌ 503 Model loading | Wait 30 seconds for model to wake up |
+    | ❌ 429 Rate limit | Wait 60 seconds (free tier limits) |
+    | ❌ ImgFlip error | Try different template or retry |
+    
+    ### 🎯 How It Works
+    
+    1. **You describe** your meme idea
+    2. **AI generates** funny top and bottom text
+    3. **ImgFlip creates** the meme with your chosen template
+    4. **Download** and share instantly!
+    
+    ### ⚡ Features
+    
+    - ✅ **Fast:** 2-5 second generation
+    - ✅ **Smart:** AI creates contextual humor
+    - ✅ **Easy:** Just describe and click
+    - ✅ **Free:** 100% free to use
+    - ✅ **No waiting:** Instant results
+    
+    ---
+    
+    <div style='text-align: center; padding: 25px; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); 
+                border-radius: 10px; color: white;'>
+        <p style='font-size: 1.2em; margin: 0;'><strong>🤗 100% Free & Open Source</strong></p>
+        <p style='margin: 10px 0;'>Built with: HuggingFace Inference API • ImgFlip API • Gradio</p>
+        <p style='margin: 10px 0;'>⭐ Star this Space if you find it useful!</p>
+        <p style='margin: 0; font-size: 0.9em;'><em>Free tier has rate limits - be patient between generations</em></p>
+    </div>
+    """)
+    
+    # Event handler
     btn.click(
         fn=lambda i, t, m: create_meme(i, t, MODELS[m]),
         inputs=[idea, template, model],
         outputs=[output_img, output_status]
     )
 
+# Launch
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(show_error=True)