Update a.py

a.py

@@ -1,66 +1,195 @@
-import gradio as gr
-import requests
 import os
+import time
+import gradio as gr
 from byteplussdkarkruntime import Ark
+import requests
 
-API_KEY = os.environ.get("Key", "")
+# Get API key from Hugging Face secret "Key" and CLEAN it
+API_KEY = os.environ.get("Key", "").strip()  # The .strip() fixes the newline issue!
+print(f"✅ Key loaded, length: {len(API_KEY)}")
 
-def test_both(prompt_text):
-    results = []
-    
-    # TEST 1: Direct BytePlus URL
+# Initialize client with proxy
+client = Ark(
+    base_url="https://1hit.no/proxy/proxy.php",  # Using proxy
+    api_key=API_KEY,
+    timeout=30.0,
+    max_retries=3,
+)
+
+# Fresh new prompts
+DEFAULT_PROMPTS = {
+    "Cinematic Nature": "Majestic drone shot soaring above misty mountains at golden hour, sunlight breaking through clouds, cinematic 4k quality, smooth motion --duration 5 --camerafixed false",
+    "Urban Exploration": "Dynamic drone flight through narrow alleyways of a neon-lit Tokyo street at night, rain-slicked surfaces reflecting lights, cyberpunk aesthetic --duration 5 --camerafixed false",
+    "Action Sequence": "High-speed drone chasing a sports car through a winding coastal road, dramatic cliffside views, action movie style --duration 5 --camerafixed false",
+    "Abstract Art": "Surreal drone flight through floating geometric shapes in a dreamlike void, pastel colors, ethereal atmosphere --duration 5 --camerafixed false",
+    "Wildlife Documentary": "Drone following a herd of elephants across the African savanna at sunset, National Geographic style, majestic wildlife cinematography --duration 5 --camerafixed false",
+    "Sports Highlight": "Drone racing alongside professional skiers carving through fresh powder in the Alps, high-energy sports broadcast style --duration 5 --camerafixed false",
+    "Beach Paradise": "Drone gliding over turquoise waters and white sand beaches, palm trees swaying, tropical paradise aerial view --duration 5 --camerafixed false",
+    "Ancient Temple": "Drone circling around ancient Angkor Wat temple at dawn, mystical atmosphere, historical documentary style --duration 5 --camerafixed false"
+}
+
+def poll_via_json(task_id):
+    """Fallback method: poll io.json for results"""
+    json_url = "https://1hit.no/proxy/io.json"
     try:
-        client_direct = Ark(
-            base_url="https://ark.ap-southeast.bytepluses.com/api/v3",
-            api_key=API_KEY,
-            timeout=10
-        )
-        result_direct = client_direct.content_generation.tasks.create(
-            model="seedance-1-5-pro-251215",
-            content=[{"type": "text", "text": prompt_text}]
-        )
-        results.append(f"✅ DIRECT: Task created {result_direct.id}")
-    except Exception as e:
-        results.append(f"❌ DIRECT: {str(e)[:100]}")
+        response = requests.get(json_url)
+        data = response.json()
+        if task_id in data:
+            task_data = data[task_id]
+            if task_data.get('status') == 'succeeded':
+                return task_data.get('video_url')
+            elif task_data.get('status') == 'failed':
+                return None
+    except:
+        pass
+    return None
+
+def generate_video(image_path, prompt_text, progress=gr.Progress()):
+    """Generate video with all features"""
     
-    # TEST 2: Proxy URL
-    try:
-        client_proxy = Ark(
-            base_url="https://1hit.no/proxy/proxy.php",
-            api_key=API_KEY,
-            timeout=10
-        )
-        result_proxy = client_proxy.content_generation.tasks.create(
-            model="seedance-1-5-pro-251215",
-            content=[{"type": "text", "text": prompt_text}]
-        )
-        results.append(f"✅ PROXY: Task created {result_proxy.id}")
-    except Exception as e:
-        results.append(f"❌ PROXY: {str(e)[:100]}")
+    if not API_KEY:
+        yield "❌ API Key not configured. Please add 'Key' secret.", None
+        return
+    
+    if image_path is None:
+        yield "⚠️ Please upload an image first", None
+        return
     
-    # TEST 3: Direct requests (no SDK)
     try:
-        url = "https://ark.ap-southeast.bytepluses.com/api/v3/contents/generations/tasks"
-        headers = {
-            "Content-Type": "application/json",
-            "Authorization": f"Bearer {API_KEY}"
-        }
-        data = {
-            "model": "seedance-1-5-pro-251215",
-            "content": [{"type": "text", "text": prompt_text}]
-        }
-        r = requests.post(url, headers=headers, json=data, timeout=10)
-        results.append(f"✅ REQUESTS: Status {r.status_code}")
+        progress(0, desc="Preparing image...")
+        
+        # Get HF temp URL
+        image_url = f"/file={image_path}"
+        print(f"Using image URL: {image_url}")
+        
+        yield "✅ Image ready! Creating video request...", None
+        
+        progress(0.2, desc="Creating request...")
+        
+        # Create task
+        try:
+            create_result = client.content_generation.tasks.create(
+                model="seedance-1-5-pro-251215",
+                content=[
+                    {"type": "text", "text": prompt_text},
+                    {"type": "image_url", "image_url": {"url": image_url}}
+                ]
+            )
+        except Exception as e:
+            yield f"❌ API Error: {str(e)}", None
+            return
+        
+        task_id = create_result.id
+        print(f"Task created: {task_id}")
+        yield f"✅ Task created: {task_id}", None
+        
+        progress(0.3, desc="Polling for results...")
+        
+        # Poll for results
+        attempts = 0
+        max_attempts = 120
+        used_fallback = False
+        
+        while attempts < max_attempts:
+            try:
+                get_result = client.content_generation.tasks.get(task_id=task_id)
+                status = get_result.status
+                
+                if status == "succeeded":
+                    progress(1.0, desc="Complete!")
+                    video_url = get_result.content.video_url if hasattr(get_result, 'content') else None
+                    yield "✅ Video generated successfully!", video_url
+                    return
+                    
+                elif status == "failed":
+                    error_msg = get_result.error if hasattr(get_result, 'error') else "Unknown error"
+                    yield f"❌ Failed: {error_msg}", None
+                    return
+                else:
+                    progress(0.3 + (attempts/max_attempts)*0.7, desc=f"Status: {status}")
+                    yield f"⏳ Status: {status}... (attempt {attempts + 1})", None
+                    time.sleep(1)
+                    attempts += 1
+                    
+            except Exception as e:
+                if not used_fallback:
+                    print("SDK polling failed, trying JSON fallback...")
+                    used_fallback = True
+                
+                video_url = poll_via_json(task_id)
+                if video_url:
+                    yield "✅ Video generated successfully! (via JSON)", video_url
+                    return
+                elif video_url is None:
+                    yield "❌ Task failed (via JSON)", None
+                    return
+                else:
+                    yield f"⏳ Status: processing... (JSON fallback)", None
+                    time.sleep(5)
+                    attempts += 1
+        
+        yield "⏰ Timeout after 2 minutes", None
+        
     except Exception as e:
-        results.append(f"❌ REQUESTS: {str(e)[:100]}")
-    
-    return "\n\n".join(results)
+        yield f"❌ Error: {str(e)}", None
 
-with gr.Blocks() as demo:
-    gr.Markdown("# BytePlus Connection Test")
-    prompt = gr.Textbox("test", label="Prompt")
-    btn = gr.Button("Test All")
-    output = gr.Textbox(label="Results", lines=10)
-    btn.click(fn=test_both, inputs=prompt, outputs=output)
+def update_prompt(choice):
+    return DEFAULT_PROMPTS.get(choice, "")
+
+# Interface
+with gr.Blocks(title="BytePlus Video Generator", theme=gr.themes.Soft()) as demo:
+    
+    gr.Markdown("# 🎥 BytePlus Video Generator")
+    
+    with gr.Row():
+        if API_KEY:
+            gr.Markdown("✅ **API Key:** Configured")
+        else:
+            gr.Markdown("❌ **API Key:** Not configured - please add 'Key' secret")
+    
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(
+                label="Upload Starting Image",
+                type="filepath",
+                height=300
+            )
+            
+            shot_type = gr.Dropdown(
+                choices=list(DEFAULT_PROMPTS.keys()),
+                label="🎬 Shot Type",
+                value="Cinematic Nature"
+            )
+            
+            prompt = gr.Textbox(
+                label="📝 Custom Prompt",
+                lines=3,
+                value=DEFAULT_PROMPTS["Cinematic Nature"]
+            )
+            
+            shot_type.change(fn=update_prompt, inputs=shot_type, outputs=prompt)
+            
+            generate_btn = gr.Button("🚀 Generate Video", variant="primary", size="lg")
+        
+        with gr.Column():
+            status = gr.Textbox(label="Status", lines=5)
+            video_output = gr.Video(label="Generated Video")
+    
+    # Quick shot buttons
+    gr.Markdown("### Quick Shot Select")
+    with gr.Row():
+        gr.Button("🏔️ Nature").click(fn=lambda: "Cinematic Nature", outputs=shot_type)
+        gr.Button("🌃 City").click(fn=lambda: "Urban Exploration", outputs=shot_type)
+        gr.Button("🏎️ Action").click(fn=lambda: "Action Sequence", outputs=shot_type)
+        gr.Button("🎨 Abstract").click(fn=lambda: "Abstract Art", outputs=shot_type)
+        gr.Button("🦁 Wildlife").click(fn=lambda: "Wildlife Documentary", outputs=shot_type)
+        gr.Button("⛷️ Sports").click(fn=lambda: "Sports Highlight", outputs=shot_type)
+    
+    generate_btn.click(
+        fn=generate_video,
+        inputs=[image_input, prompt],
+        outputs=[status, video_output]
+    )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0")