Update app.py

app.py

@@ -1,161 +1,143 @@
-import asyncio
 import gradio as gr
-from browser_use import Agent
-from langchain_openai import ChatOpenAI
-import re
+from huggingface_hub import InferenceClient
 import os
 import requests
+import random
 
-# Model mapping
-MODELS = {
-    "Llama 3.2 3B (Fast)": "meta-llama/Llama-3.2-3B-Instruct",
-    "Qwen 2.5 72B": "Qwen/Qwen2.5-72B-Instruct",
-    "Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3",
-    "Phi-3.5 Mini": "microsoft/Phi-3.5-mini-instruct",
+# Initialize HF client
+def get_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 = {
+    "Drake": "181913649",
+    "Distracted Boyfriend": "112126428",
+    "Two Buttons": "87743020",
+    "Expanding Brain": "93895088",
+    "Change My Mind": "129242436",
+    "Disaster Girl": "97984",
+    "Success Kid": "61544",
+    "Bad Luck Brian": "61585",
+    "Philosoraptor": "61516",
+    "One Does Not Simply": "61579",
 }
 
-async def generate_meme_async(query: str, model_name: str):
-    """Generate meme using AI browser automation"""
-    
-    hf_token = os.environ.get("HF_TOKEN", "")
-    
-    if not hf_token:
-        yield None, "❌ **Setup Required**\n\nAdd HF_TOKEN to Space Settings → Repository Secrets\n\nGet token: https://huggingface.co/settings/tokens"
-        return
-    
-    if not query or len(query.strip()) < 5:
-        yield None, "❌ Please enter a detailed meme idea (minimum 5 characters)"
-        return
+def generate_meme_text(idea: str, model: str):
+    """Generate meme text using AI"""
+    client, error = get_client()
+    if error:
+        return None, None, error
     
     try:
-        yield None, f"🔄 Initializing {model_name}..."
-        
-        llm = ChatOpenAI(
-            base_url="https://api-inference.huggingface.co/v1/",
-            api_key=hf_token,
-            model=MODELS[model_name],
-            temperature=0.3,
-            max_tokens=2048,
-            timeout=180,
-        )
-        
-        yield None, f"🤖 Using {model_name}\n🌐 Starting browser..."
-
-        task = f"""Create a meme about: "{query}"
+        prompt = f"""Generate meme text for: "{idea}"
 
-Steps:
-1. Go to https://imgflip.com/memegenerator
-2. Select a popular meme template that fits: "{query}"
-3. Click on the template
-4. Add Top Text (setup) and Bottom Text (punchline) about: "{query}"
-5. Click "Generate Meme"
-6. Return the meme URL
+Return ONLY in this exact format:
+TOP: [text for top of meme]
+BOTTOM: [text for bottom of meme]
 
-Make it funny and relevant to: {query}
-Return only the URL."""
+Make it funny, relatable, and max 10 words each line."""
 
-        yield None, "🎨 AI creating meme...\n\n⏰ Takes 30-90 seconds"
-
-        agent = Agent(
-            task=task,
-            llm=llm,
-            max_actions_per_step=10,
-            use_vision=False
-        )
-
-        history = await agent.run()
-        result = history.final_result()
+        response = ""
+        for message in client.chat_completion(
+            messages=[{"role": "user", "content": prompt}],
+            model=model,
+            max_tokens=200,
+            stream=True,
+        ):
+            if message.choices[0].delta.content:
+                response += message.choices[0].delta.content
         
-        yield None, "🔍 Extracting meme URL..."
+        # Parse response
+        lines = response.strip().split('\n')
+        top_text = ""
+        bottom_text = ""
         
-        # Find meme URL
-        patterns = [
-            r'https://i\.imgflip\.com/(\w+)\.jpg',
-            r'https://imgflip\.com/i/(\w+)',
-            r'imgflip\.com/i/(\w+)',
-        ]
+        for line in lines:
+            if line.startswith("TOP:"):
+                top_text = line.replace("TOP:", "").strip()
+            elif line.startswith("BOTTOM:"):
+                bottom_text = line.replace("BOTTOM:", "").strip()
         
-        meme_id = None
-        for pattern in patterns:
-            match = re.search(pattern, result)
-            if match:
-                meme_id = match.group(1)
-                break
+        if not top_text or not bottom_text:
+            return None, None, "❌ Could not parse AI response. Try again."
         
-        if meme_id:
-            meme_url = f"https://i.imgflip.com/{meme_id}.jpg"
-            
-            try:
-                yield None, "📥 Downloading meme..."
-                response = requests.get(meme_url, timeout=10)
-                if response.status_code == 200:
-                    temp_path = f"/tmp/meme_{meme_id}.jpg"
-                    with open(temp_path, 'wb') as f:
-                        f.write(response.content)
-                    yield temp_path, f"✅ **Success!**\n\nGenerated with {model_name}\n\n🔗 {meme_url}"
-                else:
-                    yield None, f"✅ Meme created!\n\n🔗 **Link:** {meme_url}"
-            except:
-                yield None, f"✅ Meme created!\n\n🔗 **Link:** {meme_url}"
-        else:
-            yield None, f"⚠️ Could not find meme URL\n\nAgent output:\n{result[:300]}..."
+        return top_text, bottom_text, None
         
     except Exception as e:
-        error = str(e)
-        
-        if "rate limit" in error.lower() or "429" in error:
-            yield None, "❌ **Rate Limit**\n\nWait 60 seconds and try again\n\nTip: Use Llama 3.2 3B (faster)"
-        elif "token" in error.lower() or "401" in error:
-            yield None, "❌ **Auth Failed**\n\nCheck HF_TOKEN in Space Secrets\n\nToken needs 'read' permission"
-        elif "timeout" in error.lower():
-            yield None, "❌ **Timeout**\n\nTry Llama 3.2 3B or simplify your idea"
-        else:
-            yield None, f"❌ **Error**\n\n{error[:200]}...\n\nTry: Llama 3.2 3B or wait 30s"
+        return None, None, f"❌ Error: {str(e)}"
 
-def generate_meme_wrapper(query: str, model_name: str):
-    """Sync wrapper for Gradio"""
-    try:
-        loop = asyncio.get_event_loop()
-        if loop.is_closed():
-            raise RuntimeError
-    except:
-        loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(loop)
-    
-    async def run():
-        async for result in generate_meme_async(query, model_name):
-            yield result
+def create_meme(idea: str, template: str, model: str):
+    """Generate complete meme"""
+    if not idea:
+        return None, "❌ Enter a meme idea!"
+    
+    # Get AI-generated text
+    top, bottom, error = generate_meme_text(idea, model)
+    if error:
+        return None, error
+    
+    # Get template ID
+    template_id = MEME_TEMPLATES.get(template, "181913649")
+    
+    # Create meme using ImgFlip API
+    url = "https://api.imgflip.com/caption_image"
+    
+    payload = {
+        'template_id': template_id,
+        'username': 'imgflip_huggingface',
+        'password': 'imgflip_huggingface',
+        'text0': top,
+        'text1': bottom
+    }
     
-    async def collect():
-        async for item in run():
-            yield item
-    
-    for result in loop.run_until_complete(async_list(collect())):
-        yield result
+    try:
+        response = requests.post(url, data=payload, timeout=10)
+        data = response.json()
+        
+        if data['success']:
+            meme_url = data['data']['url']
+            
+            # Download image
+            img_response = requests.get(meme_url, timeout=10)
+            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}"
+            else:
+                return None, f"✅ Meme created!\n\nURL: {meme_url}"
+        else:
+            return None, f"❌ ImgFlip API error: {data.get('error_message', 'Unknown')}"
+            
+    except Exception as e:
+        return None, f"❌ Error creating meme: {str(e)}"
 
-async def async_list(gen):
-    """Convert async gen to list"""
-    items = []
-    async for item in gen:
-        items.append(item)
-        yield item
+# Models
+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",
+}
 
 # Examples
 examples = [
-    ["When you finally fix the bug at 3 AM", "Llama 3.2 3B (Fast)"],
-    ["Junior dev vs Senior dev", "Llama 3.2 3B (Fast)"],
-    ["My code in dev vs production", "Llama 3.2 3B (Fast)"],
-    ["Trying to explain AI to parents", "Mistral 7B"],
-    ["Me before vs after coffee", "Llama 3.2 3B (Fast)"],
+    ["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"],
 ]
 
 # UI
 with gr.Blocks(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 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>Free HuggingFace Models + Browser Automation</h3>
+        <h3>Powered by HuggingFace + ImgFlip API</h3>
     </div>
     """)
     
@@ -163,19 +145,26 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI Meme Generator") as demo:
     
     with gr.Row():
         with gr.Column(scale=2):
-            query_input = gr.Textbox(
+            idea = gr.Textbox(
                 label="🎨 Meme Idea",
-                placeholder="When the client says 'just one small change'...",
-                lines=3
+                placeholder="When you find a bug in production...",
+                lines=2
             )
             
-            model_dropdown = gr.Dropdown(
-                choices=list(MODELS.keys()),
-                value="Llama 3.2 3B (Fast)",
-                label="🤖 AI Model"
-            )
+            with gr.Row():
+                template = gr.Dropdown(
+                    choices=list(MEME_TEMPLATES.keys()),
+                    value="Drake",
+                    label="🖼️ Template"
+                )
+                
+                model = gr.Dropdown(
+                    choices=list(MODELS.keys()),
+                    value="Llama 3.2 3B",
+                    label="🤖 AI Model"
+                )
             
-            generate_btn = gr.Button("🚀 Generate", variant="primary", size="lg")
+            btn = gr.Button("🚀 Generate Meme", variant="primary", size="lg")
         
         with gr.Column(scale=1):
             gr.Markdown("""
@@ -183,62 +172,36 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI Meme Generator") as demo:
             
             **1. Get Token:**
             - [HF Tokens](https://huggingface.co/settings/tokens)
-            - Click "New token"
+            - Create "Read" token
             - Copy it
             
             **2. Add to Space:**
             - Settings → Secrets
             - Name: `HF_TOKEN`
             - Value: (paste)
+            - Restart
             
             **3. Generate!**
-            - Describe meme
-            - Wait 30-90s
+            - Pick template
+            - Enter idea
+            - Click generate
             
-            **Tips:**
-            - Use Llama 3.2 3B
-            - Be specific
-            - Include emotion
+            ⚡ **Instant results!**
             """)
     
-    with gr.Row():
-        output_image = gr.Image(label="🖼️ Generated Meme", type="filepath", height=400)
-    
-    status_output = gr.Textbox(label="📊 Status", lines=6)
+    output_img = gr.Image(label="Generated Meme", type="filepath")
+    output_status = gr.Textbox(label="Status", lines=4)
     
     gr.Examples(
         examples=examples,
-        inputs=[query_input, model_dropdown],
-        label="💡 Examples"
+        inputs=[idea, template, model]
     )
     
-    gr.Markdown("""
-    ---
-    
-    ### 🔧 Troubleshooting
-    
-    | Issue | Fix |
-    |-------|-----|
-    | Token not found | Add HF_TOKEN in Secrets |
-    | Rate limit | Wait 60s |
-    | Slow | Use Llama 3.2 3B |
-    | Build error | Check Dockerfile committed |
-    
-    ---
-    
-    <div style='text-align: center; padding: 20px; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); border-radius: 10px; color: white;'>
-        <p><strong>🤗 100% Free & Open Source</strong></p>
-        <p>HuggingFace • Browser-Use • LangChain • Gradio</p>
-        <p>⭐ Star if you like it!</p>
-    </div>
-    """)
-    
-    generate_btn.click(
-        fn=generate_meme_wrapper,
-        inputs=[query_input, model_dropdown],
-        outputs=[output_image, status_output],
-        show_progress=True
+    btn.click(
+        fn=lambda i, t, m: create_meme(i, t, MODELS[m]),
+        inputs=[idea, template, model],
+        outputs=[output_img, output_status]
     )
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860, show_error=True)
+    demo.launch()