135 lines
5.0 KiB
Python
135 lines
5.0 KiB
Python
import os
|
|
import shutil
|
|
from datetime import datetime
|
|
import google.generativeai as genai
|
|
import PIL.Image
|
|
import time
|
|
|
|
# --------------------- CONFIGURATION ---------------------
|
|
photo_folder = r"C:\Users\Antoine\PycharmProjects\PHOTO\photo a organiser"
|
|
dossier_folder = r"C:\Users\Antoine\PycharmProjects\PHOTO\dossier"
|
|
|
|
genai.configure(api_key="AIzaSyDq2LmX_fwKGAwxGAtmBfX940vT2wDQzBU")
|
|
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:
|
|
base = os.path.splitext(filename)[0]
|
|
parts = base.split("_")
|
|
time_part = parts[-1]
|
|
return datetime.strptime(time_part, "%H%M%S")
|
|
except Exception:
|
|
return None
|
|
|
|
def analyze_group_photos(group_paths):
|
|
"""
|
|
Envoie les images et demande de lier les valeurs aux noms de fichiers.
|
|
"""
|
|
prompt = (
|
|
"Analyze these images one by one. For each image, if you see a 6-character Barcode "
|
|
"or a 5-character LCLC, identify it.\n"
|
|
"Return the results in this exact format for each relevant file:\n"
|
|
"FILENAME: [filename], TYPE: [BARCODE or LCLC], VALUE: [value]\n"
|
|
"If a file has nothing, don't list it."
|
|
)
|
|
|
|
content = [prompt]
|
|
|
|
# On ouvre les images (avec gestion de fermeture automatique)
|
|
images_to_close = []
|
|
for path in group_paths:
|
|
img = PIL.Image.open(path)
|
|
content.append(f"Filename: {os.path.basename(path)}")
|
|
content.append(img)
|
|
images_to_close.append(img)
|
|
|
|
try:
|
|
# Note: Utilisez 'gemini-1.5-flash' car 2.5 n'existe pas encore
|
|
response = model.generate_content(content)
|
|
res_text = response.text
|
|
print(f"--- Gemini Analysis ---\n{res_text}\n-----------------------")
|
|
|
|
# Fermeture des images pour libérer les fichiers
|
|
for i in images_to_close: i.close()
|
|
|
|
return res_text
|
|
except Exception as e:
|
|
print(f"Error calling Gemini: {e}")
|
|
return ""
|
|
|
|
# --------------------- 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))
|
|
|
|
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]
|
|
difference = abs((time_obj - current_group["reference_time"]).total_seconds())
|
|
if difference <= 60:
|
|
current_group["photos"].append(photo)
|
|
current_group["reference_time"] = time_obj
|
|
else:
|
|
groups.append({"photos": [photo], "reference_time": time_obj})
|
|
|
|
# --------------------- ANALYSE ET RENOMMAGE ---------------------
|
|
for i, group in enumerate(groups, start=1):
|
|
print(f"Analyzing Group {i}...")
|
|
group_paths = [os.path.join(photo_folder, p) for p in group["photos"]]
|
|
analysis_result = analyze_group_photos(group_paths)
|
|
|
|
group_folder_path = os.path.join(dossier_folder, str(i))
|
|
os.makedirs(group_folder_path, exist_ok=True)
|
|
|
|
# Créer un dictionnaire pour mapper Filename -> NouveauNom
|
|
# Exemple: {"IMG_123.jpg": "ABC123.jpg"}
|
|
filename_mapping = {}
|
|
|
|
# Parsing de la réponse de Gemini
|
|
for line in analysis_result.splitlines():
|
|
if "FILENAME:" in line and "VALUE:" in line:
|
|
try:
|
|
fname = line.split("FILENAME:")[1].split(",")[0].strip()
|
|
vtype = line.split("TYPE:")[1].split(",")[0].strip()
|
|
val = line.split("VALUE:")[1].strip()
|
|
|
|
# On vérifie la validité des données
|
|
if (vtype == "BARCODE" and len(val) == 6) or (vtype == "LCLC" and len(val) == 5):
|
|
filename_mapping[fname] = val
|
|
except:
|
|
continue
|
|
|
|
for photo_name in group["photos"]:
|
|
src_path = os.path.join(photo_folder, photo_name)
|
|
extension = os.path.splitext(photo_name)[1]
|
|
|
|
# On vérifie si Gemini a trouvé une valeur spécifique pour CE fichier
|
|
if photo_name in filename_mapping:
|
|
new_name = f"{filename_mapping[photo_name]}{extension}"
|
|
else:
|
|
# Sinon on garde le nom original
|
|
new_name = photo_name
|
|
|
|
dst_path = os.path.join(group_folder_path, new_name)
|
|
|
|
# Gestion des doublons (si deux photos ont le même code)
|
|
counter = 1
|
|
base_name = os.path.splitext(new_name)[0]
|
|
while os.path.exists(dst_path):
|
|
dst_path = os.path.join(group_folder_path, f"{base_name}_{counter}{extension}")
|
|
counter += 1
|
|
|
|
shutil.move(src_path, dst_path) |