import os
import shutil
from datetime import datetime
import google.generativeai as genai
import PIL.Image

# --------------------- CONFIGURATION ---------------------
photo_folder = r"C:\Users\Antoine\PycharmProjects\PHOTO\photo a organiser"
dossier_folder = r"C:\Users\Antoine\PycharmProjects\PHOTO\dossier"

# REMPLACER PAR VOTRE CLÉ
genai.configure(api_key="AIzaSyDq2LmX_fwKGAwxGAtmBfX940vT2wDQzBU")

# Gemini 1.5 Flash is excellent for OCR/Barcodes
model = genai.GenerativeModel('models/gemini-2.5-flash') 

if not os.path.exists(dossier_folder):
    os.makedirs(dossier_folder)

# --------------------- FONCTIONS ---------------------
def get_time_from_filename(filename):
    try:
        # Supposant un format : QuelqueChose_123045.jpg (HHMMSS)
        base = os.path.splitext(filename)[0]
        parts = base.split("_")
        time_part = parts[-1] # On prend la dernière partie avant l'extension
        return datetime.strptime(time_part, "%H%M%S")
    except Exception:
        return None

def read_barcode(photo_path):
    """
    Détecte le contenu d'un code-barre. 
    Retourne la valeur ou 'inconnu' s'il n'y en a pas.
    """
    try:
        with PIL.Image.open(photo_path) as img:
            response = model.generate_content([
                "Retourne uniquement le texte ou les chiffres du code-barre présent sur l'image. "
                "Si aucun code-barre n'est visible, répond 'inconnu'.", 
                img
            ])
            res = response.text.strip().lower()
            if "inconnu" in res or len(res) > 50: # Sécurité si le modèle divague
                return "inconnu"
            return response.text.strip().replace(" ", "")
    except Exception as e:
        print(f"Erreur lors de la lecture de {photo_path} : {e}")
        return "inconnu"

# --------------------- TRAITEMENT ---------------------
photos_raw = [f for f in os.listdir(photo_folder) if f.lower().endswith((".jpg", ".jpeg", ".png"))]
photo_times = []

for photo in photos_raw:
    time_obj = get_time_from_filename(photo)
    if time_obj:
        photo_times.append((photo, time_obj))

# Trier par heure
photo_times.sort(key=lambda x: x[1])

groups = []
for photo, time_obj in photo_times:
    if not groups:
        groups.append({"photos": [photo], "reference_time": time_obj})
    else:
        current_group = groups[-1]
        # On compare avec la dernière photo du groupe actuel
        difference = abs((time_obj - current_group["reference_time"]).total_seconds())
        
        if difference <= 60:
            current_group["photos"].append(photo)
            current_group["reference_time"] = time_obj # Update reference to the latest photo
        else:
            groups.append({"photos": [photo], "reference_time": time_obj})

# --------------------- DÉPLACEMENT ET SOUS-DOSSIERS ---------------------
for i, group in enumerate(groups, start=1):
    # Dossier principal pour le groupe (ex: Dossier_1)
    group_folder_name = f"{i}"
    group_folder_path = os.path.join(dossier_folder, group_folder_name)
    os.makedirs(group_folder_path, exist_ok=True)
    
    print(f"Traitement du groupe {i} ({len(group['photos'])} photos)...")

    for photo_name in group["photos"]:
        src_path = os.path.join(photo_folder, photo_name)
        
        # Vérifier si CETTE photo est un code-barre
        barcode_value = read_barcode(src_path)
        
        if barcode_value != "inconnu":
            # Nettoyage du nom pour Windows
            barcode_clean = "".join(c for c in barcode_value if c.isalnum() or c in "-_").strip()
            
            # Création du SOUS-DOSSIER dans le dossier du groupe
            subfolder_path = os.path.join(group_folder_path, barcode_clean)
            os.makedirs(subfolder_path, exist_ok=True)
            
            dst_path = os.path.join(subfolder_path, photo_name)
            print(f"  -> Code-barre détecté ({barcode_clean}), déplacé dans sous-dossier.")
        else:
            # Photo normale, reste à la racine du dossier groupe
            dst_path = os.path.join(group_folder_path, photo_name)

        shutil.move(src_path, dst_path)

print("\nOpération terminée avec succès !")