Update pages/v2.py

pages/v2.py

@@ -18,7 +18,24 @@ from concurrent.futures import ThreadPoolExecutor
 from functools import lru_cache
 import numpy as np
 
-# Fonction pour charger les données de fichiers GeoJSON/KML
+# Liste complète des dates disponibles
+AVAILABLE_DATES = [
+    18641231, 18701231, 18801231, 18901231, 18941231, 18951231, 18961231, 18971231, 18981231, 18991231,
+    19001231, 19011231, 19021231, 19031231, 19041231, 19051231, 19061231, 19071231, 19081231, 19091231,
+    19101231, 19111231, 19121231, 19131231, 19141231, 19151231, 19161231, 19171231, 19181231, 19191231,
+    19201231, 19211231, 19221231, 19231231, 19241231, 19251231, 19261231, 19271231, 19281231, 19291231,
+    19301231, 19311231, 19321231, 19331231, 19341231, 19351231, 19361231, 19371231, 19381231, 19391231,
+    19401231, 19411231, 19421231, 19431231, 19441231, 19451231, 19461231, 19471231, 19481231, 19491231,
+    19501231, 19511231, 19521231, 19531231, 19541231, 19551231, 19561231, 19571231, 19581231, 19591231,
+    19601231, 19611231, 19621231, 19631231, 19641231, 19651231, 19661231, 19671231, 19681231, 19691231,
+    19701231, 19711231, 19721231, 19731231, 19741231, 19751231, 19761231, 19771231, 19781231, 19791231,
+    19801231, 19811231, 19821231, 19831231, 19841231, 19851231, 19861231, 19871231, 19881231, 19891231,
+    19901231, 19911231, 19921231, 19931231, 19941231, 19951231, 19961231, 19971231, 19981231, 19991231,
+    20001231, 20011231, 20021231, 20031231, 20041231, 20051231, 20061231, 20071231, 20081231, 20091231,
+    20101231, 20111231, 20121231, 20131231, 20141231, 20151231, 20161231, 20171231, 20181231, 20191231,
+    20201231, 20211231
+]
+
 @st.cache_data
 def uploaded_file_to_gdf(data):
     import tempfile
@@ -39,7 +56,6 @@ def uploaded_file_to_gdf(data):
 
     return gdf
 
-# Fonction pour construire l'URL de requête WMS
 @lru_cache(maxsize=128)
 def get_wms_url(bbox, width, height, time):
     url = "https://wms.geo.admin.ch/"
@@ -59,7 +75,6 @@ def get_wms_url(bbox, width, height, time):
     }
     return url + "?" + "&".join(f"{k}={v}" for k, v in params.items())
 
-# Fonction pour ajouter une date à l'image
 def add_date_to_image(image, date):
     draw = ImageDraw.Draw(image)
     font = ImageFont.load_default()
@@ -76,7 +91,6 @@ def add_date_to_image(image, date):
     draw.text((x, y), text, font=font, fill="white")
     return image
 
-# Fonction asynchrone pour télécharger une image
 async def fetch_image(session, url, date, semaphore):
     async with semaphore:
         try:
@@ -89,14 +103,12 @@ async def fetch_image(session, url, date, semaphore):
             print(f"Erreur lors de la récupération de l'image pour la date {date}: {str(e)}")
     return None
 
-# Téléchargement des images en fonction des années disponibles
 async def download_images(bbox, width, height, available_years):
-    semaphore = asyncio.Semaphore(20)  # Limité à 20 requêtes simultanées
+    semaphore = asyncio.Semaphore(20)
     async with aiohttp.ClientSession() as session:
         tasks = [fetch_image(session, get_wms_url(bbox, width, height, date), date, semaphore) for date in available_years]
         return await asyncio.gather(*tasks)
 
-# Traitement des images et génération des fichiers ZIP par lots
 def process_images_stream(images, format_option, speed, temp_dir, batch_size=20):
     results = {}
 
@@ -134,7 +146,6 @@ def process_images_stream(images, format_option, speed, temp_dir, batch_size=20)
 
     return results
 
-# Générer un lien de téléchargement pour les fichiers ZIP
 def get_binary_file_downloader_html(bin_file, file_label='File'):
     with open(bin_file, 'rb') as f:
         data = f.read()
@@ -142,7 +153,6 @@ def get_binary_file_downloader_html(bin_file, file_label='File'):
     href = f'<a href="data:application/octet-stream;base64,{bin_str}" download="{os.path.basename(bin_file)}">Télécharger {file_label}</a>'
     return href
 
-# Application Streamlit principale
 def app():
     st.title("Générateur de Timelapse Historique Suisse")
 
@@ -171,10 +181,9 @@ def app():
         )
 
         with st.form("submit_form"):
-            start_year = st.selectbox("Sélectionnez l'année de début:", [1864, 2021])
-            end_year = st.selectbox("Sélectionnez l'année de fin:", [1864, 2021], index=len([1864, 2021]) - 1)
+            start_year = st.selectbox("Sélectionnez l'année de début:", [date // 10000 for date in AVAILABLE_DATES])
+            end_year = st.selectbox("Sélectionnez l'année de fin:", [date // 10000 for date in AVAILABLE_DATES], index=len(AVAILABLE_DATES) - 1)
             
-            # Options de taille d'image
             size_choice = st.selectbox("Choisissez la taille de l'image:", ["HD (720p)", "Full HD (1080p)"])
             width, height = (1280, 720) if size_choice == "HD (720p)" else (1920, 1080)
             
@@ -187,7 +196,7 @@ def app():
         if data:
             gdf = uploaded_file_to_gdf(data)
             bbox = tuple(gdf.to_crs(epsg=2056).total_bounds)
-            available_years = [18641231, 20211231]  # Exemple d'années disponibles dans le code de l'application originale
+            available_years = [date for date in AVAILABLE_DATES if start_year <= date // 10000 <= end_year]
             
             images = asyncio.run(download_images(bbox, width, height, available_years))