Upload app.py

app.py

@@ -7,7 +7,7 @@ Features:
   - Translates + narrates into 36 languages (preset or cloned voice)
   - Generates AI video per scene via HappyHorse 1.0
   - Three video modes: text-to-video, image-to-video, auto scene prompts
-  - Three HappyHorse API providers: fal.ai, happyhorse.app, DashScope
+  - Two HappyHorse API providers: happyhorse.app, DashScope (Wan 2.1 fallback)
   - Combines narrated audio + video scenes into final MP4
 
 Deploy as a Hugging Face Space:
@@ -417,23 +417,37 @@ def narrate_scene_cloned(client, text, voice_id, language, lang_config, translat
 # ==========================================
 # HAPPYHORSE VIDEO GENERATION
 # ==========================================
+def sanitize_prompt(prompt, max_length=2000):
+    """Clean and truncate prompt for HappyHorse API."""
+    # Replace smart quotes and special chars
+    prompt = prompt.replace("\u2018", "'").replace("\u2019", "'")
+    prompt = prompt.replace("\u201c", '"').replace("\u201d", '"')
+    prompt = prompt.replace("\u2014", " -- ").replace("\u2013", " - ")
+    # Remove any control characters
+    prompt = re.sub(r'[\x00-\x1f\x7f-\x9f]', ' ', prompt)
+    # Collapse multiple spaces
+    prompt = re.sub(r'\s+', ' ', prompt).strip()
+    # Truncate
+    if len(prompt) > max_length:
+        prompt = prompt[:max_length].rsplit(' ', 1)[0]
+    return prompt
+
+
 def generate_video_happyhorse_app(prompt, api_key, duration=5, aspect="16:9", image_url=None):
     """Generate video via happyhorse.app."""
     headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
 
-    mode = "text-to-video"
-    if image_url and not image_url.startswith("data:"):
-        mode = "image-to-video"
+    clean_prompt = sanitize_prompt(prompt, max_length=2000)
 
     payload = {
         "model": "happyhorse-1.0/video",
-        "prompt": prompt,
-        "mode": mode,
-        "duration": duration,
+        "prompt": clean_prompt,
+        "mode": "pro",
+        "duration": int(duration),
         "aspect_ratio": aspect,
         "sound": False,
     }
-    if mode == "image-to-video":
+    if image_url and not image_url.startswith("data:"):
         payload["image_urls"] = [image_url]
 
     generate_url = HAPPYHORSE_PROVIDERS["happyhorse.app"]["generate"]
@@ -509,19 +523,32 @@ def generate_video_happyhorse_app(prompt, api_key, duration=5, aspect="16:9", im
 
 
 def generate_video_dashscope(prompt, api_key, duration=5, aspect="16:9", image_url=None):
-    """Generate video via DashScope Bailian (async task API)."""
+    """Generate video via DashScope Bailian (async task API).
+    Uses wan2.1-t2v-plus since happyhorse-1.0 is not yet available on DashScope.
+    """
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json",
         "X-DashScope-Async": "enable",
     }
-    payload = {
-        "model": "happyhorse-1.0",
-        "input": {"prompt": prompt},
-        "parameters": {"duration": duration, "aspect_ratio": aspect},
-    }
+
+    clean_prompt = sanitize_prompt(prompt, max_length=2000)
+
+    # Use Wan 2.1 T2V Plus (available on DashScope) as HappyHorse is not live yet
     if image_url and not image_url.startswith("data:"):
-        payload["input"]["image_url"] = image_url
+        model = "wan2.1-i2v-plus"
+        payload = {
+            "model": model,
+            "input": {"prompt": clean_prompt, "img_url": image_url},
+            "parameters": {"resolution": "1280*720"},
+        }
+    else:
+        model = "wan2.1-t2v-plus"
+        payload = {
+            "model": model,
+            "input": {"prompt": clean_prompt},
+            "parameters": {"size": "1280*720"},
+        }
 
     generate_url = HAPPYHORSE_PROVIDERS["DashScope (Bailian)"]["generate"]
     print(f"[DashScope] Submitting to {generate_url}")
@@ -930,7 +957,6 @@ def generate_wrapper(text_input, file_input, lang, voice_mode, preset_voice,
 
 with gr.Blocks(
     title="Visual Storybook Generator",
-    theme=gr.themes.Soft(primary_hue="amber", secondary_hue="orange", neutral_hue="stone"),
 ) as demo:
 
     gr.Markdown(DESCRIPTION)
@@ -956,7 +982,7 @@ with gr.Blocks(
             with gr.Tab("Video"):
                 video_provider = gr.Dropdown(
                     choices=list(HAPPYHORSE_PROVIDERS.keys()),
-                    value="fal.ai",
+                    value="happyhorse.app",
                     label="HappyHorse API Provider",
                     info="Each provider needs its own API key in Secrets",
                 )
@@ -980,6 +1006,7 @@ with gr.Blocks(
                     video_duration = gr.Slider(minimum=3, maximum=10, value=5, step=1, label="Video Duration (sec/scene)")
 
             generate_btn = gr.Button("Generate Visual Storybook", variant="primary", size="lg")
+            test_btn = gr.Button("Test Video API Connection", variant="secondary", size="sm")
 
         # -- RIGHT: Output --
         with gr.Column(scale=1):
@@ -987,12 +1014,94 @@ with gr.Blocks(
             stats_output = gr.Markdown(label="Stats")
             with gr.Accordion("Scene Transcripts", open=False):
                 transcript_output = gr.Markdown()
+            test_output = gr.Markdown(label="API Test Result", visible=True)
 
     # Events
     sample_btn.click(fn=lambda: SAMPLE_TEXT, outputs=text_input)
     voice_mode.change(fn=on_voice_mode_change, inputs=voice_mode, outputs=[preset_voice, clone_audio])
     video_mode.change(fn=on_video_mode_change, inputs=video_mode, outputs=[scene_images])
 
+    def test_video_api(provider):
+        """Test the video API connection with a simple request."""
+        config = HAPPYHORSE_PROVIDERS[provider]
+        api_key = os.environ.get(config["key_env"], "")
+
+        if not api_key:
+            return f"**FAILED:** `{config['key_env']}` is not set in Secrets."
+
+        results = []
+        results.append(f"**Provider:** {provider}")
+        results.append(f"**API Key:** `{api_key[:8]}...{api_key[-4:]}`")
+
+        # Step 1: Submit a short test generation
+        if provider == "happyhorse.app":
+            headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
+            payload = {
+                "model": "happyhorse-1.0/video",
+                "prompt": "A golden sunset over calm ocean waves",
+                "mode": "std",
+                "duration": 3,
+                "aspect_ratio": "16:9",
+                "sound": False,
+            }
+            url = config["generate"]
+        else:
+            headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json", "X-DashScope-Async": "enable"}
+            payload = {
+                "model": "wan2.1-t2v-plus",
+                "input": {"prompt": "A golden sunset over calm ocean waves"},
+                "parameters": {"size": "1280*720"},
+            }
+            url = config["generate"]
+
+        results.append(f"**URL:** `{url}`")
+        results.append(f"**Payload:** `{json.dumps(payload)}`")
+
+        try:
+            resp = http_requests.post(url, json=payload, headers=headers, timeout=30)
+            results.append(f"**HTTP Status:** {resp.status_code}")
+            results.append(f"**Response:** ```{resp.text[:500]}```")
+
+            if resp.status_code == 200:
+                data = resp.json()
+                task_id = (
+                    data.get("data", {}).get("task_id")
+                    or data.get("task_id")
+                    or data.get("output", {}).get("task_id")
+                )
+                if task_id:
+                    results.append(f"**Task ID:** `{task_id}` -- API is working! Task submitted successfully.")
+
+                    # Quick poll once
+                    time.sleep(5)
+                    if provider == "happyhorse.app":
+                        s_resp = http_requests.get(
+                            f"{config['status']}?task_id={task_id}",
+                            headers=headers, timeout=15,
+                        )
+                    else:
+                        s_resp = http_requests.get(
+                            f"{config['status']}/{task_id}",
+                            headers={"Authorization": f"Bearer {api_key}"}, timeout=15,
+                        )
+                    results.append(f"**Status poll:** {s_resp.status_code}")
+                    results.append(f"**Status body:** ```{s_resp.text[:500]}```")
+                else:
+                    results.append(f"**WARNING:** 200 OK but no task_id found in response")
+            elif resp.status_code == 401:
+                results.append("**ERROR:** Invalid API key (401 Unauthorized)")
+            elif resp.status_code == 402:
+                results.append("**ERROR:** Insufficient credits (402 Payment Required)")
+            else:
+                results.append(f"**ERROR:** Unexpected status {resp.status_code}")
+
+        except Exception as e:
+            results.append(f"**EXCEPTION:** {str(e)}")
+
+        return "\n\n".join(results)
+
+    test_btn.click(fn=test_video_api, inputs=[video_provider], outputs=[test_output])
+
     generate_btn.click(
         fn=generate_wrapper,
         inputs=[text_input, file_input, target_lang, voice_mode, preset_voice,
@@ -1012,4 +1121,4 @@ with gr.Blocks(
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(theme=gr.themes.Soft(primary_hue="amber", secondary_hue="orange", neutral_hue="stone"))