imagebind2 / downloader.py
raghavd99's picture
data update
3ceda73
import requests
import os
from pathlib import Path
# URL of the raw audio file on GitHub
audio_file_urls = [
"https://github.com/raghavdixit99/assets/raw/main/bird_audio.wav",
"https://github.com/raghavdixit99/assets/raw/main/dragon-growl-37570.wav",
"https://github.com/raghavdixit99/assets/raw/main/car_audio.wav",
"https://github.com/raghavdixit99/assets/raw/main/acoustic-guitar.wav",
"https://github.com/raghavdixit99/assets/raw/main/witch.wav",
"https://github.com/raghavdixit99/assets/raw/main/thunder-25689.wav",
]
image_urls = [
"https://github.com/raghavdixit99/assets/assets/34462078/abf47cc4-d979-4aaa-83be-53a2115bf318",
"https://github.com/raghavdixit99/assets/assets/34462078/93be928e-522b-4e37-889d-d4efd54b2112",
"https://github.com/raghavdixit99/assets/assets/34462078/025deaff-632a-4829-a86c-3de6e326402f",
"https://github.com/raghavdixit99/assets/assets/34462078/a20bff32-155c-4bad-acf1-97856c493099",
"https://github.com/raghavdixit99/assets/assets/34462078/4f7dadd8-b38c-4c14-ac8a-5a2e74414f6a",
"https://github.com/raghavdixit99/assets/assets/34462078/ac11eeab-7b2b-4db3-981b-d5fed08d9bc2",
]
base_path = os.path.dirname(os.path.abspath(__file__))
# Local path where you want to save the .wav file
def dowload_and_save_audio():
audio_pths = []
for url in audio_file_urls:
filename = url.split("/")[-1]
local_file_path = Path(f"{base_path}/test_inputs/{filename}")
local_file_path.parent.mkdir(parents=True, exist_ok=True)
# Perform the GET request
response = requests.get(url)
# Check if the request was successful
if response.status_code == 200:
# Write the content of the response to a local file
with open(local_file_path, "wb") as audio_file:
audio_file.write(response.content)
audio_pths.append(str(local_file_path))
print(
f"Audio file downloaded successfully and saved as '{local_file_path}'."
)
else:
print(f"Failed to download file. Status code: {response.status_code}")
return audio_pths
def dowload_and_save_image():
image_paths = []
for url in image_urls:
filename = url.split("/")[-1]
local_file_path = Path(f"{base_path}/test_inputs/{filename}.jpeg")
local_file_path.parent.mkdir(parents=True, exist_ok=True)
# Perform the GET request
response = requests.get(url)
# Check if the request was successful
if response.status_code == 200:
# Write the content of the response to a local file
with open(local_file_path, "wb") as image_file:
image_file.write(response.content)
image_paths.append(str(local_file_path))
print(
f"Image file downloaded successfully and saved as '{local_file_path}'."
)
else:
print(f"Failed to download file. Status code: {response.status_code}")
return image_paths