app.py

import os
import tempfile
import gradio as gr
from huggingface_hub import HfApi
import logging

# 로깅 설정
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

# 환경 변수에서 Hugging Face 토큰 읽기
hf_token = os.getenv("TOKEN")
if not hf_token:
    logging.error("Hugging Face token is not set. Please set the HF_TOKEN environment variable.")
    exit()

# Hugging Face API 초기화
api = HfApi()

def upload_file_to_hf_space(uploaded_file):
    user_id = "seawolf2357"
    space_name = "video"
    repo_id = f"{user_id}/{space_name}"

    # 임시 파일 생성 및 업로드된 파일 데이터 쓰기
    with tempfile.NamedTemporaryFile(delete=True, suffix='.mp4', mode='wb') as tmp_file:
        # 업로드된 파일의 바이트 데이터를 임시 파일에 쓰기
        if isinstance(uploaded_file, bytes):
            tmp_file.write(uploaded_file)
        else:
            # Gradio로부터 받은 파일 데이터가 bytes가 아닌 경우 처리
            data = uploaded_file.read()  # 파일 객체에서 바이트 데이터 읽기
            tmp_file.write(data)
        
        tmp_file.flush()
        file_path = tmp_file.name

        logging.info(f"Uploading {file_path} to Hugging Face Spaces")
        try:
            api.upload_file(
                path_or_fileobj=file_path,
                path_in_repo=os.path.basename(file_path),
                repo_id=repo_id,
                token=hf_token,
            )
            uploaded_file_url = f"https://huggingface.co/spaces/{repo_id}/blob/main/{os.path.basename(file_path)}"
            logging.info(f"File uploaded successfully: {uploaded_file_url}")
            return uploaded_file_url
        except Exception as e:
            logging.error(f"Failed to upload file: {e}")
            return "Failed to upload file."

# Gradio 인터페이스 설정 및 실행
iface = gr.Interface(
    fn=upload_file_to_hf_space,
    inputs=gr.File(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. Please ensure the file is an MP4 format."
)

iface.launch(debug=True)