Update app7.py

app7.py

@@ -1,377 +1,146 @@
 import gradio as gr
+import requests
 import os
 import time
-import json
-from PIL import Image
 import logging
-import sys
+from PIL import Image
+import io
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Print versions for debugging
-logger.info(f"Gradio version: {gr.__version__}")
-logger.info(f"Python version: {sys.version}")
-
-# Get API key from environment variable "Key"
-api_key = os.environ.get("Key", "")
-if not api_key:
-    logger.warning("⚠️ API Key not set in environment variable 'Key'!")
-    logger.info("Will use input field instead")
-else:
-    logger.info("✅ API Key loaded from environment")
+# Get API key (match your secret name - 'key' or 'Key')
+API_KEY = os.environ.get("key", os.environ.get("Key", ""))
+if not API_KEY:
+    logger.error("❌ No API key found")
 
-# Import BytePlus SDK
-try:
-    import byteplussdkcore
-    from byteplussdkarkruntime import Ark
-    SDK_AVAILABLE = True
-    logger.info("✅ BytePlus SDK loaded successfully")
-except ImportError as e:
-    SDK_AVAILABLE = False
-    logger.warning(f"⚠️ BytePlus SDK not available: {e}")
+def upload_to_temp_url(image):
+    """Upload image to temporary hosting and return URL"""
+    # For now, we'll need a temporary hosting solution
+    # Option 1: Use imgbb.com API (need another key)
+    # Option 2: Use Hugging Face's temp files (but need public URL)
+    # Option 3: Keep URL input for now until we set up hosting
+    
+    # For simplicity now, we'll keep URL input
+    # But we can add image preview
+    return None
 
-def initialize_sdk_config():
-    """Initialize SDK configuration as per docs"""
-    try:
-        configuration = byteplussdkcore.Configuration()
-        configuration.client_side_validation = True
-        configuration.schema = "http"
-        configuration.debug = False
-        configuration.logger_file = "sdk.log"
-        byteplussdkcore.Configuration.set_default(configuration)
-        return True
-    except Exception as e:
-        logger.error(f"SDK config error: {e}")
-        return False
-
-def generate_video(input_api_key, prompt_text, image_url, model_id, progress=gr.Progress()):
-    """Generate video using the BytePlus SDK"""
-    
-    # Use input API key if provided and not "key", otherwise fall back to environment variable
-    effective_api_key = input_api_key if input_api_key and input_api_key != "key" else api_key
-    
-    if not effective_api_key:
-        yield "⚠️ Please enter your BytePlus API key or set it in the 'Key' environment variable", None
-        return
+def generate_video(image, prompt_text, model_id, progress=gr.Progress()):
+    """Generate video using image upload and prompt"""
     
-    if not SDK_AVAILABLE:
-        yield "❌ BytePlus SDK not available. Please check installation.", None
+    if not API_KEY:
+        yield "❌ API key not configured", None
         return
     
     try:
-        progress(0, desc="Initializing SDK...")
-        
-        # Initialize SDK config
-        try:
-            initialize_sdk_config()
-        except:
-            pass  # Continue even if config fails
-        
-        # Set environment variable for SDK (it expects ARK_API_KEY)
-        os.environ["ARK_API_KEY"] = effective_api_key
-        
-        # Initialize client
-        client = Ark(
-            base_url="https://ark.ap-southeast.bytepluses.com/api/v3",
-            api_key=os.environ.get("ARK_API_KEY"),
-        )
-        
-        progress(0.1, desc="Creating video generation request...")
-        logger.info("🚀 Creating video generation request...")
-        
-        # Create task
-        create_result = client.content_generation.tasks.create(
-            model=model_id,
-            content=[
-                {
-                    "type": "text",
-                    "text": prompt_text
-                },
-                {
-                    "type": "image_url",
-                    "image_url": {
-                        "url": image_url
-                    }
-                }
-            ]
-        )
-        
-        task_id = create_result.id
-        logger.info(f"📋 Task ID: {task_id}")
+        progress(0, desc="Preparing request...")
         
-        yield f"⏳ Task created with ID: {task_id}. Waiting for completion...", None
+        # For now, we need a publicly accessible image URL
+        # This is a limitation - we need to host the uploaded image somewhere
+        # Let's use a placeholder message until we implement hosting
+        if image is None:
+            yield "⚠️ Please upload an image", None
+            return
         
-        # Polling
-        max_attempts = 60
-        attempts = 0
+        # For demo purposes, we'll keep URL input for now
+        # But we show the uploaded image preview
+        yield "⏳ Image uploaded. For now, please also provide a public URL (hosting service needed)", None
+        return
         
-        while attempts < max_attempts:
-            progress(0.1 + (attempts / max_attempts) * 0.8, 
-                    desc=f"Polling status: {attempts + 1}/{max_attempts}")
-            
-            get_result = client.content_generation.tasks.get(task_id=task_id)
-            status = get_result.status
-            
-            if status == "succeeded":
-                progress(1.0, desc="Complete!")
-                logger.info("✅ Task succeeded!")
-                
-                # Extract video URL
-                video_url = None
-                if hasattr(get_result, 'output') and get_result.output:
-                    if isinstance(get_result.output, list) and len(get_result.output) > 0:
-                        video_url = get_result.output[0].get('video_url')
-                    elif hasattr(get_result.output, 'video_url'):
-                        video_url = get_result.output.video_url
-                    elif isinstance(get_result.output, dict):
-                        video_url = get_result.output.get('video_url')
-                
-                if video_url:
-                    yield f"✅ Video generated successfully!", video_url
-                else:
-                    yield f"✅ Task completed but no video URL in response", None
-                break
-                
-            elif status == "failed":
-                error_msg = get_result.error if hasattr(get_result, 'error') else "Unknown error"
-                yield f"❌ Task failed: {error_msg}", None
-                break
-            else:
-                yield f"⏳ Current status: {status}, waiting... ({attempts + 1}/{max_attempts})", None
-                time.sleep(1)
-                attempts += 1
+        # The actual API call would go here once we have URL hosting
         
-        if attempts >= max_attempts:
-            yield "⏰ Timeout: Task took too long to complete. Please try again.", None
-            
     except Exception as e:
-        logger.error(f"Error: {e}")
+        logger.error(f"❌ Error: {str(e)}")
         yield f"❌ Error: {str(e)}", None
 
-# Custom CSS
-custom_css = """
-.gradio-container {
-    max-width: 1200px !important;
-    margin: auto !important;
-}
-.video-container {
-    border-radius: 8px;
-    overflow: hidden;
-}
-.env-badge {
-    background: #e3f2fd;
-    color: #0d47a1;
-    padding: 5px 10px;
-    border-radius: 4px;
-    font-size: 0.9em;
-    margin-bottom: 10px;
-}
-"""
-
-# Create Gradio interface
-with gr.Blocks(title="BytePlus Video Generator", fill_width=True) as demo:
+# Simple, clean interface
+with gr.Blocks(title="BytePlus Video Generator", theme=gr.themes.Soft()) as demo:
     
     gr.Markdown("""
     # 🎥 BytePlus Video Generator
-    Generate stunning videos from images and text prompts using BytePlus AI
+    Upload an image and describe the video you want to generate
     """)
     
-    # Show API key source
-    if api_key:
-        gr.Markdown(f"""
-        <div class="env-badge">
-        ✅ API Key loaded from environment variable 'Key'
-        </div>
-        """)
-    else:
-        gr.Markdown("""
-        <div class="env-badge" style="background: #fff3e0; color: #bf360c;">
-        ⚠️ No API Key in environment variable 'Key'. Please enter below.
-        </div>
-        """)
-    
-    # SDK Status
-    with gr.Row():
-        if SDK_AVAILABLE:
-            gr.Markdown("✅ **SDK Status:** Connected to BytePlus SDK")
-        else:
-            gr.Markdown("❌ **SDK Status:** SDK not available")
-    
-    gr.Markdown("---")
-    
     with gr.Row():
-        with gr.Column(scale=1):
-            # API Key input - will use env var as default if available
-            api_key_input = gr.Textbox(
-                label="🔑 BytePlus API Key",
-                placeholder="Enter your BytePlus API key here",
-                type="password",
-                value="" if api_key else "key",  # Empty if env var exists, otherwise "key"
-                info="Using env var 'Key' by default. Enter only to override."
-            )
-            
-            # Model ID
-            model_id = gr.Textbox(
-                label="🤖 Model ID",
-                value="seedance-1-5-pro-251215",
-                info="Model ID for video generation"
-            )
-            
-            # Image URL input
-            image_url = gr.Textbox(
-                label="🔗 Image URL",
-                placeholder="Enter public image URL",
-                value="https://ark-doc.tos-ap-southeast-1.bytepluses.com/seepro_i2v%20.png",
-                info="Image must be publicly accessible"
-            )
-            
-            # Image preview
-            image_preview = gr.Image(
-                label="Image Preview",
+        with gr.Column():
+            # Image upload (input only)
+            image_input = gr.Image(
+                label="📤 Upload Starting Image",
                 type="pil",
-                height=200
+                height=300
             )
             
             # Text prompt
             prompt = gr.Textbox(
-                label="📝 Text Prompt",
-                placeholder="Describe your video...",
-                value="At breakneck speed, drones thread through intricate obstacles or stunning natural wonders, delivering an immersive, heart-pounding flying experience. --duration 5 --camerafixed false",
-                lines=4
+                label="📝 Video Description",
+                placeholder="Describe what you want to see...",
+                lines=3,
+                value="At breakneck speed, drones thread through intricate obstacles --duration 5 --camerafixed false"
+            )
+            
+            # Model ID (hidden or simplified)
+            model_id = gr.Textbox(
+                label="Model ID",
+                value="seedance-1-5-pro-251215",
+                visible=False  # Hide it for simplicity
             )
             
             # Generate button
             generate_btn = gr.Button("🚀 Generate Video", variant="primary", size="lg")
         
-        with gr.Column(scale=1):
-            # Status output
+        with gr.Column():
+            # Status messages
             status = gr.Textbox(
-                label="📊 Generation Status",
-                lines=5
-            )
-            
-            # Video output - REMOVED the unsupported parameters
-            video = gr.Video(
-                label="🎬 Generated Video"
-                # show_download_button and show_share_button removed as they're not in Gradio 6.5.1
-            )
-            
-            # Video URL
-            video_url = gr.Textbox(
-                label="🔗 Video URL",
+                label="Status",
+                lines=2,
                 interactive=False
             )
             
-            # Add download button separately if needed
-            with gr.Row():
-                download_btn = gr.DownloadButton(
-                    label="📥 Download Video",
-                    variant="secondary",
-                    visible=False  # Hidden until video is generated
-                )
+            # Video output (player only, not uploader)
+            video_output = gr.Video(
+                label="🎬 Generated Video",
+                interactive=False,  # Can't upload, only view
+                show_download_button=True,  # Allow download
+                height=300
+            )
     
-    # Example prompts
+    # Example prompts (optional)
     gr.Markdown("---")
-    gr.Markdown("## 📋 Example Prompts")
-    
+    gr.Markdown("### Example Prompts")
     with gr.Row():
-        example1 = gr.Button("🌄 Nature Drone", size="sm")
-        example2 = gr.Button("🏙️ City Racing", size="sm")
-        example3 = gr.Button("🌊 Ocean Waves", size="sm")
-    
-    # Function to update image preview
-    def update_preview(url):
-        try:
-            if url and url.startswith(('http://', 'https://')):
-                import requests
-                from io import BytesIO
-                
-                response = requests.get(url, timeout=5)
-                img = Image.open(BytesIO(response.content))
-                return img
-            return None
-        except Exception as e:
-            logger.error(f"Preview error: {e}")
-            return None
-    
-    # Example prompt functions
-    def set_nature():
-        return "Aerial drone shot flying over majestic mountains at sunrise, cinematic lighting, smooth motion --duration 5 --camerafixed false"
-    
-    def set_city():
-        return "Fast-paced drone racing through futuristic city streets with neon lights, dynamic angles, high speed --duration 5 --camerafixed false"
-    
-    def set_ocean():
-        return "Drone following surfers riding massive waves, slow motion, dramatic ocean views, golden hour --duration 5 --camerafixed false"
-    
-    # Function to update download button when video is generated
-    def update_download_button(video_path):
-        if video_path and video_path != "None" and video_path != "":
-            return gr.update(visible=True, value=video_path)
-        return gr.update(visible=False)
-    
-    # Connect events
-    image_url.change(
-        fn=update_preview,
-        inputs=image_url,
-        outputs=image_preview
-    )
-    
-    example1.click(fn=set_nature, outputs=prompt)
-    example2.click(fn=set_city, outputs=prompt)
-    example3.click(fn=set_ocean, outputs=prompt)
-    
-    # Generate
-    generate_event = generate_btn.click(
-        fn=generate_video,
-        inputs=[api_key_input, prompt, image_url, model_id],
-        outputs=[status, video]
-    )
-    
-    # Update video URL and download button
-    generate_event.then(
-        fn=lambda url: url if url else "",
-        inputs=[video],
-        outputs=[video_url]
-    ).then(
-        fn=update_download_button,
-        inputs=[video],
-        outputs=[download_btn]
-    )
-    
-    # Clear button
-    clear_btn = gr.Button("🗑️ Clear", variant="secondary")
-    clear_btn.click(
-        fn=lambda: (
-            "" if api_key else "key",  # Reset API key input
-            "https://ark-doc.tos-ap-southeast-1.bytepluses.com/seepro_i2v%20.png",
-            "Enter your prompt here...",
-            "",
-            None,
-            "",
-            gr.update(visible=False)
-        ),
-        outputs=[api_key_input, image_url, prompt, status, video, video_url, download_btn]
-    )
+        gr.Button("Nature Drone").click(
+            fn=lambda: "Aerial drone shot over mountains at sunrise, cinematic --duration 5 --camerafixed false",
+            outputs=prompt
+        )
+        gr.Button("City Race").click(
+            fn=lambda: "Fast drone racing through futuristic city streets --duration 5 --camerafixed false",
+            outputs=prompt
+        )
+        gr.Button("Ocean Waves").click(
+            fn=lambda: "Drone following waves at golden hour --duration 5 --camerafixed false",
+            outputs=prompt
+        )
     
-    # Footer
-    gr.Markdown("---")
+    # For now, show a note about image hosting
     gr.Markdown("""
-    <div style='text-align: center'>
-        <p>Powered by BytePlus SDK | Gradio 6.5.1</p>
-        <p style='font-size: 0.8em; color: gray;'>API key sourced from environment variable 'Key'</p>
-    </div>
+    ---
+    ⚠️ **Note:** Currently requires publicly accessible image URLs. 
+    Working on direct image upload support!
     """)
+    
+    # Placeholder function for now
+    def placeholder_gen(image, prompt, model):
+        if image is None:
+            return "Please upload an image first", None
+        return "Image uploaded! (URL hosting coming soon)", None
+    
+    # Connect the button
+    generate_btn.click(
+        fn=placeholder_gen,
+        inputs=[image_input, prompt, model_id],
+        outputs=[status, video_output]
+    )
 
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        theme=gr.themes.Soft(
-            primary_hue="blue",
-            secondary_hue="purple",
-        ),
-        css=custom_css
-    )
+    demo.launch(server_name="0.0.0.0")