Update app.py

app.py

@@ -15,13 +15,59 @@ load_dotenv()
 B_KEY = os.getenv("B_KEY")
 
 # URLs
-API_URL = "https://api.sync.so/v2/generate"  # Updated API endpoint
+API_URL = "https://api.sync.so/v2/generate"
+
+def get_media_resolution(url):
+    print(f"\n[DEBUG] Getting resolution for: {url}")
+    
+    # Download the file to a temporary location
+    response = requests.get(url)
+    if response.status_code != 200:
+        print(f"[ERROR] Failed to download media. Status code: {response.status_code}")
+        return None
+        
+    temp_path = f"temp_media_{uuid.uuid4()}"
+    with open(temp_path, 'wb') as f:
+        f.write(response.content)
+    
+    # Get resolution using FFprobe
+    cmd = [
+        'ffprobe', 
+        '-v', 'error',
+        '-select_streams', 'v:0',
+        '-show_entries', 'stream=width,height',
+        '-of', 'json',
+        temp_path
+    ]
+    
+    try:
+        result = subprocess.run(cmd, capture_output=True, text=True)
+        os.remove(temp_path)  # Clean up temp file
+        
+        if result.returncode == 0:
+            data = json.loads(result.stdout)
+            if 'streams' in data and data['streams']:
+                width = data['streams'][0].get('width')
+                height = data['streams'][0].get('height')
+                if width and height:
+                    print(f"[DEBUG] Detected resolution: {width}x{height}")
+                    return [width, height]
+    except Exception as e:
+        print(f"[ERROR] Failed to get resolution: {str(e)}")
+        if os.path.exists(temp_path):
+            os.remove(temp_path)
+    
+    print("[DEBUG] Failed to detect resolution, using default")
+    return [1280, 720]  # Default resolution
 
 def lipsync_api_call(video_url, audio_url):
     print(f"\n[DEBUG] Starting lipsync_api_call")
     print(f"[DEBUG] Video URL: {video_url}")
     print(f"[DEBUG] Audio URL: {audio_url}")
     
+    # Get the resolution of the input video/image
+    resolution = get_media_resolution(video_url)
+    
     headers = {
         "Content-Type": "application/json",
         "x-api-key": B_KEY
@@ -45,7 +91,7 @@ def lipsync_api_call(video_url, audio_url):
             "output_format": "mp4",
             "sync_mode": "bounce",
             "fps": 24,
-            "output_resolution": [1280, 720]
+            "output_resolution": resolution
         }
     }
     
@@ -60,94 +106,14 @@ def lipsync_api_call(video_url, audio_url):
         print(f"[ERROR] API call failed: {str(e)}")
         raise
 
-def check_job_status(job_id):
-    print(f"\n[DEBUG] Checking job status for ID: {job_id}")
-    headers = {"x-api-key": B_KEY}
-    max_attempts = 3000
-    attempt = 0
-    
-    while attempt < max_attempts:
-        try:
-            response = requests.get(f"{API_URL}/{job_id}", headers=headers)
-            print(f"[DEBUG] Status check attempt {attempt + 1}")
-            print(f"[DEBUG] Status response: {response.text}")
-            
-            data = response.json()
-            status = data.get("status")
-            print(f"[DEBUG] Current status: {status}")
-            
-            if status == "COMPLETED":
-                print(f"[DEBUG] Job completed. Output URL: {data.get('outputUrl')}")
-                return data.get("outputUrl")
-            elif status == "FAILED" or status == "CANCELED":
-                print(f"[ERROR] Job failed or was canceled. Error: {data.get('error')}")
-                return None
-            
-            attempt += 1
-            time.sleep(10)
-            
-        except Exception as e:
-            print(f"[ERROR] Status check failed: {str(e)}")
-            return None
-    
-    print("[ERROR] Max attempts reached")
-    return None
-
-def get_media_duration(file_path):
-    print(f"\n[DEBUG] Getting duration for: {file_path}")
-    cmd = ['ffprobe', '-v', 'error', '-show_entries', 'format=duration', '-of', 'default=noprint_wrappers=1:nokey=1', file_path]
-    result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-    duration = float(result.stdout.strip())
-    print(f"[DEBUG] Media duration: {duration} seconds")
-    return duration
-
-def combine_audio_video(video_path, audio_path, output_path):
-    print(f"\n[DEBUG] Combining audio and video")
-    print(f"[DEBUG] Video path: {video_path}")
-    print(f"[DEBUG] Audio path: {audio_path}")
-    print(f"[DEBUG] Output path: {output_path}")
-    
-    video_duration = get_media_duration(video_path)
-    audio_duration = get_media_duration(audio_path)
-
-    if video_duration > audio_duration:
-        print("[DEBUG] Video longer than audio - trimming video")
-        cmd = [
-            'ffmpeg', '-i', video_path, '-i', audio_path,
-            '-t', str(audio_duration),
-            '-map', '0:v', '-map', '1:a',
-            '-c:v', 'copy', '-c:a', 'aac',
-            '-y', output_path
-        ]
-    else:
-        print("[DEBUG] Audio longer than video - looping video")
-        loop_count = int(audio_duration // video_duration) + 1
-        cmd = [
-            'ffmpeg', '-stream_loop', str(loop_count), '-i', video_path, '-i', audio_path,
-            '-t', str(audio_duration),
-            '-map', '0:v', '-map', '1:a',
-            '-c:v', 'copy', '-c:a', 'aac',
-            '-shortest', '-y', output_path
-        ]
-
-    print(f"[DEBUG] FFmpeg command: {' '.join(cmd)}")
-    result = subprocess.run(cmd, capture_output=True, text=True)
-    print(f"[DEBUG] FFmpeg stdout: {result.stdout}")
-    print(f"[DEBUG] FFmpeg stderr: {result.stderr}")
-
-def is_image_url(url):
-    parsed = urlparse(url)
-    path = parsed.path.lower()
-    result = path.endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp', '.tiff', '.webp', '.heic', '.svg', '.ico'))
-    print(f"\n[DEBUG] Checking if URL is image: {url}")
-    print(f"[DEBUG] Result: {result}")
-    return result
-
 def create_video_from_image(image_url, output_path, duration=10):
     print(f"\n[DEBUG] Creating video from image")
     print(f"[DEBUG] Image URL: {image_url}")
     print(f"[DEBUG] Output path: {output_path}")
     
+    # Get the resolution before creating the video
+    resolution = get_media_resolution(image_url)
+    
     response = requests.get(image_url)
     if response.status_code != 200:
         print(f"[ERROR] Failed to download image. Status code: {response.status_code}")
@@ -162,7 +128,7 @@ def create_video_from_image(image_url, output_path, duration=10):
     cmd = [
         'ffmpeg', '-loop', '1', '-i', temp_image_path,
         '-c:v', 'libx264', '-t', str(duration), '-pix_fmt', 'yuv420p',
-        '-vf', 'scale=trunc(iw/2)*2:trunc(ih/2)*2',
+        '-vf', f'scale={resolution[0]}:{resolution[1]}',
         '-y', output_path
     ]