Update app.py

app.py

@@ -312,7 +312,60 @@ def make_nasa_soho_image_trigger():
     #html_in+= make_nasa_soho_videos()
     html_in+="</div>"
     return html_in
-def count_days(date1,date2):
+from datetime import date, timedelta, datetime
+
+def generate_sdo_urls(date1_str: str, date2_str: str) -> list[str]:
+    """
+    Generates a list of SDO browse image URLs for all days
+    inclusive of date1_str and date2_str.
+
+    The URL format is assumed to be:
+    https://sdo.gsfc.nasa.gov/assets/img/browse/YYYY/MM/DD/
+
+    Args:
+        date1_str: Start date in "YYYYMMDD" format (e.g., "20230115").
+        date2_str: End date in "YYYYMMDD" format (e.g., "20230117").
+
+    Returns:
+        A list of URL strings.
+        Returns an empty list if date1_str > date2_str or if dates are invalid.
+    """
+    url_box = []
+    base_url = "https://sdo.gsfc.nasa.gov/assets/img/browse"
+
+    try:
+        # Convert input strings to date objects
+        # datetime.strptime creates a datetime object, .date() gets just the date part
+        start_date = datetime.strptime(date1_str, "%Y%m%d").date()
+        end_date = datetime.strptime(date2_str, "%Y%m%d").date()
+    except ValueError:
+        print(f"Error: Invalid date format. Please use YYYYMMDD. Received: '{date1_str}', '{date2_str}'")
+        return [] # Return empty list for invalid date format
+
+    # Handle cases where start_date is after end_date
+    if start_date > end_date:
+        print(f"Warning: Start date ({date1_str}) is after end date ({date2_str}). Returning an empty list.")
+        return []
+
+    current_date = start_date
+    while current_date <= end_date:
+        # Format year, month, and day with leading zeros if necessary
+        # .year, .month, .day are integers
+        year_str = f"{current_date.year:04}"  # Ensures YYYY format (though usually not needed for year)
+        month_str = f"{current_date.month:02}" # Ensures MM format (e.g., 1 -> 01)
+        day_str = f"{current_date.day:02}"   # Ensures DD format (e.g., 5 -> 05)
+
+        # Construct the URL
+        out_url = f"{base_url}/{year_str}/{month_str}/{day_str}/"
+        url_box.append(out_url)
+
+        # Move to the next day
+        current_date += timedelta(days=1)
+
+    return url_box
+
+
+def count_days_og(date1,date2):
     url_box=[]
     in_year=f"{date1[0:4]}"
     in_month=f"{date1[4:6]}"
@@ -364,6 +417,7 @@ def count_days(date1,date2):
 
     
 def nasa_sdo_images(obj,size,date1,date2):
+    print(generate_sdo_urls(date1,date2))
     url_box=[]
     html_in=f"<style>{css}</style><div class='img_box_soho'>"
     print(date1 + " - " + date2)