Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
+from huggingface_hub import InferenceClient, HfFolder
 import os
 import requests
 import random
@@ -9,78 +9,92 @@ import re
 
 def get_client():
     """Get Hugging Face client, handle token error."""
-    token = os.environ.get("HF_TOKEN", "")
+    token = os.environ.get("HF_TOKEN")
+    if not token:
+        # As a fallback for local testing, you might use this, but it's better to use secrets in Spaces
+        try:
+            token = HfFolder.get_token()
+        except Exception:
+            pass
     if not token:
         return None, "❌ **HuggingFace Token Required**\n\n**Setup:**\n1. Go to Space Settings → Repository Secrets\n2. Add secret: Name=`HF_TOKEN`, Value=(your token)\n3. Get token: https://huggingface.co/settings/tokens\n4. Restart Space"
     return InferenceClient(token=token), None
 
-def generate_meme_text(idea: str, model: str):
-    """Generate meme text using a STABLE AI model."""
+def generate_meme_text(idea: str):
+    """
+    Generate meme text by trying a list of models until one succeeds.
+    This is the auto-healing part.
+    """
     client, error = get_client()
     if error:
-        return None, None, error
-    
-    try:
-        # A simple, direct prompt for a text-to-text model like FLAN-T5
-        prompt = f"""Generate two short, funny lines for a meme about: "{idea}".
-
-Use this exact format:
-Top: [text] | Bottom: [text]"""
-
-        # Using text_generation which is compatible with these models
-        response = client.text_generation(
-            prompt,
-            model=model,
-            max_new_tokens=60,
-            temperature=0.9,
-            do_sample=True,
-        )
-        
-        # --- Robust Parsing Logic ---
-        top_text = ""
-        bottom_text = ""
-        
-        # Best case: split by |
-        if "|" in response:
-            parts = response.split("|")
-            top_text = parts[0].replace("Top:", "").strip()
-            bottom_text = parts[1].replace("Bottom:", "").strip()
-        else: # Fallback if '|' is not present
-            lines = response.strip().split('\n')
-            if len(lines) > 0:
-                top_text = lines[0].replace("Top:", "").strip()
-            if len(lines) > 1:
-                bottom_text = lines[1].replace("Bottom:", "").strip()
-
-        # Final fallback if parsing fails completely
-        if not top_text or not bottom_text:
-            words = idea.split()
-            mid_point = max(1, len(words) // 2)
-            top_text = ' '.join(words[:mid_point])
-            bottom_text = ' '.join(words[mid_point:])
+        return None, None, error, None
+
+    # List of models to try in order of preference. This makes the app resilient.
+    MODELS_TO_TRY = [
+        "mistralai/Mistral-7B-Instruct-v0.2",
+        "HuggingFaceH4/zephyr-7b-beta",
+        "google/flan-t5-large",
+        "tiiuae/falcon-7b-instruct",
+    ]
+
+    failed_models = []
+    for model_id in MODELS_TO_TRY:
+        try:
+            prompt = f"""Task: Create a funny, two-line meme caption for the idea: "{idea}".
+
+Format your response exactly like this, with no extra text:
+Top: [The first line of the meme]
+Bottom: [The second line, the punchline]"""
+
+            response = client.text_generation(
+                prompt,
+                model=model_id,
+                max_new_tokens=60,
+                temperature=0.8,
+                do_sample=True,
+            )
             
-        return top_text.strip(), bottom_text.strip(), None
-        
-    except Exception as e:
-        error_msg = str(e)
-        if "rate limit" in error_msg:
-            return None, None, "❌ **Rate Limit Exceeded**\n\nWait 60 seconds and try again. Free tier has limits."
-        elif "503" in error_msg:
-            return None, None, f"❌ **Model is Loading**\n\nWait 30 seconds for the model to wake up and try again."
-        elif "404" in error_msg:
-             return None, None, f"❌ **Model Not Found**\n\nThe model '{model}' is not available on the free tier right now. This should be rare with this model."
-        else:
-            return None, None, f"❌ **AI Error:** {error_msg[:200]}"
+            # --- Robust Parsing Logic ---
+            top_text, bottom_text = "",""
+            top_match = re.search(r"Top:\s*(.*)", response, re.IGNORECASE)
+            bottom_match = re.search(r"Bottom:\s*(.*)", response, re.IGNORECASE)
+
+            if top_match: top_text = top_match.group(1).strip().strip('[]"')
+            if bottom_match: bottom_text = bottom_match.group(1).strip().strip('[]"')
 
-def create_meme(idea: str, template: str, model: str):
+            if not top_text or not bottom_text:
+                lines = response.strip().split('\n')
+                if len(lines) >= 1: top_text = lines[0].replace("Top:", "").strip()
+                if len(lines) >= 2: bottom_text = lines[1].replace("Bottom:", "").strip()
+            
+            if not top_text.strip() or not bottom_text.strip():
+                continue # If parsing fails, treat it as a model failure and try the next one
+
+            # SUCCESS!
+            return top_text.strip(), bottom_text.strip(), None, model_id
+
+        except Exception as e:
+            error_msg = str(e).lower()
+            if "404" in error_msg or "503" in error_msg or "is currently loading" in error_msg:
+                # This model is offline or loading, try the next one
+                failed_models.append(model_id.split('/')[-1])
+                continue
+            else:
+                # A different error occurred (like rate limit), so we stop
+                return None, None, f"❌ **AI Error:** {str(e)[:200]}", model_id
+
+    # If the loop finishes, all models failed.
+    return None, None, f"❌ **All AI Models Are Offline**\n\nI'm sorry, all reliable free models are currently unavailable or busy. This is a Hugging Face issue. Please try again in a few minutes.\n\n**Models Tried:** {', '.join(failed_models)}", None
+
+def create_meme(idea: str, template: str):
     """Main function to generate the complete meme."""
     if not idea or len(idea.strip()) < 3:
         return None, "❌ Please enter a meme idea (at least 3 characters)!"
-    
-    top, bottom, error = generate_meme_text(idea, model)
+
+    top, bottom, error, model_used = generate_meme_text(idea)
     if error:
         return None, error
-    
+
     template_id = MEME_TEMPLATES.get(template, "181913649")
     url = "https://api.imgflip.com/caption_image"
     
@@ -93,14 +107,14 @@ def create_meme(idea: str, template: str, model: str):
     }
     
     try:
-        response = requests.post(url, data=payload, timeout=15)
+        response = requests.post(url, data=payload, timeout=20)
         response.raise_for_status()
         data = response.json()
         
         if data.get('success'):
             meme_url = data['data']['url']
             
-            img_response = requests.get(meme_url, timeout=15)
+            img_response = requests.get(meme_url, timeout=20)
             if img_response.status_code == 200:
                 temp_path = f"/tmp/meme_{random.randint(1000, 9999)}.jpg"
                 with open(temp_path, 'wb') as f:
@@ -111,7 +125,7 @@ def create_meme(idea: str, template: str, model: str):
 📝 **Top Text:** {top}
 📝 **Bottom Text:** {bottom}
 
-🤖 **AI Model:** {model}
+🤖 **AI Model Used:** {model_used.split('/')[-1]}
 🔗 **Direct URL:** {meme_url}"""
                 return temp_path, status_message
             else:
@@ -126,11 +140,6 @@ def create_meme(idea: str, template: str, model: str):
 
 # --- CONFIGURATION & UI ---
 
-# ✅ A STABLE, RELIABLE, ALWAYS-ON FREE MODEL
-MODELS = {
-    "FLAN-T5 (Reliable & Fast)": "google/flan-t5-large",
-}
-
 # Popular meme templates
 MEME_TEMPLATES = {
     "Drake": "181913649",
@@ -139,13 +148,14 @@ MEME_TEMPLATES = {
     "Expanding Brain": "93895088",
     "Success Kid": "61544",
     "Batman Slapping Robin": "438680",
+    "Change My Mind": "129242436",
 }
 
 # Example prompts
 examples = [
-    ["When you fix a bug you don't understand", "Success Kid", "FLAN-T5 (Reliable & Fast)"],
-    ["Writing code vs. explaining it in the meeting", "Drake", "FLAN-T5 (Reliable & Fast)"],
-    ["My code on my machine vs. my code in production", "Distracted Boyfriend", "FLAN-T5 (Reliable & Fast)"],
+    ["When you fix a bug you don't understand"],
+    ["Writing code vs. explaining it in the meeting"],
+    ["My code on my machine vs. my code in production"],
 ]
 
 # Gradio UI
@@ -155,7 +165,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI Meme Generator") as demo:
     <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 a Stable & Reliable Free AI Model 🤗</h3>
+        <h3>Auto-Healing AI that ALWAYS Finds a Working Model 🤗</h3>
     </div>
     """)
     
@@ -163,31 +173,29 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI Meme Generator") as demo:
     
     with gr.Row():
         with gr.Column(scale=2):
-            idea_input = gr.Textbox(label="🎨 Your Meme Idea", placeholder="Example: When the CI/CD pipeline finally passes...", lines=2)
-            with gr.Row():
-                template_dropdown = gr.Dropdown(choices=list(MEME_TEMPLATES.keys()), value="Drake", label="🖼️ Meme Template")
-                # Removed the model dropdown since we are forcing one stable model
-                model_display = gr.Textbox(value="FLAN-T5 (Reliable & Fast)", label="🤖 AI Model (Free & Stable)", interactive=False)
+            idea_input = gr.Textbox(label="🎨 Your Meme Idea", placeholder="Example: When the CI/CD pipeline finally passes...", lines=3)
+            template_dropdown = gr.Dropdown(choices=list(MEME_TEMPLATES.keys()), value="Drake", label="🖼️ Meme Template")
             generate_button = gr.Button("🚀 Generate Meme", variant="primary", size="lg")
         
         with gr.Column(scale=1):
             gr.Markdown("""
-            ### 📖 How to Use
-            1.  **Enter Idea:** Describe the meme.
-            2.  **Pick Template:** Choose a format.
-            3.  **Click Generate!**
+            ### 📖 How it Works
+            This app is now **resilient**. It has a list of reliable free models and automatically tries them until it finds one that is online.
+            
+            1.  **Enter Idea & Pick Template.**
+            2.  **Click Generate.**
+            3.  The app does the hard work of finding an active AI model for you!
             
-            **Why only one AI model?**
-            To ensure this app **always works**, it uses a stable, reliable free model (`FLAN-T5`) instead of popular models that are often offline on the free tier.
+            **No more model selection. No more 404 errors.**
             """)
     
     output_image = gr.Image(label="🖼️ Your Generated Meme", type="filepath", show_download_button=True)
     output_status = gr.Textbox(label="📊 Status & Details", lines=6, show_copy_button=True)
     
-    gr.Examples(examples=examples, inputs=[idea_input, template_dropdown, model_display])
+    gr.Examples(examples=examples, inputs=[idea_input])
     
     generate_button.click(
-        fn=lambda idea, template: create_meme(idea, template, MODELS["FLAN-T5 (Reliable & Fast)"]),
+        fn=create_meme,
         inputs=[idea_input, template_dropdown],
         outputs=[output_image, output_status]
     )