Update app.py

app.py

@@ -1,61 +1,35 @@
-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."
-)
+from moviepy.editor import ImageSequenceClip
+import tempfile
+import os
 
-iface.launch(debug=True)
+def images_to_video(image_files):
+    # 임시 디렉터리 생성
+    temp_dir = tempfile.mkdtemp()
+    
+    # 이미지 파일을 임시 디렉터리로 저장
+    image_paths = []
+    for i, image_file in enumerate(image_files):
+        image_path = os.path.join(temp_dir, f"image_{i}.png")
+        with open(image_path, "wb") as f:
+            f.write(image_file)
+        image_paths.append(image_path)
+    
+    # 이미지 시퀀스로부터 비디오 클립 생성 (각 이미지의 지속 시간은 2초)
+    clip = ImageSequenceClip(image_paths, fps=0.5)  # fps=0.5 => 각 이미지는 2초 동안 보임
+    
+    # 비디오 파일 저장
+    output_video_path = os.path.join(temp_dir, "output.mp4")
+    clip.write_videofile(output_video_path, fps=24)  # 24fps로 출력 비디오 저장
+    
+    # 생성된 비디오 파일 경로 반환
+    return output_video_path
+
+# Gradio 인터페이스 정의
+with gr.Blocks() as demo:
+    with gr.Row():
+        file_input = gr.File(label="Upload images", type="file", accept="image/*", multiple=True)
+        video_output = gr.Video(label="Output video")
+        file_input.change(images_to_video, inputs=file_input, outputs=video_output)
+
+demo.launch()