Update app.py

app.py

@@ -2,8 +2,6 @@ import gradio as gr
 import requests
 import time
 import os
-import re
-
 
 # 환경 변수에서 API 키를 읽어옵니다.
 api_key = os.getenv("API_KEY")
@@ -11,26 +9,10 @@ api_key = os.getenv("API_KEY")
 # 여기에 실제 API 엔드포인트 주소를 할당합니다.
 api_url = "https://api.synclabs.so/video"
 
-def convert_google_drive_link(share_link):
-    patterns = [
-        r"/file/d/([a-zA-Z0-9_-]+)",
-        r"open\?id=([a-zA-Z0-9_-]+)",
-        r"uc\?id=([a-zA-Z0-9_-]+)&export=download"
-    ]
-    
-    for pattern in patterns:
-        match = re.search(pattern, share_link)
-        if match:
-            file_id = match.group(1)
-            return f"https://drive.google.com/uc?export=download&id={file_id}"
-    
-    # 매치되는 패턴이 없는 경우
-    raise ValueError("Google Drive 공유 링크에서 파일 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)
+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}"
 
     payload = {
         "audioUrl": audio_download_link,
@@ -67,9 +49,12 @@ def create_video(video_share_link, audio_share_link):
 
 iface = gr.Interface(
     fn=create_video,
-    inputs=[gr.Textbox(label="비디오 파일의 구글 드라이브 공유 링크"), gr.Textbox(label="오디오 파일의 구글 드라이브 공유 링크")],
+    inputs=[
+        gr.Textbox(label="비디오 파일의 Google Drive ID"), 
+        gr.Textbox(label="오디오 파일의 Google Drive ID")
+    ],
     outputs=gr.Textbox(label="생성된 비디오 다운로드 링크"),
-    description="구글 드라이브 공유된 mp4 파일과 오디오 파일을 합성하여 비디오를 생성합니다."
+    description="Google Drive 공유된 mp4 파일과 오디오 파일의 ID를 입력하여 비디오를 생성합니다."
 )
 
 # Gradio 앱 실행