"""
InfiniteTalk - Powered by WaveSpeed AI
Auto-generated by Space Generator
"""
import gradio as gr
import requests
import time
from typing import Dict, Any
# ============ API Configuration ============
WAVESPEED_API_BASE = "https://api.wavespeed.ai/api/v3"
UPLOAD_ENDPOINT = "https://api.wavespeed.ai/api/v3/media/upload/binary"
MODEL_ENDPOINT = "wavespeed-ai/infinitetalk"
POLL_INTERVAL = 1.5
POLL_MAX_SECONDS = 300
# ============ CSS ============
CUSTOM_CSS = """
/* ===== Base Styles ===== */
html, body, .gradio-container {
background: linear-gradient(145deg, #f5f3ff 0%, #ede9fe 50%, #e0e7ff 100%) !important;
font-family: 'Inter', -apple-system, BlinkMacSystemFont, sans-serif !important;
}
.gradio-container {
max-width: 960px !important;
margin: 0 auto !important;
padding: 24px !important;
}
/* ===== Hero Section ===== */
.hero-container {
text-align: center;
padding: 48px 20px 36px;
}
.hero-badge {
display: inline-block;
background: linear-gradient(135deg, #8b5cf6, #7c3aed);
padding: 10px 24px;
border-radius: 50px;
font-size: 0.7rem;
color: #fff;
font-weight: 700;
letter-spacing: 1.5px;
margin-bottom: 20px;
box-shadow: 0 4px 20px rgba(139, 92, 246, 0.35);
}
.hero-title {
font-size: 3rem;
font-weight: 800;
margin: 0 0 16px 0;
background: linear-gradient(135deg, #6d28d9 0%, #8b5cf6 50%, #a78bfa 100%);
-webkit-background-clip: text;
-webkit-text-fill-color: transparent;
background-clip: text;
letter-spacing: -0.5px;
}
.hero-desc {
font-size: 1.05rem;
color: #64748b;
max-width: 100%;
margin: 0 auto 20px;
line-height: 1.6;
}
.hero-badges {
display: flex;
gap: 28px;
justify-content: center;
flex-wrap: wrap;
}
.hero-badges span {
display: flex;
align-items: center;
gap: 8px;
color: #475569;
font-size: 0.9rem;
font-weight: 600;
}
/* ===== Hero Image ===== */
.hero-image {
max-width: 100%;
max-height: 300px;
border-radius: 16px;
margin: 24px auto;
box-shadow: 0 8px 32px rgba(139, 92, 246, 0.15);
}
/* ===== Main Card ===== */
.main-card {
background: #ffffff;
border: 1px solid rgba(139, 92, 246, 0.1);
border-radius: 20px;
padding: 28px;
margin-bottom: 20px;
box-shadow: 0 4px 24px rgba(139, 92, 246, 0.08), 0 1px 3px rgba(0, 0, 0, 0.04);
transition: box-shadow 0.3s ease;
}
.main-card:hover {
box-shadow: 0 8px 32px rgba(139, 92, 246, 0.12), 0 2px 6px rgba(0, 0, 0, 0.04);
}
/* ===== API Key Section ===== */
.api-key-row {
display: flex;
align-items: center;
justify-content: space-between;
margin-bottom: 12px;
}
.api-key-label {
display: flex;
align-items: center;
gap: 10px;
color: #1e293b;
font-weight: 700;
font-size: 1rem;
}
.get-key-btn {
padding: 10px 20px;
background: linear-gradient(135deg, #8b5cf6, #7c3aed);
border: none;
border-radius: 10px;
color: #fff !important;
text-decoration: none;
font-weight: 600;
font-size: 0.85rem;
transition: all 0.25s ease;
box-shadow: 0 4px 12px rgba(139, 92, 246, 0.3);
}
.get-key-btn:hover {
transform: translateY(-2px);
box-shadow: 0 6px 20px rgba(139, 92, 246, 0.4);
color: #fff;
}
/* ===== Section Title ===== */
.section-title {
color: #1e293b;
font-weight: 700;
font-size: 1rem;
display: flex;
align-items: center;
gap: 10px;
margin-bottom: 16px;
}
/* ===== Upload Area ===== */
.upload-area {
border: 2px dashed rgba(139, 92, 246, 0.3) !important;
border-radius: 16px !important;
background: linear-gradient(145deg, #faf5ff 0%, #f5f3ff 100%) !important;
transition: all 0.3s ease !important;
min-height: 220px !important;
}
.upload-area:hover {
border-color: rgba(139, 92, 246, 0.5) !important;
background: linear-gradient(145deg, #f5f3ff 0%, #ede9fe 100%) !important;
}
/* ===== Result Area ===== */
.result-area {
border: 2px solid rgba(139, 92, 246, 0.15) !important;
border-radius: 16px !important;
background: #fafafa !important;
min-height: 220px !important;
}
/* ===== Button Styling ===== */
.primary-btn {
width: 100%;
margin-top: 20px !important;
background: linear-gradient(135deg, #8b5cf6, #7c3aed) !important;
border: none !important;
color: #fff !important;
font-weight: 700 !important;
font-size: 1rem !important;
padding: 14px 28px !important;
border-radius: 12px !important;
box-shadow: 0 4px 16px rgba(139, 92, 246, 0.35) !important;
transition: all 0.25s ease !important;
cursor: pointer !important;
}
.primary-btn:hover {
transform: translateY(-2px) !important;
box-shadow: 0 8px 24px rgba(139, 92, 246, 0.45) !important;
}
/* ===== CTA Section ===== */
.cta-container {
text-align: center;
padding: 44px 32px;
background: linear-gradient(135deg, #8b5cf6 0%, #7c3aed 50%, #6d28d9 100%);
border-radius: 20px;
margin-top: 8px;
box-shadow: 0 8px 32px rgba(139, 92, 246, 0.35);
position: relative;
overflow: hidden;
}
.cta-container::before {
content: '';
position: absolute;
top: -50%;
right: -50%;
width: 100%;
height: 100%;
background: radial-gradient(circle, rgba(255,255,255,0.1) 0%, transparent 60%);
pointer-events: none;
}
.cta-title {
color: #fff;
font-size: 1.5rem;
font-weight: 800;
margin: 0 0 8px 0;
position: relative;
}
.cta-desc {
color: rgba(255, 255, 255, 0.9);
font-size: 1rem;
margin: 0 0 24px 0;
position: relative;
}
.cta-btn {
display: inline-block;
padding: 14px 36px;
background: #fff;
border-radius: 12px;
color: #7c3aed !important;
text-decoration: none;
font-weight: 700;
font-size: 1rem;
transition: all 0.25s ease;
box-shadow: 0 4px 16px rgba(0, 0, 0, 0.1);
position: relative;
}
.cta-btn:hover {
transform: translateY(-3px);
box-shadow: 0 8px 24px rgba(0, 0, 0, 0.15);
color: #6d28d9 !important;
}
/* ===== Hide Elements ===== */
footer { display: none !important; }
/* ===== Input Styling ===== */
.gradio-container input[type="password"],
.gradio-container input[type="text"] {
border: 2px solid #e2e8f0 !important;
border-radius: 12px !important;
padding: 14px 16px !important;
font-size: 0.95rem !important;
transition: all 0.2s ease !important;
}
.gradio-container input[type="password"]:focus,
.gradio-container input[type="text"]:focus {
border-color: #8b5cf6 !important;
box-shadow: 0 0 0 3px rgba(139, 92, 246, 0.15) !important;
outline: none !important;
}
"""
# ============ API Functions ============
def upload_image(api_key: str, file_path: str) -> str:
headers = {"Authorization": f"Bearer {api_key.strip()}"}
with open(file_path, "rb") as f:
resp = requests.post(UPLOAD_ENDPOINT, headers=headers, files={"file": f}, timeout=60)
if resp.status_code == 401:
raise Exception("Invalid API Key")
elif resp.status_code >= 400:
raise Exception(f"Upload failed: {resp.status_code}")
data = resp.json()
if data.get("code") != 200:
raise Exception(data.get("message", "Upload failed"))
return data.get("data", {}).get("download_url")
def call_api(api_key: str, endpoint: str, payload: Dict[str, Any]) -> Dict[str, Any]:
headers = {"Authorization": f"Bearer {api_key.strip()}", "Content-Type": "application/json"}
resp = requests.post(f"{WAVESPEED_API_BASE}/{endpoint}", json=payload, headers=headers, timeout=30)
if resp.status_code == 401:
raise Exception("Invalid API Key")
elif resp.status_code == 429:
raise Exception("Quota exceeded")
elif resp.status_code >= 400:
raise Exception(f"API error: {resp.status_code}")
data = resp.json()
if data.get("code") != 200:
raise Exception(data.get("message", "Unknown error"))
return data.get("data", {})
def poll_result(api_key: str, request_id: str) -> Dict[str, Any]:
headers = {"Authorization": f"Bearer {api_key.strip()}"}
url = f"{WAVESPEED_API_BASE}/predictions/{request_id}/result"
start_time = time.time()
while time.time() - start_time < POLL_MAX_SECONDS:
resp = requests.get(url, headers=headers, timeout=30)
if resp.status_code >= 400:
raise Exception("Failed to get result")
result = resp.json().get("data", {})
status = result.get("status", "")
if status == "completed":
return result
elif status == "failed":
raise Exception("Generation failed")
time.sleep(POLL_INTERVAL)
raise Exception("Timeout")
def download_video(url: str) -> str:
"""下载视频到临时文件，返回文件路径"""
import tempfile
import os
try:
resp = requests.get(url, timeout=120, stream=True)
if resp.status_code != 200:
return None
# 从 URL 或 Content-Type 确定扩展名
ext = ".mp4"
if "webm" in url.lower() or "webm" in resp.headers.get("content-type", ""):
ext = ".webm"
# 创建临时文件
fd, path = tempfile.mkstemp(suffix=ext)
with os.fdopen(fd, 'wb') as f:
for chunk in resp.iter_content(chunk_size=8192):
f.write(chunk)
return path
except Exception as e:
print(f"Download error: {e}")
return None
def process(api_key: str, image_path: str, audio_path: str):
if not api_key or not api_key.strip():
gr.Warning("Please enter your API Key")
return None
if not image_path:
gr.Warning("Please upload an image")
return None
if not audio_path:
gr.Warning("Please upload an audio file")
return None
try:
gr.Info("Uploading image...")
image_url = upload_image(api_key, image_path)
gr.Info("Uploading audio...")
audio_url = upload_image(api_key, audio_path) # 同样用 upload 接口
gr.Info("Generating video...")
payload = {
"image": image_url,
"audio": audio_url,
"resolution": "480p",
"seed": -1,
}
result = call_api(api_key, MODEL_ENDPOINT, payload)
request_id = result.get("id")
if not request_id:
gr.Warning("Failed to start")
return None
final_result = poll_result(api_key, request_id)
outputs = final_result.get("outputs", [])
if outputs:
gr.Info("Downloading video...")
video_path = download_video(outputs[0])
if video_path:
gr.Info("Done!")
return video_path
return None
except Exception as e:
error_msg = str(e).lower()
if "invalid" in error_msg or "401" in error_msg:
gr.Warning("Invalid API Key")
elif "quota" in error_msg or "429" in error_msg:
gr.Warning("Quota exceeded")
elif "timeout" in error_msg:
gr.Warning("Timeout - please try again")
else:
gr.Warning(str(e))
return None
# ============ Gradio UI ============
with gr.Blocks(css=CUSTOM_CSS, title="InfiniteTalk - WaveSpeed") as demo:
# Hero Section
gr.HTML("""
WAVESPEED AI
InfiniteTalk
Generate talking videos from image and audio. Try on wavespeed
""")
# API Key Card
with gr.Group(elem_classes="main-card"):
gr.HTML("""
""")
api_key_input = gr.Textbox(
placeholder="Enter your WaveSpeed API key",
type="password",
show_label=False
)
# Main Content Card
with gr.Group(elem_classes="main-card"):
with gr.Row():
# Left Column - Input
with gr.Column(scale=1):
gr.HTML("""
""")
image_input = gr.Image(
label="Upload Image",
type="filepath",
source="upload",
elem_classes=["upload-area"]
)
audio_input = gr.Audio(
label="Upload Audio",
type="filepath"
)
submit_btn = gr.Button("Process", variant="primary", elem_classes="primary-btn")
# Right Column - Output
with gr.Column(scale=1):
gr.HTML("""
""")
output_video = gr.Video(
label="",
interactive=False
)
# CTA Section
gr.HTML("""
""")
# Event binding
submit_btn.click(
fn=process,
inputs=[api_key_input, image_input, audio_input],
outputs=output_video,
)
if __name__ == "__main__":
demo.launch(server_name="0.0.0.0", server_port=7860)