Update app.py

app.py

@@ -9,8 +9,23 @@ API_KEY = os.getenv("API_KEY")
 # 실제 API 엔드포인트 주소
 api_url = "https://api.synclabs.so/video"
 
+def check_status(job_id):
+    status_url = f"{api_url}/{job_id}"
+    headers = {"accept": "application/json", "x-api-key": API_KEY}
+    
+    while True:
+        response = requests.get(status_url, headers=headers)
+        if response.status_code == 200:
+            status = response.json()["status"]
+            if status == "COMPLETED":
+                return response.json()["url"]
+            elif status == "FAILED":
+                return "비디오 생성 실패."
+        else:
+            return "상태 확인 중 오류 발생."
+        time.sleep(10)  # 10초 간격으로 상태 체크
+
 def create_video(video_file_id, audio_file_id):
-    # 사용자로부터 입력받은 파일 ID를 기반으로 다운로드 링크 생성
     video_download_link = f"https://drive.google.com/uc?export=download&id={video_file_id}"
     audio_download_link = f"https://drive.google.com/uc?export=download&id={audio_file_id}"
 
@@ -23,46 +38,22 @@ def create_video(video_file_id, audio_file_id):
         "model": "sync-1"
     }
 
-    headers = {
-        "accept": "application/json",
-        "x-api-key": API_KEY,
-        "Content-Type": "application/json"
-    }
+    headers = {"accept": "application/json", "x-api-key": API_KEY, "Content-Type": "application/json"}
 
-    # API 호출 및 응답 처리
+    # API 호출
     response = requests.post(api_url, json=payload, headers=headers)
     if response.status_code in [200, 201]:
         job_id = response.json()["id"]
-        status_url = f"{api_url}/{job_id}"
-
-        # 작업 ID를 반환
-        yield f"작업 ID: {job_id}. 생성 상태를 확인중입니다..."
-        
-        while True:
-            status_response = requests.get(status_url, headers=headers)
-            if status_response.status_code == 200:
-                status = status_response.json()["status"]
-                if status == "COMPLETED":
-                    download_url = status_response.json()["url"]
-                    return download_url  # 생성된 비디오의 다운로드 링크 반환
-                elif status == "FAILED":
-                    return "비디오 생성 실패."
-                time.sleep(10)  # 상태 확인을 위해 잠시 대기
-            else:
-                return "상태 확인 중 오류 발생."
+        # 상태 확인 및 최종 결과 반환
+        download_url = check_status(job_id)
+        return download_url
     else:
         return "API 호출 실패."
 
 iface = gr.Interface(
     fn=create_video,
-    inputs=[
-        gr.Textbox(label="비디오 파일의 Google Drive ID"), 
-        gr.Textbox(label="오디오 파일의 Google Drive ID")
-    ],
-    outputs=[
-        gr.Textbox(label="작업 ID 및 상태"),
-        gr.Video(label="생성된 비디오")
-    ],
+    inputs=[gr.Textbox(label="비디오 파일의 Google Drive ID"), gr.Textbox(label="오디오 파일의 Google Drive ID")],
+    outputs=gr.Textbox(label="생성된 비디오 다운로드 링크"),
     description="Google Drive 공유된 mp4 파일과 오디오 파일의 ID를 입력하여 비디오를 생성합니다."
 )