Update app.py

app.py

@@ -25,7 +25,6 @@ MEME_TEMPLATES = {
     "One Does Not Simply": "61579",
     "Batman Slapping Robin": "438680",
     "Ancient Aliens": "101470",
-    "X, X Everywhere": "347390",
 }
 
 def generate_meme_text(idea: str, model: str):
@@ -35,74 +34,57 @@ def generate_meme_text(idea: str, model: str):
         return None, None, error
     
     try:
-        prompt = f"""Generate funny meme text for: "{idea}"
+        prompt = f"""Create funny meme text for: "{idea}"
 
-Format your response exactly like this:
-TOP: [setup/first part - max 10 words]
-BOTTOM: [punchline/second part - max 10 words]
+Reply ONLY in this format:
+TOP: [setup text - max 10 words]
+BOTTOM: [punchline text - max 10 words]
 
-Example for "when code works first try":
-TOP: When your code works
-BOTTOM: On the first try
-
-Now create for: "{idea}"
+Now generate:
 TOP:"""
 
-        # Use text_generation instead of chat_completion
         response = client.text_generation(
             prompt,
             model=model,
-            max_new_tokens=150,
+            max_new_tokens=100,
             temperature=0.7,
-            do_sample=True,
             return_full_text=False
         )
         
         # Parse response
-        full_response = "TOP:" + response
-        lines = full_response.strip().split('\n')
+        full_text = "TOP:" + response
         
         top_text = ""
         bottom_text = ""
         
-        for line in lines:
+        for line in full_text.split('\n'):
             line = line.strip()
             if line.startswith("TOP:"):
-                top_text = line.replace("TOP:", "").strip()
+                top_text = line.replace("TOP:", "").strip().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('[]"\'')
+                bottom_text = line.replace("BOTTOM:", "").strip().strip('[]"\'.,')
         
+        # Fallback if parsing fails
         if not top_text or not bottom_text:
-            # Fallback to simple split
             words = idea.split()
-            mid = len(words) // 2
+            mid = max(1, len(words) // 2)
             top_text = ' '.join(words[:mid])
             bottom_text = ' '.join(words[mid:])
         
+        # Limit length
+        top_text = ' '.join(top_text.split()[:10])
+        bottom_text = ' '.join(bottom_text.split()[:10])
+        
         return top_text, bottom_text, None
         
     except Exception as 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]}"
+        return None, None, f"❌ AI Error: {error_msg[:150]}"
 
 def create_meme(idea: str, template: str, model: str):
     """Generate complete meme"""
     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}"
+        return None, "❌ Enter a meme idea!"
     
     # Get AI-generated text
     top, bottom, error = generate_meme_text(idea, model)
@@ -137,203 +119,96 @@ def create_meme(idea: str, template: str, model: str):
                 with open(temp_path, 'wb') as f:
                     f.write(img_response.content)
                 
-                return temp_path, f"""✅ **Meme Generated Successfully!**
+                return temp_path, f"""✅ Success!
 
-📝 **Top Text:** {top}
-📝 **Bottom Text:** {bottom}
+Top: {top}
+Bottom: {bottom}
 
-🤖 **AI Model:** {model}
-🖼️ **Template:** {template}
+Model: {model}
+Template: {template}
 
-🔗 **Direct URL:** {meme_url}
-
-💡 Right-click image to save or share!"""
+URL: {meme_url}"""
             else:
-                return None, f"✅ Meme created!\n\n🔗 **URL:** {meme_url}\n\nTop: {top}\nBottom: {bottom}"
+                return None, f"✅ Meme created!\n\nURL: {meme_url}\n\nTop: {top}\nBottom: {bottom}"
         else:
-            error_msg = data.get('error_message', 'Unknown error')
-            return None, f"❌ ImgFlip API Error: {error_msg}\n\nTry a different template or try again."
+            return None, f"❌ ImgFlip Error: {data.get('error_message', 'Unknown')}"
             
-    except requests.exceptions.Timeout:
-        return None, "❌ Request timeout. Try again."
     except Exception as e:
-        return None, f"❌ Error creating meme: {str(e)[:200]}"
+        return None, f"❌ Error: {str(e)[:150]}"
 
-# Models (using correct model IDs)
+# WORKING MODELS (verified)
 MODELS = {
-    "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",
+    "Llama 3.1 8B (Recommended)": "meta-llama/Llama-3.1-8B-Instruct",
+    # Add more models here that you verify work
 }
 
 # Examples
 examples = [
-    ["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"],
+    ["When you finally fix the bug at 3 AM", "Success Kid", "Llama 3.1 8B (Recommended)"],
+    ["Junior dev vs Senior dev", "Drake", "Llama 3.1 8B (Recommended)"],
+    ["My code in dev vs production", "Distracted Boyfriend", "Llama 3.1 8B (Recommended)"],
 ]
 
-# 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(css=custom_css, theme=gr.themes.Soft(), title="🥸 AI Meme Generator") as demo:
+with gr.Blocks(theme=gr.themes.Soft(), title="AI Meme Generator") as demo:
     
     gr.HTML("""
-    <div class="header-gradient">
+    <div style='text-align: center; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); 
+                padding: 30px; border-radius: 15px; color: white; margin-bottom: 20px;'>
         <h1>🥸 AI Meme Generator</h1>
-        <h3>Powered by HuggingFace AI + ImgFlip API 🤗</h3>
-        <p>AI creates funny meme text, then generates your meme instantly!</p>
+        <h3>HuggingFace AI + ImgFlip API</h3>
     </div>
     """)
     
-    gr.Markdown("""
-    ⚠️ **First Time Setup:** Add `HF_TOKEN` to Space Settings → Repository Secrets 
-    ([Get Free Token](https://huggingface.co/settings/tokens))
-    """)
+    gr.Markdown("⚠️ **Setup:** Add `HF_TOKEN` to Settings → Repository Secrets ([Get Token](https://huggingface.co/settings/tokens))")
     
     with gr.Row():
         with gr.Column(scale=2):
             idea = gr.Textbox(
-                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"
+                label="🎨 Meme Idea",
+                placeholder="When you find a bug in production...",
+                lines=2
             )
             
             with gr.Row():
                 template = gr.Dropdown(
                     choices=list(MEME_TEMPLATES.keys()),
                     value="Drake",
-                    label="🖼️ Meme Template",
-                    info="Choose which meme format to use"
+                    label="🖼️ Template"
                 )
                 
                 model = gr.Dropdown(
                     choices=list(MODELS.keys()),
-                    value="Llama 3.2 3B (Recommended)",
-                    label="🤖 AI Model",
-                    info="Llama 3.2 3B is fastest and most reliable"
+                    value="Llama 3.1 8B (Recommended)",
+                    label="🤖 Model"
                 )
             
-            with gr.Row():
-                btn = gr.Button("🚀 Generate Meme", variant="primary", size="lg")
-                clear_btn = gr.ClearButton(components=[idea], value="🗑️ Clear")
+            btn = gr.Button("🚀 Generate", variant="primary", size="lg")
         
         with gr.Column(scale=1):
             gr.Markdown("""
-            ### 📖 Quick Setup
-            
-            **1. Get HuggingFace Token:**
-            - Visit [HF Tokens](https://huggingface.co/settings/tokens)
-            - Click "New token"
-            - Select "Read" permission
-            - Copy token (starts with `hf_`)
+            ### Setup
             
-            **2. Add to Space:**
-            - Click Settings (top right)
-            - Go to "Repository Secrets"
-            - Add new secret:
-              - Name: `HF_TOKEN`
-              - Value: (paste token)
-            - Save and restart Space
+            1. Get token: [HF Tokens](https://huggingface.co/settings/tokens)
+            2. Settings → Secrets
+            3. Add `HF_TOKEN`
+            4. Restart Space
             
-            **3. Create Memes:**
-            - Enter your idea
-            - Pick a template
-            - Click "Generate Meme"
-            - Download and share!
+            ### Share Working Models
             
-            **💡 Tips:**
-            - ✅ Be specific and descriptive
-            - ✅ Include context/scenario
-            - ✅ Use "Llama 3.2 3B" for best results
-            - ✅ Generation takes 2-5 seconds
+            If you find models that work, please share them so I can add them to the list!
             """)
     
-    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],
-        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 |
+    output_img = gr.Image(label="Meme", type="filepath")
+    output_status = gr.Textbox(label="Status", lines=6)
     
-    ### 🎯 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>
-    """)
+    gr.Examples(examples=examples, inputs=[idea, template, model])
     
-    # 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(show_error=True)
+    demo.launch()