Update app.py

app.py

@@ -12,7 +12,6 @@ from diffusers import StableVideoDiffusionPipeline
 from diffusers.utils import export_to_video
 
 # ── ENV SETUP ────────────────────────────────────────────────────────────────
-# Support BOTH HF_TOKEN and HF_KEY (screenshot shows HF_KEY)
 HF_TOKEN = (
     os.environ.get("HF_TOKEN", "")
     or os.environ.get("HF_KEY", "")
@@ -20,9 +19,17 @@ HF_TOKEN = (
 
 if HF_TOKEN:
     login(token=HF_TOKEN)
-    print("✅ HuggingFace login successful.")
+    print(f"✅ HF login OK — token starts: {HF_TOKEN[:8]}...")
 else:
-    print("⚠️  No HF token found (tried HF_TOKEN and HF_KEY).")
+    print("⚠️  No HF token found.")
+
+# ── DEBUG: Print ALL env vars that contain KEY or TOKEN ──────────────────────
+print("\n=== ENV DEBUG (secrets check) ===")
+for k, v in os.environ.items():
+    if any(word in k.upper() for word in ["KEY", "TOKEN", "GEMINI", "API"]):
+        masked = v[:6] + "..." + v[-4:] if len(v) > 10 else "TOO_SHORT"
+        print(f"  {k} = {masked}  (len={len(v)})")
+print("=================================\n")
 
 # ── DEVICE SETUP ─────────────────────────────────────────────────────────────
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
@@ -42,10 +49,24 @@ print("✅  SVD model ready.")
 
 # ── GEMINI HELPER ─────────────────────────────────────────────────────────────
 def call_gemini(pil_image: Image.Image, user_desc: str, language: str, style: str) -> dict:
-    # Read key fresh every call — never cache at startup
-    api_key = os.environ.get("GEMINI_API_KEY", "").strip()
+    # Try every possible env var name user might have used
+    api_key = (
+        os.environ.get("GEMINI_API_KEY", "")
+        or os.environ.get("GEMINI_KEY", "")
+        or os.environ.get("GOOGLE_API_KEY", "")
+        or os.environ.get("API_KEY", "")
+    ).strip()
+
+    print(f"[Gemini] Using key: {api_key[:6]}...{api_key[-4:]} (len={len(api_key)})")
+
     if not api_key:
-        raise ValueError("GEMINI_API_KEY secret is missing. Add it in Space Settings → Secrets.")
+        raise ValueError(
+            "No Gemini API key found! Tried: GEMINI_API_KEY, GEMINI_KEY, "
+            "GOOGLE_API_KEY, API_KEY. Please set one in Space Secrets."
+        )
+
+    if len(api_key) < 20:
+        raise ValueError(f"Key looks too short (len={len(api_key)}). Check for copy-paste errors.")
 
     client = genai.Client(api_key=api_key)
 
@@ -66,7 +87,7 @@ def call_gemini(pil_image: Image.Image, user_desc: str, language: str, style: st
 Language rule : {lang_map.get(language, lang_map['English'])}
 Style rule     : {style_map.get(style, style_map['Fun'])}
 
-CRITICAL: Return ONLY a raw JSON object. No markdown fences. No ```json. No explanation. Pure JSON only.
+CRITICAL: Return ONLY a raw JSON object. No markdown. No ```json. No explanation. Pure JSON only.
 {{
   "hook":         "attention-grabbing opening line (1-2 sentences)",
   "script":       "full 15-20 second voiceover script",
@@ -79,7 +100,7 @@ CRITICAL: Return ONLY a raw JSON object. No markdown fences. No ```json. No expl
     image_bytes = buf.getvalue()
 
     response = client.models.generate_content(
-        model="gemini-2.5-flash",   # ← confirmed available in your API key
+        model="gemini-2.5-flash",
         contents=[
             types.Part.from_bytes(data=image_bytes, mime_type="image/jpeg"),
             types.Part.from_text(text=prompt),
@@ -87,8 +108,6 @@ CRITICAL: Return ONLY a raw JSON object. No markdown fences. No ```json. No expl
     )
 
     raw = response.text.strip()
-
-    # Strip markdown fences if present
     if "```" in raw:
         raw = raw.split("```")[1]
         if raw.lower().startswith("json"):
@@ -101,7 +120,6 @@ CRITICAL: Return ONLY a raw JSON object. No markdown fences. No ```json. No expl
 # ── VIDEO GENERATION ──────────────────────────────────────────────────────────
 def generate_video(pil_image: Image.Image) -> str:
     img = pil_image.convert("RGB").resize((1024, 576))
-
     frames = svd_pipe(
         image=img,
         num_frames=14,
@@ -109,7 +127,6 @@ def generate_video(pil_image: Image.Image) -> str:
         decode_chunk_size=4,
         generator=torch.manual_seed(42),
     ).frames[0]
-
     tmp = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False)
     export_to_video(frames, tmp.name, fps=7)
     return tmp.name
@@ -122,7 +139,6 @@ def generate_ad(image, user_desc, language, style):
 
     pil_image = image if isinstance(image, Image.Image) else Image.fromarray(image)
 
-    # STEP 1 — Gemini ad copy
     try:
         ad_data = call_gemini(pil_image, user_desc or "", language, style)
     except Exception as e:
@@ -132,7 +148,6 @@ def generate_ad(image, user_desc, language, style):
     script = ad_data.get("script", "")
     cta    = ad_data.get("cta", "")
 
-    # STEP 2 — Video generation
     try:
         video_path = generate_video(pil_image)
     except Exception as e:
@@ -154,10 +169,8 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="violet")) as demo:
 
     with gr.Row():
         with gr.Column(scale=1):
-            image_input = gr.Image(
-                label="📸 Upload Product Image", type="pil", height=300
-            )
-            desc_input = gr.Textbox(
+            image_input = gr.Image(label="📸 Upload Product Image", type="pil", height=300)
+            desc_input  = gr.Textbox(
                 label="📝 Describe your product (optional)",
                 placeholder="e.g. Organic honey sourced from Himalayan farms …",
                 lines=3,
@@ -187,8 +200,8 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="violet")) as demo:
 
     gr.Markdown(
         "---\n**How it works:** "
-        "1️⃣ Gemini 1.5 Flash reads your image → hook, script, CTA. "
-        "2️⃣ Stable Video Diffusion → short cinematic video. "
+        "1️⃣ Gemini 2.5 Flash reads your image → hook, script, CTA. "
+        "2️⃣ Stable Video Diffusion → short cinematic video clip. "
         "3️⃣ Ready-to-post reel! 🎉"
     )