Update pages/lastimev4.py

pages/lastimev4.py

@@ -117,7 +117,7 @@ async def download_images(bbox, width, height, available_years):
         tasks = [fetch_image(session, get_wms_url(bbox, width, height, date), date, semaphore) for date in available_years]
         return await asyncio.gather(*tasks)
 
-def process_images_stream(images, format_option, speed, temp_dir, available_years, bbox, width, height):
+def process_images_stream(images, format_option, speed, temp_dir):
     results = {}
 
     if "GIF" in format_option:
@@ -136,31 +136,15 @@ def process_images_stream(images, format_option, speed, temp_dir, available_year
                     writer.append_data(np.array(img))
         results["MP4"] = mp4_path
 
-    if "Images individuelles (ZIP)" in format_option:
+    if "Individual Images (ZIP)" in format_option:
         zip_path = os.path.join(temp_dir, "images.zip")
         with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
-            info_file_content = []
-            for i, (img, year) in enumerate(zip(images, available_years)):
+            for i, img in enumerate(images):
                 if img is not None:
-                    # Conversion de l'image en mode RGB
-                    rgb_img = img.convert('RGB')
-                    # Sauvegarde de l'image en JPEG
-                    img_path = os.path.join(temp_dir, f"image_{year}.jpg")
-                    rgb_img.save(img_path, format="JPEG", quality=85)  # Ajustez la qualité selon vos besoins
+                    img_path = os.path.join(temp_dir, f"image_{i}.png")
+                    img.save(img_path)
                     zipf.write(img_path, os.path.basename(img_path))
-                    os.remove(img_path)
-
-                    # Création du lien WMS pour cette image
-                    wms_url = get_wms_url(bbox, width, height, year)
-                    info_file_content.append(f"{year}: {wms_url}")
-
-            # Ajout du fichier texte avec les informations
-            info_file_path = os.path.join(temp_dir, "image_info.txt")
-            with open(info_file_path, 'w') as f:
-                f.write("\n".join(info_file_content))
-            zipf.write(info_file_path, "image_info.txt")
-            os.remove(info_file_path)
-
+                    os.unlink(img_path)
         results["ZIP"] = zip_path
 
     return results
@@ -265,7 +249,7 @@ def app():
                     logger.info(f"Récupération réussie de {len(images)} images")
                     with tempfile.TemporaryDirectory() as temp_dir:
                         with st.spinner('Traitement des images en cours... Cela peut prendre un certain temps pour les grandes images.'):
-                            results = process_images_stream(images, format_option, speed, temp_dir, available_years, bbox, width, height)
+                            results = process_images_stream(images, format_option, speed, temp_dir)
 
                         for format, path in results.items():
                             if os.path.exists(path):
@@ -273,7 +257,7 @@ def app():
                                     st.success("Images individuelles (ZIP) créées avec succès!")
                                 else:
                                     st.success(f"Timelapse {format} créé avec succès!")
-                                st.markdown(get_binary_file_downloader_html(path, f'{"Images individuelles (ZIP)" if format == "ZIP" else f"Timelapse {format}"}'), unsafe_allow_html=True)
+                                st.markdown(get_binary_file_downloader_html(path, f'Timelapse {format if format != "ZIP" else "Images individuelles (ZIP)"}'), unsafe_allow_html=True)
                             else:
                                 st.error(f"Le fichier {format} n'a pas été créé avec succès.")
                 else: