Update app.py

app.py

@@ -4,7 +4,21 @@ import json
 import re
 from moviepy.editor import VideoFileClip
 from moviepy.audio.AudioClip import AudioClip
+import requests
 
+def search_pexels_images(query):
+    API_KEY = '5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62'  # Pexels API 키를 여기에 입력하세요.
+    url = f"https://api.pexels.com/v1/search?query={query}&per_page=5"  # 검색 결과 5개만 가져옵니다.
+    
+    headers = {
+        "Authorization": API_KEY
+    }
+    response = requests.get(url, headers=headers)
+    data = response.json()
+    
+    images_urls = [photo['src']['medium'] for photo in data['photos']]
+    return images_urls
+    
 def extract_audio(video_in):
     input_video = video_in
     output_audio = 'audio.wav'
@@ -136,24 +150,47 @@ css="""
 }
 """
 
-image_examples = [
-    ["./dog.jpg"],
-    ["./bird.jpg"]
-]
+def show_search_results(query):
+    images_urls = search_pexels_images(query)
+    return images_urls
 
-with gr.Blocks() as demo:
+with gr.Blocks() as app:
     with gr.Column():
-        gr.Markdown("### Image to SFX\nCompare MAGNet, AudioLDM2, and AudioGen sound effects generation from image caption.")
-        image_in = gr.Image(label="Upload Image", type="filepath")
-        chosen_model = gr.Radio(label="Choose a model", choices=["MAGNet", "AudioLDM-2", "AudioGen"], value="AudioLDM-2")
-        submit_btn = gr.Button("Generate Sound Effect")
-        audio_o = gr.Audio(label="Audio Output")
-
+        gr.Markdown("### Image SFX Generator with Pexels Image Search")
+        search_query = gr.Textbox(label="사진 검색")
+        search_btn = gr.Button("검색")
+        images_output = gr.Gallery(label="검색 결과 이미지")
+        
+    search_btn.click(
+        fn=show_search_results,
+        inputs=search_query,
+        outputs=images_output
+    )
+app.launch(debug=True)
+
+with gr.Blocks(css=css) as demo:
+    with gr.Column(elem_id="col-container"):
+        gr.HTML("""
+        <h2 style="text-align: center;">
+            Image to SFX
+        </h2>
+        <p style="text-align: center;">
+            Compare MAGNet, AudioLDM2 and AudioGen sound effects generation from image caption.
+        </p>
+        """)
+        
+        with gr.Column():
+            image_in = gr.Image(sources=["upload"], type="filepath", label="Image input", value="oiseau.png")
+            with gr.Row():
+                chosen_model = gr.Radio(label="Choose a model", choices=["MAGNet", "AudioLDM-2", "AudioGen"], value="AudioLDM-2")
+                submit_btn = gr.Button("Submit")
+        with gr.Column():
+            audio_o = gr.Audio(label="Audio output")
+    
     submit_btn.click(
         fn=infer,
         inputs=[image_in, chosen_model],
-        outputs=audio_o
+        outputs=[audio_o]
     )
 
-# launch 메소드 호출 시 examples 파라미터를 사용하여 이미지 예제 추가
-demo.launch(examples=image_examples, debug=True)
+demo.queue(max_size=10).launch(debug=True)