Update app.py

app.py

@@ -1,184 +1,198 @@
-import asyncio
-import gradio as gr
-from browser_use import Agent
-from langchain_huggingface import HuggingFaceEndpoint
-import re
-import os
-
-# Model mapping
-MODELS = {
-    "Qwen 2.5 72B (Recommended)": "Qwen/Qwen2.5-72B-Instruct",
-    "Meta Llama 3.1 70B": "meta-llama/Llama-3.1-70B-Instruct",
-    "Mixtral 8x7B": "mistralai/Mixtral-8x7B-Instruct-v0.1",
-    "Phi-3 Medium": "microsoft/Phi-3-medium-4k-instruct",
-}
-
-async def generate_meme_async(query: str, model_name: str, hf_token: str):
-    """Async function to generate meme"""
-    
-    # Get token from environment if not provided
-    if not hf_token:
-        hf_token = os.environ.get("HF_TOKEN", "")
-    
-    if not hf_token:
-        return None, "❌ Error: HuggingFace token is required! Get one from https://huggingface.co/settings/tokens"
-    
-    if not query:
-        return None, "❌ Error: Please enter a meme idea!"
-    
-    try:
-        # Initialize HuggingFace LLM
-        llm = HuggingFaceEndpoint(
-            repo_id=MODELS[model_name],
-            huggingfacehub_api_token=hf_token,
-            temperature=0.3,
-            max_new_tokens=2048,
-            top_k=50,
-            top_p=0.95
-        )
-
-        task_description = (
-            f"You are a meme generator expert. Generate a meme for: '{query}'\n\n"
-            "Steps:\n"
-            "1. Go to https://imgflip.com/memetemplates\n"
-            "2. Search for ONE main action verb from the query (like 'laugh', 'cry', 'think')\n"
-            "3. Pick a fitting meme template and click 'Add Caption'\n"
-            "4. Write Top Text (setup) and Bottom Text (punchline) about: '{query}'\n"
-            "5. Preview and adjust if needed\n"
-            "6. Click 'Generate meme'\n"
-            "7. Copy and return the image URL\n"
-        ).format(query=query)
-
-        agent = Agent(
-            task=task_description,
-            llm=llm,
-            max_actions_per_step=5,
-            max_failures=25,
-            use_vision=False
-        )
-
-        history = await agent.run()
-        final_result = history.final_result()
-        
-        # Extract meme URL
-        url_match = re.search(r'https://imgflip\.com/i/(\w+)', final_result)
-        if url_match:
-            meme_id = url_match.group(1)
-            meme_url = f"https://i.imgflip.com/{meme_id}.jpg"
-            return meme_url, f"✅ Success! Meme generated using {model_name}"
-        
-        return None, "❌ Could not generate meme. Try rephrasing your idea or use a different model."
-        
-    except Exception as e:
-        return None, f"❌ Error: {str(e)}\n\nTip: Make sure your HuggingFace token is valid!"
-
-def generate_meme(query: str, model_name: str, hf_token: str):
-    """Wrapper function for Gradio"""
-    return asyncio.run(generate_meme_async(query, model_name, hf_token))
-
-# Example meme ideas
-examples = [
-    ["When you finally fix the bug at 3 AM", "Qwen 2.5 72B (Recommended)"],
-    ["Junior dev vs Senior dev looking at the same error", "Meta Llama 3.1 70B"],
-    ["My code in development vs in production", "Qwen 2.5 72B (Recommended)"],
-    ["Trying to explain AI to my parents", "Mixtral 8x7B"],
-    ["Me before coffee vs after coffee", "Qwen 2.5 72B (Recommended)"],
-]
-
-# Create Gradio interface
-with gr.Blocks(theme=gr.themes.Soft(), title="🥸 AI Meme Generator") as demo:
-    
-    gr.Markdown("""
-    # 🥸 AI Meme Generator
-    ### Powered by Free HuggingFace Models 🤗
-    
-    This AI agent automatically browses ImgFlip and creates memes based on your description!
-    """)
-    
-    with gr.Row():
-        with gr.Column(scale=2):
-            query_input = gr.Textbox(
-                label="🎨 Describe Your Meme Idea",
-                placeholder="Example: When the client says 'just one small change'...",
-                lines=3
-            )
-            
-            model_dropdown = gr.Dropdown(
-                choices=list(MODELS.keys()),
-                value="Qwen 2.5 72B (Recommended)",
-                label="🤖 Select AI Model (All Free!)",
-                info="Choose which model to use for meme generation"
-            )
-            
-            token_input = gr.Textbox(
-                label="🔑 HuggingFace Token (Optional if set in Space secrets)",
-                placeholder="hf_...",
-                type="password",
-                value=os.environ.get("HF_TOKEN", "")
-            )
-            
-            with gr.Row():
-                generate_btn = gr.Button("🚀 Generate Meme", variant="primary", size="lg")
-                clear_btn = gr.ClearButton(components=[query_input], value="🗑️ Clear")
-        
-        with gr.Column(scale=1):
-            gr.Markdown("""
-            ### 📖 Quick Guide
-            
-            **First Time?**
-            1. Get FREE token: [HuggingFace Tokens](https://huggingface.co/settings/tokens)
-            2. Paste it above
-            3. Describe your meme
-            4. Click Generate!
-            
-            **Tips for Best Results:**
-            - Be specific and descriptive
-            - Include the emotion/scenario
-            - Think about popular meme formats
-            - Qwen 2.5 72B works best!
-            
-            **Examples:**
-            - "Confident vs reality"
-            - "Before/after situations"
-            - "Expectations vs reality"
-            - "Relatable daily struggles"
-            """)
-    
-    with gr.Row():
-        output_image = gr.Image(label="Generated Meme", type="filepath")
-    
-    status_output = gr.Textbox(label="Status", lines=2)
-    
-    gr.Markdown("""
-    ### 💡 Example Meme Ideas (Click to try!)
-    """)
-    
-    gr.Examples(
-        examples=examples,
-        inputs=[query_input, model_dropdown],
-        label="Click any example to auto-fill"
-    )
-    
-    gr.Markdown("""
-    ---
-    <div style='text-align: center;'>
-        <p><strong>🤗 100% Free & Open Source</strong></p>
-        <p>Powered by HuggingFace Inference API • Browser-Use • Gradio</p>
-        <p>⭐ Star on GitHub if you like this project!</p>
-    </div>
-    """)
-    
-    # Connect button to function
-    generate_btn.click(
-        fn=generate_meme,
-        inputs=[query_input, model_dropdown, token_input],
-        outputs=[output_image, status_output]
-    )
-
-# Launch the app
-if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False
+import asyncio
+import gradio as gr
+from browser_use import Agent
+from langchain_huggingface import HuggingFaceEndpoint
+import re
+import os
+import subprocess
+import sys
+
+# Install Playwright on startup
+def install_playwright():
+    try:
+        subprocess.run([sys.executable, "-m", "playwright", "install", "chromium"], check=True)
+        subprocess.run([sys.executable, "-m", "playwright", "install-deps", "chromium"], check=True)
+        print("✅ Playwright installed successfully")
+    except Exception as e:
+        print(f"⚠️ Playwright installation warning: {e}")
+
+# Run installation
+install_playwright()
+
+# Model mapping
+MODELS = {
+    "Qwen 2.5 72B (Recommended)": "Qwen/Qwen2.5-72B-Instruct",
+    "Meta Llama 3.1 70B": "meta-llama/Llama-3.1-70B-Instruct",
+    "Mixtral 8x7B": "mistralai/Mixtral-8x7B-Instruct-v0.1",
+    "Phi-3 Medium": "microsoft/Phi-3-medium-4k-instruct",
+}
+
+async def generate_meme_async(query: str, model_name: str, hf_token: str):
+    """Async function to generate meme"""
+    
+    # Get token from environment if not provided
+    if not hf_token:
+        hf_token = os.environ.get("HF_TOKEN", "")
+    
+    if not hf_token:
+        return None, "❌ Error: HuggingFace token is required! Get one from https://huggingface.co/settings/tokens"
+    
+    if not query:
+        return None, "❌ Error: Please enter a meme idea!"
+    
+    try:
+        # Initialize HuggingFace LLM
+        llm = HuggingFaceEndpoint(
+            repo_id=MODELS[model_name],
+            huggingfacehub_api_token=hf_token,
+            temperature=0.3,
+            max_new_tokens=2048,
+            top_k=50,
+            top_p=0.95
+        )
+
+        task_description = (
+            f"You are a meme generator expert. Generate a meme for: '{query}'\n\n"
+            "Steps:\n"
+            "1. Go to https://imgflip.com/memetemplates\n"
+            "2. Search for ONE main action verb from the query (like 'laugh', 'cry', 'think')\n"
+            "3. Pick a fitting meme template and click 'Add Caption'\n"
+            "4. Write Top Text (setup) and Bottom Text (punchline) about: '{query}'\n"
+            "5. Preview and adjust if needed\n"
+            "6. Click 'Generate meme'\n"
+            "7. Copy and return the image URL\n"
+        ).format(query=query)
+
+        agent = Agent(
+            task=task_description,
+            llm=llm,
+            max_actions_per_step=5,
+            max_failures=25,
+            use_vision=False
+        )
+
+        history = await agent.run()
+        final_result = history.final_result()
+        
+        # Extract meme URL
+        url_match = re.search(r'https://imgflip\.com/i/(\w+)', final_result)
+        if url_match:
+            meme_id = url_match.group(1)
+            meme_url = f"https://i.imgflip.com/{meme_id}.jpg"
+            return meme_url, f"✅ Success! Meme generated using {model_name}"
+        
+        return None, "❌ Could not generate meme. Try rephrasing your idea or use a different model."
+        
+    except Exception as e:
+        return None, f"❌ Error: {str(e)}\n\nTip: Make sure your HuggingFace token is valid!"
+
+def generate_meme(query: str, model_name: str, hf_token: str):
+    """Wrapper function for Gradio"""
+    return asyncio.run(generate_meme_async(query, model_name, hf_token))
+
+# Example meme ideas
+examples = [
+    ["When you finally fix the bug at 3 AM", "Qwen 2.5 72B (Recommended)"],
+    ["Junior dev vs Senior dev looking at the same error", "Meta Llama 3.1 70B"],
+    ["My code in development vs in production", "Qwen 2.5 72B (Recommended)"],
+    ["Trying to explain AI to my parents", "Mixtral 8x7B"],
+    ["Me before coffee vs after coffee", "Qwen 2.5 72B (Recommended)"],
+]
+
+# Create Gradio interface
+with gr.Blocks(theme=gr.themes.Soft(), title="🥸 AI Meme Generator") as demo:
+    
+    gr.Markdown("""
+    # 🥸 AI Meme Generator
+    ### Powered by Free HuggingFace Models 🤗
+    
+    This AI agent automatically browses ImgFlip and creates memes based on your description!
+    """)
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            query_input = gr.Textbox(
+                label="🎨 Describe Your Meme Idea",
+                placeholder="Example: When the client says 'just one small change'...",
+                lines=3
+            )
+            
+            model_dropdown = gr.Dropdown(
+                choices=list(MODELS.keys()),
+                value="Qwen 2.5 72B (Recommended)",
+                label="🤖 Select AI Model (All Free!)",
+                info="Choose which model to use for meme generation"
+            )
+            
+            token_input = gr.Textbox(
+                label="🔑 HuggingFace Token (Optional if set in Space secrets)",
+                placeholder="hf_...",
+                type="password",
+                value=os.environ.get("HF_TOKEN", "")
+            )
+            
+            with gr.Row():
+                generate_btn = gr.Button("🚀 Generate Meme", variant="primary", size="lg")
+                clear_btn = gr.ClearButton(components=[query_input], value="🗑️ Clear")
+        
+        with gr.Column(scale=1):
+            gr.Markdown("""
+            ### 📖 Quick Guide
+            
+            **First Time?**
+            1. Get FREE token: [HuggingFace Tokens](https://huggingface.co/settings/tokens)
+            2. Paste it above
+            3. Describe your meme
+            4. Click Generate!
+            
+            **Tips for Best Results:**
+            - Be specific and descriptive
+            - Include the emotion/scenario
+            - Think about popular meme formats
+            - Qwen 2.5 72B works best!
+            
+            **Examples:**
+            - "Confident vs reality"
+            - "Before/after situations"
+            - "Expectations vs reality"
+            - "Relatable daily struggles"
+            """)
+    
+    with gr.Row():
+        output_image = gr.Image(label="Generated Meme", type="filepath")
+    
+    status_output = gr.Textbox(label="Status", lines=2)
+    
+    gr.Markdown("""
+    ### 💡 Example Meme Ideas (Click to try!)
+    """)
+    
+    gr.Examples(
+        examples=examples,
+        inputs=[query_input, model_dropdown],
+        label="Click any example to auto-fill"
+    )
+    
+    gr.Markdown("""
+    ---
+    <div style='text-align: center;'>
+        <p><strong>🤗 100% Free & Open Source</strong></p>
+        <p>Powered by HuggingFace Inference API • Browser-Use • Gradio</p>
+        <p>⭐ Star on GitHub if you like this project!</p>
+    </div>
+    """)
+    
+    # Connect button to function
+    generate_btn.click(
+        fn=generate_meme,
+        inputs=[query_input, model_dropdown, token_input],
+        outputs=[output_image, status_output]
+    )
+
+# Launch the app
+if __name__ == "__main__":
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
     )