Create app.py

app.py

@@ -0,0 +1,58 @@
+import gradio as gr
+import requests
+import time
+
+def convert_google_drive_link(share_link):
+    file_id = share_link.split('/')[-2]
+    return f"https://drive.google.com/uc?export=download&id={file_id}"
+
+def create_video(video_share_link, audio_share_link):
+    video_download_link = convert_google_drive_link(video_share_link)
+    audio_download_link = convert_google_drive_link(audio_share_link)
+
+    # API 요청 페이로드 설정
+    payload = {
+        "audioUrl": audio_download_link,
+        "videoUrl": video_download_link,
+        "synergize": True,
+        "maxCredits": None,
+        "webhookUrl": None,
+        "model": "sync-1"
+    }
+
+    headers = {
+        "Authorization": f"Bearer {api_key}"  # Hugging Face Secrets을 통해 안전하게 관리
+    }
+
+    # API 호출 및 응답 처리
+    response = requests.post(api_url, json=payload, headers=headers)
+    if response.status_code == 200:
+        job_id = response.json()["id"]
+        status_url = f"{api_url}/{job_id}"  # 상태 확인 URL
+        
+        # 상태 체크 루프
+        while True:
+            status_response = requests.get(status_url, headers=headers)
+            if status_response.status_code == 200:
+                status = status_response.json()["status"]
+                if status == "COMPLETED":
+                    return status_response.json()["url"]
+                elif status == "FAILED":
+                    return "비디오 생성 실패."
+                time.sleep(10)  # 10초 대기
+            else:
+                return "상태 확인 중 오류 발생."
+    else:
+        return "API 호출 실패."
+
+# Gradio 인터페이스 정의
+iface = gr.Interface(
+    fn=create_video,
+    inputs=[gr.inputs.Textbox(label="비디오 파일의 구글 드라이브 공유 링크"), 
+            gr.inputs.Textbox(label="오디오 파일의 구글 드라이브 공유 링크")],
+    outputs=gr.outputs.Textbox(label="생성된 비디오 다운로드 링크"),
+    description="구글 드라이브 공유된 mp4 파일과 오디오 파일을 합성하여 비디오를 생성합니다."
+)
+
+# Hugging Face Space에 배포하기 위해 실행
+iface.launch()