Update app.py

app.py

@@ -1,36 +1,33 @@
-import cv2
-import gradio as gr
+from huggingface_hub import HfApi, HfFolder, Repository
 import os
-import tempfile
-from pathlib import Path
 
-def image_to_video(input_file, fps=30):
-    with tempfile.TemporaryDirectory() as temp_dir:
-        file_path = os.path.join(temp_dir, "image.jpg")
-        with open(file_path, 'wb') as file:
-            file.write(input_file)
+# Hugging Face 계정으로 로그인 (처음 실행 시 로그인 정보 입력)
+hf_username = "seawolf2357"  # 여기에 Hugging Face 사용자 이름 입력
+repo_name = "video"  # 여기에 리포지토리 이름 입력 (없으면 생성됨)
+file_path = "your_file_path.mp4"  # 업로드할 파일의 경로
 
-        frame = cv2.imread(file_path)
-        height, width, layers = frame.shape
-        size = (width, height)
-        
-        video_filename = os.path.join('/mnt/data', 'output_video.mp4')
-        
-        out = cv2.VideoWriter(video_filename, cv2.VideoWriter_fourcc(*'mp4v'), fps, size)
-        out.write(frame)
-        out.release()
+# API 초기화
+api = HfApi()
 
-        if not os.path.exists(video_filename):
-            raise FileNotFoundError(f"Failed to create video: {video_filename}")
+# 리포지토리 경로 설정
+repo_url = api.create_repo(
+    name=repo_name,
+    token=HfFolder.get_token(),
+    private=False,  # True로 설정하면 비공개 리포지토리 생성
+    exist_ok=True  # 이미 존재하는 경우 오류 방지
+)
 
-        return video_filename
+# 로컬에서 리포지토리 클론
+repo_local_path = "./" + repo_name
+repo = Repository(repo_local_path, clone_from=repo_url)
+os.chdir(repo_local_path)
 
-iface = gr.Interface(
-    fn=image_to_video,
-    inputs=gr.File(type="binary", label="Upload Image"),
-    outputs=gr.Video(label="Output Video"),
-    title="Image to Video Converter",
-    description="Upload an image to create a video."
-)
+# 파일 업로드
+repo.lfs_track("*.mp4")  # MP4 파일 LFS 추적
+repo.git_add(file_path)  # 파일 추가
+repo.git_commit("Upload MP4 file")  # 커밋 메시지
+repo.git_push()  # 변경 사항 푸시
 
-iface.launch()
+# 업로드된 파일의 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)