Update app.py

app.py

@@ -1,32 +1,43 @@
-from huggingface_hub import HfApi, HfFolder, Repository
 import os
+import gradio as gr
 from huggingface_hub import HfApi
 
+# 환경 변수에서 Hugging Face 토큰 읽기
+hf_token = os.getenv("HF_TOKEN")
+
+# Hugging Face API 초기화
 api = HfApi()
-repo_name = "seawolf2357"
-hf_username = "video"
-token = os.getenv("TOKEN")
 
-# 리포지토리 생성
-repo_url = api.create_repo(
-    repo_id=f"{hf_username}/{repo_name}",
-    token=token,  # 토큰 추가
-    private=False,  # True로 설정하면 비공개 리포지토리 생성
-    exist_ok=True  # 이미 존재하는 경우 오류 방지
-)
+def upload_file_to_hf_space(uploaded_file):
+    # 사용자 ID와 스페이스 이름 설정
+    user_id = "seawolf2357"
+    space_name = "video"  # 스페이스 이름을 여기에 입력하세요.
+    repo_id = f"{user_id}/{space_name}"
 
+    # 파일 업로드
+    with open(uploaded_file.name, "wb") as f:
+        f.write(uploaded_file.read())
+    
+    # Hugging Face Spaces에 파일 업로드
+    response = api.upload_file(
+        path_or_fileobj=uploaded_file.name,
+        path_in_repo=uploaded_file.name,
+        repo_id=repo_id,
+        token=hf_token,
+    )
 
-# 로컬에서 리포지토리 클론
-repo_local_path = "./" + repo_name
-repo = Repository(repo_local_path, clone_from=repo_url)
-os.chdir(repo_local_path)
+    # 업로드된 파일의 URL 반환
+    uploaded_file_url = f"https://huggingface.co/spaces/{repo_id}/blob/main/{uploaded_file.name}"
+    return uploaded_file_url
 
-# 파일 업로드
-repo.lfs_track("*.mp4")  # MP4 파일 LFS 추적
-repo.git_add(file_path)  # 파일 추가
-repo.git_commit("Upload MP4 file")  # 커밋 메시지
-repo.git_push()  # 변경 사항 푸시
+# Gradio 인터페이스 설정
+iface = gr.Interface(
+    fn=upload_file_to_hf_space,
+    inputs=gr.inputs.File(filetypes=["mp4"], label="Upload your MP4 file"),
+    outputs="text",
+    title="MP4 File Upload to Hugging Face Spaces",
+    description="Upload an MP4 file and get its URL in Hugging Face Spaces."
+)
 
-# 업로드된 파일의 URL 반환
-uploaded_file_url = f"https://huggingface.co/{hf_username}/{repo_name}/blob/main/{os.path.basename(file_path)}"
-print("Uploaded file URL:", uploaded_file_url)
+# 인터페이스 실행
+iface.launch()