Update app.py

app.py

@@ -1,6 +1,10 @@
 import gradio as gr
 import requests
 import time
+import os  # 환경 변수에 접근하기 위해 추가
+
+# 환경 변수에서 API 키를 읽어옵니다.
+api_key = os.getenv("API_KEY")
 
 def convert_google_drive_link(share_link):
     file_id = share_link.split('/')[-2]
@@ -10,7 +14,6 @@ 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,
@@ -21,16 +24,15 @@ def create_video(video_share_link, audio_share_link):
     }
 
     headers = {
-        "Authorization": f"Bearer {api_key}"  # Hugging Face Secrets을 통해 안전하게 관리
+        "Authorization": f"Bearer {api_key}"
     }
 
     # API 호출 및 응답 처리
-    response = requests.post(api_url, json=payload, headers=headers)
+    response = requests.post(api_url, json=payload, headers=headers)  # api_url을 실제 API 엔드포인트로 수정해야 합니다.
     if response.status_code == 200:
         job_id = response.json()["id"]
-        status_url = f"{api_url}/{job_id}"  # 상태 확인 URL
+        status_url = f"{api_url}/{job_id}"
         
-        # 상태 체크 루프
         while True:
             status_response = requests.get(status_url, headers=headers)
             if status_response.status_code == 200:
@@ -39,22 +41,17 @@ def create_video(video_share_link, audio_share_link):
                     return status_response.json()["url"]
                 elif status == "FAILED":
                     return "비디오 생성 실패."
-                time.sleep(10)  # 10초 대기
+                time.sleep(10)
             else:
                 return "상태 확인 중 오류 발생."
     else:
         return "API 호출 실패."
 
-# Gradio 인터페이스 정의
 iface = gr.Interface(
     fn=create_video,
-    inputs=[
-        gr.Textbox(label="비디오 파일의 구글 드라이브 공유 링크"), 
-        gr.Textbox(label="오디오 파일의 구글 드라이브 공유 링크")
-    ],
+    inputs=[gr.Textbox(label="비디오 파일의 구글 드라이브 공유 링크"), gr.Textbox(label="오디오 파일의 구글 드라이브 공유 링크")],
     outputs=gr.Textbox(label="생성된 비디오 다운로드 링크"),
     description="구글 드라이브 공유된 mp4 파일과 오디오 파일을 합성하여 비디오를 생성합니다."
 )
 
-# Hugging Face Space에 배포하기 위해 실행
 iface.launch()