Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import requests
 import os
 import re
+from datetime import datetime, timedelta
 
 # Fetch the keys from the environment variable and convert them into a list
 YOUTUBE_API_KEYS = os.getenv("YOUTUBE_API_KEYS")
@@ -20,6 +21,21 @@ def get_api_key():
     key_index = (key_index + 1) % len(YOUTUBE_API_KEYS)  # Rotate to the next key
     return api_key
 
+def get_iso8601_date(time_frame):
+    now = datetime.utcnow()
+    if time_frame == "Last Hour":
+        return (now - timedelta(hours=1)).isoformat("T") + "Z"
+    elif time_frame == "Today":
+        return (now - timedelta(days=1)).isoformat("T") + "Z"
+    elif time_frame == "This Week":
+        return (now - timedelta(weeks=1)).isoformat("T") + "Z"
+    elif time_frame == "This Month":
+        return (now - timedelta(days=30)).isoformat("T") + "Z"
+    elif time_frame == "This Year":
+        return (now - timedelta(days=365)).isoformat("T") + "Z"
+    else:
+        return ""
+
 def youtube_search(query, upload_date, video_type, duration, sort_by, max_results=50):
     search_url = "https://www.googleapis.com/youtube/v3/search"
     all_results = []
@@ -33,11 +49,20 @@ def youtube_search(query, upload_date, video_type, duration, sort_by, max_result
 
     # Add filter parameters based on user selection
     if upload_date:
-        params["publishedAfter"] = upload_date
+        published_after = get_iso8601_date(upload_date)
+        if published_after:
+            params["publishedAfter"] = published_after
+
     if video_type and video_type != "All":
         params["type"] = video_type.lower()  # API expects lowercase values
+
     if duration:
-        params["videoDuration"] = duration  # Set duration parameter for filtering
+        duration_mapping = {
+            "Short (<4 mins)": "short",
+            "Medium (4-20 mins)": "medium",
+            "Long (>20 mins)": "long"
+        }
+        params["videoDuration"] = duration_mapping.get(duration, "any")
 
     try:
         while len(all_results) < max_results:
@@ -149,26 +174,7 @@ with gr.Blocks(css="""
             search_output = gr.Gallery(label="Search Results", columns=1, height="800px", elem_id="search_output")
 
     def update_search_results(query, upload_date, video_type, duration, sort_by):
-        # Map the filter values to YouTube API parameters
-        upload_date_mapping = {
-            "Last Hour": "now-1h",
-            "Today": "now-1d",
-            "This Week": "now-7d",
-            "This Month": "now-30d",
-            "This Year": "now-365d"
-        }
-
-        duration_mapping = {
-            "Short (<4 mins)": "short",
-            "Medium (4-20 mins)": "medium",
-            "Long (>20 mins)": "long"
-        }
-
-        # Convert user-friendly values to API-compatible parameters
-        upload_date_param = upload_date_mapping.get(upload_date, "")
-        duration_param = duration_mapping.get(duration, "")
-
-        search_results = youtube_search(query, upload_date_param, video_type, duration_param, sort_by)
+        search_results = youtube_search(query, upload_date, video_type, duration, sort_by)
         gallery_items = []
         video_ids = []
         for item in search_results: