Update pages/3_ChannelAgregator.py

pages/3_ChannelAgregator.py

@@ -17,21 +17,25 @@ def extract_channel_videos(channel_id, api_key):
     video_ids = []
     next_page_token = None
 
-    while True:
-        request = youtube.search().list(
-            part="id",
-            channelId=channel_id,
-            maxResults=50,  # You can fetch up to 50 results per request
-            type="video",
-            order="date",  # Newest videos first
-            pageToken=next_page_token
-        )
-        response = request.execute()
-        video_ids += [item['id']['videoId'] for item in response.get('items', [])]
+    try:
+        while True:
+            request = youtube.search().list(
+                part="id",
+                channelId=channel_id,
+                maxResults=50,  # You can fetch up to 50 results per request
+                type="video",
+                order="date",  # Newest videos first
+                pageToken=next_page_token
+            )
+            response = request.execute()
+            video_ids += [item['id']['videoId'] for item in response.get('items', [])]
 
-        next_page_token = response.get('nextPageToken')
-        if not next_page_token:
-            break
+            next_page_token = response.get('nextPageToken')
+            if not next_page_token:
+                break
+    except Exception as e:
+        st.error(f"Failed to fetch video IDs: {str(e)}")
+        return []
 
     return video_ids
 
@@ -54,7 +58,7 @@ st.title("YouTube Channel Transcription Extractor")
 api_key = st.text_input("Enter your Google API Key", type="password")
 channel_id = st.text_input("Enter YouTube Channel ID", placeholder="Example: UCK8sQmJBp8GCxrOtXWBpyEA")
 if st.button("Fetch Video List"):
-    if api_key:
+    if api_key and channel_id:
         video_ids = extract_channel_videos(channel_id, api_key)
         video_ids_string = ", ".join(video_ids)
         edited_videos = st.text_area("Edit Video IDs (comma-separated):", value=video_ids_string, height=100)
@@ -64,4 +68,4 @@ if st.button("Fetch Video List"):
             st.text_area("Transcriptions:", value=result, height=300)
             st.download_button("Download Transcripts", data=result, file_name="transcripts.txt", mime="text/plain")
     else:
-        st.error("Please enter a valid Google API Key.")
+        st.error("Please enter a valid Google API Key and YouTube Channel ID.")