Update app.py

app.py

@@ -1,47 +1,44 @@
-import requests
 import gradio as gr
-import os
 import imageio
 import numpy as np
 from PIL import Image
-import requests
-from io import BytesIO
-import tempfile
 
-# Pexels API 키를 여기에 입력하세요
-API_KEY = os.getenv('API_KEY')
-API_URL = 'https://api.pexels.com/v1/search'
+def resize_image(image, target_width, target_height):
+    """
+    이미지를 지정된 크기로 조정합니다.
+    """
+    image_pil = Image.fromarray(image).resize((target_width, target_height), Image.ANTIALIAS)
+    return np.array(image_pil)
 
-def search_images(keyword):
-    headers = {'Authorization': API_KEY}
-    params = {'query': keyword, 'per_page': 5}  # per_page 값을 5로 설정
-    response = requests.get(API_URL, headers=headers, params=params)
-   
-    if response.status_code == 200 and response.json()['photos']:
-        photos = response.json()['photos']
-        images = []
-        for photo in photos:
-            image_url = photo['src']['original']
-            response = requests.get(image_url)
-            img = Image.open(BytesIO(response.content))
-            images.append(img)
-        
-        # 이미지를 영상으로 합치기
-        with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as tmpfile:
-            with imageio.get_writer(tmpfile.name, fps=0.5) as video:  # 2초 간격
-                for img in images:
-                    video.append_data(np.array(img))
-            return tmpfile.name  # 임시 파일의 경로 반환
-    else:
-        return "검색 결과가 없습니다."
+def create_video(images):
+    """
+    주어진 이미지 리스트로부터 비디오를 생성합니다.
+    모든 이미지는 비디오에 추가되기 전에 동일한 크기로 조정됩니다.
+    """
+    target_width = 1920
+    target_height = 1080
+    with imageio.get_writer('output_video.mp4', fps=2) as video:
+        for img in images:
+            img_resized = resize_image(img, target_width, target_height)
+            video.append_data(img_resized)
+    return 'output_video.mp4'
 
-interface = gr.Interface(
-    fn=search_images,
-    inputs=gr.Textbox(lines=2, placeholder="검색할 이미지 키워드를 입력하세요"),
-    outputs=gr.Video(label="검색 결과"),
-    title="Pexels 이미지 검색기",
-    description="Pexels에서 키워드에 맞는 이미지를 검색하고, 해당 이미지를 영상으로 변환합니다."
+def process_images(image_files):
+    """
+    업로드된 이미지 파일들을 처리하여 비디오를 생성합니다.
+    """
+    images = [imageio.imread(image_file) for image_file in image_files]
+    video_file = create_video(images)
+    return video_file
+
+# Gradio 인터페이스 정의
+iface = gr.Interface(
+    fn=process_images,
+    inputs=gr.inputs.Image(type="file", label="Upload Images", multiple=True),
+    outputs="file",
+    title="Image to Video Converter",
+    description="Upload multiple images to create a video."
 )
 
-if __name__ == "__main__":
-    interface.launch()
+# Gradio 앱 실행
+iface.launch()