Update app.py

app.py

@@ -5,6 +5,21 @@ from gradio_client import Client  # 가정: gradio_client 라이브러리가 사
 # 이미지 인식 파이프라인 로드
 image_model = pipeline("image-classification", model="google/vit-base-patch16-224")
 
+def generate_music(prompt):
+    # 음악 생성 API 호출
+    client = Client("https://haoheliu-audioldm2-text2audio-text2music.hf.space/")
+    result = client.predict(
+        prompt,
+        prompt,  # 음성 생성에 사용될 프롬프트
+        5,  # 음악의 길이 (초)
+        0,  # 가이던스 스케일
+        5,  # 시드 값
+        1,  # 생성할 waveform의 수
+        fn_index=1  # 함수 인덱스
+    )
+    # API 호출 결과 처리
+    return result
+
 def generate_voice(prompt):
     # Tango API를 사용하여 음성 생성
     client = Client("https://declare-lab-tango.hf.space/")
@@ -25,10 +40,12 @@ def classify_and_generate_voice(uploaded_image):
     
     # 음성 생성
     voice_result = generate_voice(top_prediction)
+    # 음악 생성
+    music_result = generate_music("Generate music for: " + top_prediction)
     
-    # 반환된 음성 결과를 Gradio 인터페이스로 전달
+    # 반환된 음성 및 음악 결과를 Gradio 인터페이스로 전달
     # 예: voice_result['url'] 또는 voice_result['audio_data'] 등
-    return top_prediction, voice_result
+    return top_prediction, voice_result, music_result
     
 # Gradio 인터페이스 생성
 iface = gr.Interface(