Update ups.py

ups.py

@@ -18,16 +18,6 @@ client = Ark(
 )
 
 # 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):
     """Check io.json for task status"""
@@ -48,101 +38,6 @@ def poll_via_json(task_id):
         pass
     return None
 
-def generate_video(image_path, manual_url, prompt_text, progress=gr.Progress()):
-    """Generate video with proper polling and URL override"""
-    
-    if not API_KEY:
-        yield "❌ API Key not configured. Please add 'Key' secret.", None
-        return
-    
-    # Determine which URL to use
-    if manual_url and manual_url.strip():
-        image_url = manual_url.strip()
-        source = "manual URL"
-    elif image_path:
-        image_url = f"/file={image_path}"
-        source = "uploaded image"
-    else:
-        yield "⚠️ Please upload an image or enter a URL", None
-        return
-    
-    try:
-        progress(0, desc="Preparing image...")
-        print(f"Using {source}: {image_url}")
-        yield f"✅ Using {source}...", 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 - TRY SDK FIRST, then fallback to JSON
-        attempts = 0
-        max_attempts = 120
-        used_fallback = False
-        
-        while attempts < max_attempts:
-            try:
-                # Try SDK method first
-                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(2)
-                    attempts += 1
-                    
-            except Exception as e:
-                # If SDK fails, use JSON fallback
-                if not used_fallback:
-                    print("SDK polling failed, using JSON fallback...")
-                    used_fallback = True
-                
-                # Poll JSON every 5 seconds
-                result = poll_via_json(task_id)
-                if result == "FAILED":
-                    yield "❌ Task failed (via JSON)", None
-                    return
-                elif result and result != "PROCESSING":
-                    yield "✅ Video generated successfully! (via JSON)", result
-                    return
-                else:
-                    progress(0.3 + (attempts/max_attempts)*0.7, desc="Status: processing (JSON)")
-                    yield f"⏳ Status: processing... (JSON fallback, attempt {attempts + 1})", None
-                    time.sleep(5)
-                    attempts += 1
-        
-        yield "⏰ Timeout after 2 minutes", None
-        
-    except Exception as e:
-        yield f"❌ Error: {str(e)}", None
 
 def update_prompt(choice):
     return DEFAULT_PROMPTS.get(choice, "")
@@ -278,87 +173,9 @@ def update_url_display(image_path, manual_url):
         return "No image selected"
 
 # Create the interface
-with gr.Blocks(title="BytePlus Video Generator", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="BytePlus Video Gallery", 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")
-    
-    # Create tabs
-    with gr.Tabs():
-        # Tab 1: Generate Video (UPDATED with URL display)
-        with gr.TabItem("🎬 Generate Video"):
-            with gr.Row():
-                with gr.Column():
-                    image_input = gr.Image(
-                        label="Upload Starting Image",
-                        type="filepath",
-                        height=300
-                    )
-                    
-                    # URL override section
-                    with gr.Accordion("🔗 Image URL Override", open=False):
-                        image_url_input = gr.Textbox(
-                            label="Enter image URL directly",
-                            placeholder="https://example.com/image.jpg",
-                            info="Leave empty to use uploaded image"
-                        )
-                        gr.Markdown("💡 *Using a URL will override the uploaded image*")
-                    
-                    # Current URL display
-                    with gr.Row():
-                        current_url_display = gr.Textbox(
-                            label="Current Image URL",
-                            value="No image selected",
-                            interactive=False,
-                            lines=2
-                        )
-                    
-                    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=6)
-                    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)
-            
-            # Update URL display when inputs change
-            image_input.change(
-                fn=update_url_display,
-                inputs=[image_input, image_url_input],
-                outputs=current_url_display
-            )
-            
-            image_url_input.change(
-                fn=update_url_display,
-                inputs=[image_input, image_url_input],
-                outputs=current_url_display
             )
         
         # Tab 2: Manual Polling