import os
import requests

def download_images(url_list, output_folder):
    # Kreiranje izlaznog foldera ako ne postoji
    os.makedirs(output_folder, exist_ok=True)

    # Iteriranje kroz listu URL-ova
    for i, url in enumerate(url_list):
        try:
            # Preuzimanje slike sa URL-a
            response = requests.get(url)
            if response.status_code == 200:
                # Extrakcija naziva datoteke iz URL-a
                filename = url.split("/")[-1]
                # Razdvajanje naziva datoteke na osnovu nota
                note = filename.split("_")[0]
                # Kreiranje podfoldera za notu ako ne postoji
                note_folder = os.path.join(output_folder, note)
                os.makedirs(note_folder, exist_ok=True)
                # Čuvanje slike u odgovarajućem podfolderu
                with open(os.path.join(note_folder, filename), "wb") as f:
                    f.write(response.content)
                print(f"Slika {i+1} uspešno preuzeta i sačuvana u {note_folder}.")
            else:
                print(f"Greska prilikom preuzimanja slike {i+1}. Status kod: {response.status_code}")
        except Exception as e:
            print(f"Greska prilikom preuzimanja slike {i+1}: {str(e)}")

# Lista URL-ova sa kojih ćemo preuzimati slike (primer)
url_list = [
    "https://example.com/do_image1.jpg",
    "https://example.com/re_image1.jpg",
    "https://example.com/mi_image1.jpg",
    # Dodajte ostale URL-ove prema potrebi
]

# Folder u koji ćemo sačuvati preuzete slike
output_folder = "dataset"

# Pozivanje funkcije za preuzimanje slika
download_images(url_list, output_folder)