Update app.py

app.py

@@ -7,7 +7,7 @@ import os
 API_KEY = os.getenv("API_KEY")
 
 # 실제 API 엔드포인트 주소
-api_url = "https://api.synclabs.so/video"
+api_url = os.getenv("api_url")
 
 def check_status_and_get_link(job_id):
     status_url = f"{api_url}/{job_id}"
@@ -35,7 +35,7 @@ def create_video(video_file_id, audio_file_id):
         "synergize": True,
         "maxCredits": None,
         "webhookUrl": None,
-        "model": "sync-1"
+        "model": "sync-1.5.1-beta"
     }
 
     headers = {"accept": "application/json", "x-api-key": API_KEY, "Content-Type": "application/json"}
@@ -45,9 +45,15 @@ def create_video(video_file_id, audio_file_id):
         job_id = response.json()["id"]
         download_url, message = check_status_and_get_link(job_id)
         if download_url:
-            # HTML 코드 생성: 비디오 임베드 및 다운로드 링크
-            html_code = f'<video width="640" height="360" controls><source src="{download_url}" type="video/mp4"></video><br><a href="{download_url}" target="_blank">다운로드</a>'
-            return job_id, html_code, message
+            # HTML 코드 생성: 비디오 임베드 및 다운로드 버튼
+            html_code = f'''
+            <video width="640" height="360" controls>
+                <source src="{download_url}" type="video/mp4">
+            </video>
+            <br>
+            <button onclick="window.location.href='{download_url}'">다운로드</button>
+            '''
+            return job_id, html_code, "비디오가 성공적으로 생성되었습니다."
         else:
             return job_id, "비디오 링크를 불러올 수 없습니다.", message
     else:
@@ -55,8 +61,15 @@ def create_video(video_file_id, audio_file_id):
 
 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.HTML(label="비디오"), gr.Textbox(label="메시지")],
+    inputs=[
+        gr.Textbox(label="비디오 파일의 Google Drive ID"), 
+        gr.Textbox(label="오디오 파일의 Google Drive ID")
+    ],
+    outputs=[
+        gr.Textbox(label="작업 ID"),
+        gr.HTML(label="비디오"),
+        gr.Textbox(label="메시지")
+    ],
     description="Google Drive 공유된 mp4 파일과 오디오 파일의 ID를 입력하여 비디오를 생성합니다."
 )