Add RVC code

app.py

@@ -12,10 +12,21 @@ client = InferenceClient(
     "meta-llama/Meta-Llama-3-8B-Instruct", token=os.getenv("hf_token")
 )
 
-# def download(filename):
-#     object_store_url = os.getenv("OBJECT_STORE")
-#     username = os.getenv("USERNAME")
-#     password = os.getenv("PASSWORD")
+
+def download(filename, directory):
+    object_store_url = os.getenv("OBJECT_STORE")
+    username = os.getenv("USERNAME")
+    password = os.getenv("PASSWORD")
+
+    download_url = f"{object_store_url}{directory}/{filename}"
+    response = requests.get(download_url, auth=(username, password))
+    if response.status_code == 200:
+        with open(filename, "wb") as file:
+            file.write(response.content)
+        print("File downloaded successfully")
+    else:
+        print(f"Failed to download file. Status code: {response.status_code}")
+        print(response.text)
 
 
 def process_audio(audio_data):
@@ -135,6 +146,9 @@ def respond(message, history: list[tuple[str, str]]):
     # elif "music" in decision_response:
     elif "music" in decision_response:
         responded_answer = message
+        download("SiriVT.pth", "Siri")
+        download("added_IVF617_Flat_nprobe_1_SiriVT_v1.index", "Siri")
+        download("The_Lion_King_Hakuna_Matata.mp4", "songs")
 
     else:
         # others
@@ -181,33 +195,25 @@ def tts_part():
         result = client.predict(text, audio=file("siri.wav"), api_name="/predict")
         llama_responded = 0
         responded_answer = ""
-    # else:
-    #     music_client = Client("r3gm/rvc_zero")
-    #     result_list = music_client.predict(
-    #         audio_files=[
-    #             file(
-    #                 "https://github.com/gradio-app/gradio/raw/main/test/test_files/sample_file.pdf"
-    #             )
-    #         ],
-    #         file_m=file(
-    #             "https://github.com/gradio-app/gradio/raw/main/test/test_files/sample_file.pdf"
-    #         ),
-    #         pitch_alg="rmvpe+",
-    #         pitch_lvl=0,
-    #         file_index=file(
-    #             "https://github.com/gradio-app/gradio/raw/main/test/test_files/sample_file.pdf"
-    #         ),
-    #         index_inf=0.75,
-    #         r_m_f=3,
-    #         e_r=0.25,
-    #         c_b_p=0.5,
-    #         active_noise_reduce=False,
-    #         audio_effects=False,
-    #         api_name="/run",
-    #     )
-    #     result = result_list[0]
-    #     llama_responded = 0
-    #     responded_answer = ""
+    else:
+        music_client = Client("r3gm/rvc_zero")
+        result_list = music_client.predict(
+            audio_files=[file("./The_Lion_King_Hakuna_Matata.mp4")],
+            file_m=file("./SiriVT.pth"),
+            pitch_alg="rmvpe+",
+            pitch_lvl=0,
+            file_index=file("./added_IVF617_Flat_nprobe_1_SiriVT_v1.index"),
+            index_inf=0.75,
+            r_m_f=3,
+            e_r=0.25,
+            c_b_p=0.5,
+            active_noise_reduce=False,
+            audio_effects=False,
+            api_name="/run",
+        )
+        result = result_list[0]
+        llama_responded = 0
+        responded_answer = ""
 
     return result