Update app.py

app.py

@@ -40,20 +40,23 @@ def create_video(video_file_id, audio_file_id):
 
     headers = {"accept": "application/json", "x-api-key": API_KEY, "Content-Type": "application/json"}
 
-    # API 호출
     response = requests.post(api_url, json=payload, headers=headers)
     if response.status_code in [200, 201]:
         job_id = response.json()["id"]
-        # 상태 확인 및 최종 결과 반환
         download_url, message = check_status_and_get_link(job_id)
-        return job_id, download_url, message
+        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
+        else:
+            return job_id, "비디오 링크를 불러올 수 없습니다.", message
     else:
         return "API 호출 실패", None, None
 
 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.Textbox(label="비디오 다운로드 링크"), gr.Textbox(label="메시지")],
+    outputs=[gr.Textbox(label="작업 ID"), gr.HTML(label="비디오"), gr.Textbox(label="메시지")],
     description="Google Drive 공유된 mp4 파일과 오디오 파일의 ID를 입력하여 비디오를 생성합니다."
 )