Update app.py

app.py

@@ -44,8 +44,33 @@ setup_eval_logging()
 
 # 번역기 및 Pixabay API 설정
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+
+
+
+import requests
+
+def search_pixabay_videos(query, api_key):
+    base_url = "https://pixabay.com/api/videos/"
+    params = {
+        "key": api_key,
+        "q": query,
+        "per_page": 80
+    }
+    
+    response = requests.get(base_url, params=params)
+    if response.status_code == 200:
+        data = response.json()
+        return [video['videos']['large']['url'] for video in data.get('hits', [])]
+    return []
+
+# 메인 코드에서 pixabay 관련 부분 수정
 PIXABAY_API_KEY = "33492762-a28a596ec4f286f84cd328b17"
-pixabay_video = Video(PIXABAY_API_KEY)
+
+def search_videos(query):
+    query = translate_prompt(query)
+    return search_pixabay_videos(query, PIXABAY_API_KEY)
+
+
 
 # CSS 스타일 정의
 custom_css = """