Update app6.py

app6.py

@@ -2,6 +2,8 @@ import gradio as gr
 import os
 import time
 import logging
+import socket
+import requests
 from byteplussdkarkruntime import Ark
 import byteplussdkcore
 
@@ -11,26 +13,74 @@ logger = logging.getLogger(__name__)
 
 # Get API key
 API_KEY = os.environ.get("key", "")
+logger.info(f"API Key loaded: {'Yes' if API_KEY else 'No'}")
+logger.info(f"Key length: {len(API_KEY)}")
 
-# Configure SDK exactly as per the notice
-configuration = byteplussdkcore.Configuration()
-configuration.client_side_validation = True
-configuration.schema = "http"  # Explicitly set to HTTP as per notice
-configuration.debug = False
-configuration.logger_file = "sdk.log"
-byteplussdkcore.Configuration.set_default(configuration)
+# Test basic connectivity first
+def test_connectivity():
+    """Test if we can reach the API server"""
+    try:
+        # Test DNS resolution
+        ip = socket.gethostbyname('ark.ap-southeast.bytepluses.com')
+        logger.info(f"✅ DNS resolved: {ip}")
+        
+        # Test basic HTTPS connection
+        r = requests.get("https://ark.ap-southeast.bytepluses.com", timeout=5)
+        logger.info(f"✅ Base URL reachable (status: {r.status_code})")
+        return True
+    except Exception as e:
+        logger.error(f"❌ Connection test failed: {e}")
+        return False
+
+# Run connection test on startup
+connectivity_ok = test_connectivity()
+
+# Configure SDK
+try:
+    configuration = byteplussdkcore.Configuration()
+    configuration.client_side_validation = True
+    configuration.schema = "http"
+    configuration.debug = True  # Enable debug logging
+    configuration.logger_file = "sdk.log"
+    byteplussdkcore.Configuration.set_default(configuration)
+    logger.info("✅ SDK configured")
+except Exception as e:
+    logger.error(f"❌ SDK config failed: {e}")
 
 # Initialize client
-client = Ark(
-    base_url="https://ark.ap-southeast.bytepluses.com/api/v3",
-    api_key=API_KEY,
-)
+try:
+    client = Ark(
+        base_url="https://ark.ap-southeast.bytepluses.com/api/v3",
+        api_key=API_KEY,
+    )
+    logger.info("✅ Client initialized")
+except Exception as e:
+    logger.error(f"❌ Client init failed: {e}")
+    client = None
 
-def generate_video(prompt_text, image_url):
-    """Generate video using properly configured SDK"""
+def generate_video(prompt_text, image_url, progress=gr.Progress()):
+    """Generate video with detailed debugging"""
+    
+    if not API_KEY:
+        yield "❌ No API key", None
+        return
+    
+    if not client:
+        yield "❌ Client not initialized", None
+        return
+    
+    if not connectivity_ok:
+        yield "❌ Cannot reach API server", None
+        return
     
     try:
-        # Create task exactly like original code
+        progress(0, desc="Creating task...")
+        logger.info("Creating task...")
+        
+        # Log the request details
+        logger.info(f"Prompt: {prompt_text[:50]}...")
+        logger.info(f"Image URL: {image_url}")
+        
         create_result = client.content_generation.tasks.create(
             model="seedance-1-5-pro-251215",
             content=[
@@ -48,31 +98,53 @@ def generate_video(prompt_text, image_url):
         )
         
         task_id = create_result.id
+        logger.info(f"✅ Task created: {task_id}")
         yield f"Task created: {task_id}", None
         
+        progress(0.3, desc="Polling...")
+        
         # Poll for results
-        while True:
-            get_result = client.content_generation.tasks.get(task_id=task_id)
-            status = get_result.status
-            
-            if status == "succeeded":
-                video_url = get_result.output[0].get('video_url') if get_result.output else None
-                yield "Success!", video_url
-                break
-            elif status == "failed":
-                yield f"Failed: {get_result.error}", None
-                break
-            else:
-                yield f"Status: {status}...", None
-                time.sleep(1)
+        attempts = 0
+        while attempts < 60:
+            try:
+                get_result = client.content_generation.tasks.get(task_id=task_id)
+                status = get_result.status
+                logger.info(f"Status: {status}")
                 
+                if status == "succeeded":
+                    progress(1.0, desc="Complete!")
+                    video_url = get_result.output[0].get('video_url') if get_result.output else None
+                    yield "Success!", video_url
+                    return
+                elif status == "failed":
+                    yield f"Failed: {get_result.error}", None
+                    return
+                else:
+                    progress(0.3 + (attempts/60)*0.7, desc=f"Status: {status}")
+                    yield f"Status: {status}...", None
+                    time.sleep(1)
+                    attempts += 1
+            except Exception as e:
+                logger.error(f"Polling error: {e}")
+                yield f"Polling error: {e}", None
+                return
+                
+        yield "Timeout", None
+        
     except Exception as e:
+        logger.error(f"❌ Error: {e}", exc_info=True)
         yield f"Error: {str(e)}", None
 
 # Simple interface
 with gr.Blocks() as demo:
     gr.Markdown("# BytePlus Video Generator")
     
+    # Show connection status
+    if connectivity_ok:
+        gr.Markdown("✅ **API Server:** Reachable")
+    else:
+        gr.Markdown("❌ **API Server:** Cannot connect - check network")
+    
     with gr.Row():
         with gr.Column():
             prompt = gr.Textbox(
@@ -87,7 +159,7 @@ with gr.Blocks() as demo:
             btn = gr.Button("Generate", variant="primary")
         
         with gr.Column():
-            status = gr.Textbox(label="Status")
+            status = gr.Textbox(label="Status", lines=5)
             video = gr.Video(label="Generated Video")
     
     btn.click(
@@ -96,4 +168,4 @@ with gr.Blocks() as demo:
         outputs=[status, video]
     )
 
-demo.launch()
+demo.launch(server_name="0.0.0.0")