Update app.py

app.py

@@ -1,57 +1,33 @@
-import requests
 import gradio as gr
 import imageio
-import os
-from PIL import Image
 import numpy as np
-import time
-
-# 환경 변수에서 API 키를 불러옵니다. 시스템에 'API_KEY' 환경 변수를 설정해야 합니다.
-API_KEY = os.environ.get("18BkVTfHD8DRxtR7uoO5RMu6O2P5PywUMoHTK4eRMz7MAEBQlOI5qRTC")
-API_URL = 'https://api.pexels.com/v1'
-
-def download_images(keyword):
-    headers = {'Authorization': API_KEY}
-    params = {'query': keyword, 'per_page': 5}  # 5장의 이미지를 검색합니다.
-    response = requests.get(f"{API_URL}/search", headers=headers, params=params)
-    images = []
-    
-    if response.status_code == 200:
-        photos = response.json()['photos']
-        for photo in photos:
-            image_url = photo['src']['original']
-            img_data = requests.get(image_url).content
-            image_path = f"temp_{photos.index(photo)}.jpg"
-            with open(image_path, 'wb') as handler:
-                handler.write(img_data)
-            images.append(image_path)
-    
-    return images
+from PIL import Image
 
-def create_video(images):
-    video_path = "output.mp4"
-    with imageio.get_writer(video_path, fps=0.5) as writer:  # 2초 간격으로 설정
-        for image_path in images:
-            image = Image.open(image_path)
-            image = image.resize((640, 480))  # 모든 이미지를 동일한 크기로 조정
-            writer.append_data(np.array(image))
-            os.remove(image_path)  # 이미지 파일 삭제
+# 이미지를 영상으로 변환하는 함수
+def create_video(image_paths, video_length=1800, fps=1):
+    video_path = "output_video.mp4"
+    with imageio.get_writer(video_path, fps=fps) as writer:
+        for _ in range(video_length):
+            for image_path in image_paths:
+                image = Image.open(image_path)
+                image = image.resize((640, 480))  # 모든 이미지를 동일한 크기로 조정
+                writer.append_data(np.array(image))
     return video_path
 
-def search_and_create_video(keyword):
-    images = download_images(keyword)
-    if images:
-        video_path = create_video(images)
-        return video_path
-    else:
-        return "검색된 이미지가 없습니다."
+# Gradio 인터페이스 함수
+def generate_video(keyword):
+    # 여기서는 이미지 수집 과정을 생략하고, 가상의 이미지 경로 리스트를 사용합니다.
+    image_paths = ["image1.jpg", "image2.jpg", "image3.jpg"]  # 가상의 이미지 경로
+    video_path = create_video(image_paths)
+    return video_path
 
+# Gradio 인터페이스 설정
 demo = gr.Interface(
-    fn=search_and_create_video, 
-    inputs=gr.Textbox(lines=2, placeholder="검색할 키워드를 입력하세요"), 
+    fn=generate_video,
+    inputs=gr.Textbox(lines=2, placeholder="검색할 키워드를 입력하세요"),
     outputs=gr.Video(label="결과 영상"),
-    title="이미지를 영상으로 변환",
-    description="입력된 키워드에 맞는 이미지를 검색하여, 이를 순차적인 영상으로 합칩니다."
+    title="키워드 기반 영상 생성",
+    description="입력된 키워드에 맞는 영상을 생성합니다."
 )
 
 if __name__ == "__main__":