Update app.py

app.py

@@ -106,6 +106,89 @@ 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}")
@@ -252,7 +335,6 @@ def process_video(video_url, audio_url, progress=gr.Progress()):
 
 def create_interface():
     css = """
-    
     """
     with gr.Blocks(css=css) as app:
         gr.Markdown("# Lipsync Video Generator")