Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import tempfile
 import gradio as gr
 from huggingface_hub import HfApi
 
@@ -9,22 +10,18 @@ hf_token = os.getenv("HF_TOKEN")
 api = HfApi()
 
 def upload_file_to_hf_space(uploaded_file):
-    # 사용자 ID와 스페이스 이름 설정
     user_id = "seawolf2357"
-    space_name = "video"  # 스페이스 이름을 여기에 입력하세요.
+    space_name = "video"
     repo_id = f"{user_id}/{space_name}"
 
-    # 파일명 설정 및 저장
-    file_path = uploaded_file.name
-    with open(file_path, "wb") as f:
-        f.write(uploaded_file.read())
-
-    # 파일 유형 검사 (MP4 파일인지 확인)
-    if not file_path.endswith('.mp4'):
-        return "Please upload an MP4 file."
+    # 임시 파일 생성 및 업로드된 파일 데이터 쓰기
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as tmp_file:
+        # Gradio 2.0 이상에서는 uploaded_file이 파일의 바이너리 데이터를 직접 제공
+        tmp_file.write(uploaded_file)
+        file_path = tmp_file.name
 
     # Hugging Face Spaces에 파일 업로드
-    response = api.upload_file(
+    api.upload_file(
         path_or_fileobj=file_path,
         path_in_repo=os.path.basename(file_path),
         repo_id=repo_id,
@@ -33,9 +30,10 @@ def upload_file_to_hf_space(uploaded_file):
 
     # 업로드된 파일의 URL 반환
     uploaded_file_url = f"https://huggingface.co/spaces/{repo_id}/blob/main/{os.path.basename(file_path)}"
+    os.unlink(file_path)  # 임시 파일 삭제
     return uploaded_file_url
 
-# Gradio 인터페이스 설정
+# Gradio 인터페이스 설정 및 실행
 iface = gr.Interface(
     fn=upload_file_to_hf_space,
     inputs=gr.File(label="Upload your MP4 file"),
@@ -44,5 +42,4 @@ iface = gr.Interface(
     description="Upload an MP4 file and get its URL in Hugging Face Spaces. Please ensure the file is an MP4 format."
 )
 
-# 인터페이스 실행
-iface.launch()
+iface.launch()