Update app.py

app.py

@@ -12,10 +12,10 @@ load_dotenv()
 
 # API Keys
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-B_KEY = os.getenv("B_KEY")
+REPLICATE_API_TOKEN = os.getenv("REPLICATE_API_TOKEN")
 
 # URLs
-API_URL = os.getenv("API_URL")
+REPLICATE_API_URL = "https://api.replicate.com/v1/predictions"
 UPLOAD_URL = os.getenv("UPLOAD_URL")
 
 def get_voices():
@@ -65,34 +65,33 @@ def upload_file(file_path):
 
 def lipsync_api_call(video_url, audio_url):
     headers = {
+        "Authorization": f"Bearer {REPLICATE_API_TOKEN}",
         "Content-Type": "application/json",
-        "x-api-key": B_KEY
+        "Prefer": "wait"
     }
 
     data = {
-        "audioUrl": audio_url,
-        "videoUrl": video_url,
-        "maxCredits": 1000,
-        "model": "sync-1.7.1-beta",
-        "synergize": True,
-        "pads": [0, 5, 0, 0],
-        "synergizerStrength": 1
+        "version": "db5a650c807b007dc5f9e5abe27c53e1b62880d1f94d218d27ce7fa802711d67",
+        "input": {
+            "face": video_url,
+            "input_audio": audio_url
+        }
     }
 
-    response = requests.post(API_URL, headers=headers, data=json.dumps(data))
+    response = requests.post(REPLICATE_API_URL, headers=headers, json=data)
     return response.json()
 
-def check_job_status(job_id):
-    headers = {"x-api-key": B_KEY}
+def check_job_status(prediction_id):
+    headers = {"Authorization": f"Bearer {REPLICATE_API_TOKEN}"}
     max_attempts = 30  # Limit the number of attempts
 
     for _ in range(max_attempts):
-        response = requests.get(f"{API_URL}/{job_id}", headers=headers)
+        response = requests.get(f"{REPLICATE_API_URL}/{prediction_id}", headers=headers)
         data = response.json()
 
-        if data["status"] == "COMPLETED":
-            return data["videoUrl"]
-        elif data["status"] == "FAILED":
+        if data["status"] == "succeeded":
+            return data["output"]
+        elif data["status"] == "failed":
             return None
 
         time.sleep(10)
@@ -183,13 +182,13 @@ def process_video(voice, url, text, progress=gr.Progress()):
         progress(0.5, desc="Initiating lipsync...")
         job_data = lipsync_api_call(video_url, audio_url)
 
-        if "error" in job_data or "message" in job_data:
-            raise Exception(job_data.get("error", job_data.get("message", "Unknown error")))
+        if "error" in job_data:
+            raise Exception(job_data.get("error", "Unknown error"))
 
-        job_id = job_data["id"]
+        prediction_id = job_data["id"]
 
         progress(0.6, desc="Processing lipsync...")
-        result_url = check_job_status(job_id)
+        result_url = check_job_status(prediction_id)
 
         if result_url:
             progress(0.9, desc="Downloading result...")
@@ -227,11 +226,8 @@ def process_video(voice, url, text, progress=gr.Progress()):
 
 def create_interface():
     voices = get_voices()
-    css = """
-    #component-0 > :not(.prose) {display: none !important;}
-    footer {display: none !important;}
-    """
-    with gr.Blocks(css=css) as app:
+    
+    with gr.Blocks() as app:
         gr.Markdown("# Generator")
         with gr.Row():
             with gr.Column():