Update pages/lecture_finder.py

pages/lecture_finder.py

@@ -1,123 +1,124 @@
-import streamlit as st
-import googleapiclient.discovery
-from dotenv import load_dotenv
-from datetime import timedelta
-
-# Load environment variables
-load_dotenv()
-
-# Set up YouTube API client
-api_service_name = "youtube"
-api_version = "v3"
-DEVELOPER_KEY = os.getenv('DEVELOPER_KEY')
-youtube = googleapiclient.discovery.build(api_service_name, api_version, developerKey=DEVELOPER_KEY)
-
-def search_youtube(query, max_results=50):
-    try:
-        request = youtube.search().list(
-            q=query,
-            type="video",
-            part="id,snippet",
-            maxResults=max_results,
-            fields="items(id(videoId),snippet(title,description,thumbnails))"
-        )
-        response = request.execute()
-        return response.get('items', [])
-    except googleapiclient.errors.HttpError as e:
-        st.error(f"An error occurred: {e}")
-        return []
-
-def get_video_details(video_id):
-    try:
-        request = youtube.videos().list(
-            part="contentDetails,statistics",
-            id=video_id,
-            fields="items(contentDetails(duration),statistics(viewCount))"
-        )
-        response = request.execute()
-        return response['items'][0] if response['items'] else None
-    except googleapiclient.errors.HttpError as e:
-        st.error(f"An error occurred while fetching video details: {e}")
-        return None
-
-def format_duration(duration):
-    duration = duration.replace('PT', '')
-    hours = 0
-    minutes = 0
-    seconds = 0
-    if 'H' in duration:
-        hours, duration = duration.split('H')
-        hours = int(hours)
-    if 'M' in duration:
-        minutes, duration = duration.split('M')
-        minutes = int(minutes)
-    if 'S' in duration:
-        seconds = int(duration.replace('S', ''))
-    return f"{hours:02d}:{minutes:02d}:{seconds:02d}"
-
-def parse_duration(duration_str):
-    parts = duration_str.split(':')
-    if len(parts) == 3:
-        return timedelta(hours=int(parts[0]), minutes=int(parts[1]), seconds=int(parts[2]))
-    elif len(parts) == 2:
-        return timedelta(minutes=int(parts[0]), seconds=int(parts[1]))
-    else:
-        return timedelta(seconds=int(parts[0]))
-
-def main():
-    st.set_page_config(page_title="S.H.E.R.L.O.C.K. Learning Assistant", page_icon="🕵️", layout="wide")
-    st.sidebar.title("S.H.E.R.L.O.C.K.")
-    st.sidebar.markdown("""
-    **S**ystematic **H**olistic **E**ducational **R**esource for **L**earning and **O**ptimizing **C**ognitive **K**nowledge
-    
-    Enhance your cognitive abilities, memory techniques, and subject-specific knowledge with AI-powered personalized learning.
-    """)
-    
-    query = st.sidebar.text_input("What would you like to learn about?", "")
-    
-    min_duration = st.sidebar.selectbox(
-        "Minimum video duration",
-        ["Any", "5:00", "10:00", "15:00", "30:00", "45:00", "1:00:00"],
-        index=0
-    )
-    
-    search_button = st.sidebar.button("Search for Learning Resources")
-    
-    st.title("Learning Resources")
-    
-    if search_button and query:
-        with st.spinner("Searching for the best learning resources..."):
-            results = search_youtube(query)
-            
-        if results:
-            filtered_results = []
-            for item in results:
-                video_id = item['id']['videoId']
-                video_details = get_video_details(video_id)
-                
-                if video_details:
-                    duration = video_details['contentDetails']['duration']
-                    formatted_duration = format_duration(duration)
-                    views = int(video_details['statistics']['viewCount'])
-                    
-                    if min_duration == "Any" or parse_duration(formatted_duration) >= parse_duration(min_duration):
-                        filtered_results.append((item, formatted_duration, views))
-                
-            if filtered_results:
-                for item, duration, views in filtered_results:
-                    col1, col2 = st.columns([1, 3])
-                    with col1:
-                        st.image(item['snippet']['thumbnails']['medium']['url'], use_column_width=True)
-                    with col2:
-                        st.markdown(f"### [{item['snippet']['title']}](https://www.youtube.com/watch?v={item['id']['videoId']})")
-                        st.markdown(f"**Duration:** {duration} | **Views:** {views:,}")
-                        st.markdown(item['snippet']['description'])
-                    
-                    st.markdown("---")
-            else:
-                st.warning("No results found matching your duration criteria. Try adjusting the minimum duration or search query.")
-        else:
-            st.warning("No results found. Please try a different search query.")
-
-if __name__ == "__main__":
+import streamlit as st
+import googleapiclient.discovery
+import os
+from dotenv import load_dotenv
+from datetime import timedelta
+
+# Load environment variables
+load_dotenv()
+
+# Set up YouTube API client
+api_service_name = "youtube"
+api_version = "v3"
+DEVELOPER_KEY = os.getenv('DEVELOPER_KEY')
+youtube = googleapiclient.discovery.build(api_service_name, api_version, developerKey=DEVELOPER_KEY)
+
+def search_youtube(query, max_results=50):
+    try:
+        request = youtube.search().list(
+            q=query,
+            type="video",
+            part="id,snippet",
+            maxResults=max_results,
+            fields="items(id(videoId),snippet(title,description,thumbnails))"
+        )
+        response = request.execute()
+        return response.get('items', [])
+    except googleapiclient.errors.HttpError as e:
+        st.error(f"An error occurred: {e}")
+        return []
+
+def get_video_details(video_id):
+    try:
+        request = youtube.videos().list(
+            part="contentDetails,statistics",
+            id=video_id,
+            fields="items(contentDetails(duration),statistics(viewCount))"
+        )
+        response = request.execute()
+        return response['items'][0] if response['items'] else None
+    except googleapiclient.errors.HttpError as e:
+        st.error(f"An error occurred while fetching video details: {e}")
+        return None
+
+def format_duration(duration):
+    duration = duration.replace('PT', '')
+    hours = 0
+    minutes = 0
+    seconds = 0
+    if 'H' in duration:
+        hours, duration = duration.split('H')
+        hours = int(hours)
+    if 'M' in duration:
+        minutes, duration = duration.split('M')
+        minutes = int(minutes)
+    if 'S' in duration:
+        seconds = int(duration.replace('S', ''))
+    return f"{hours:02d}:{minutes:02d}:{seconds:02d}"
+
+def parse_duration(duration_str):
+    parts = duration_str.split(':')
+    if len(parts) == 3:
+        return timedelta(hours=int(parts[0]), minutes=int(parts[1]), seconds=int(parts[2]))
+    elif len(parts) == 2:
+        return timedelta(minutes=int(parts[0]), seconds=int(parts[1]))
+    else:
+        return timedelta(seconds=int(parts[0]))
+
+def main():
+    st.set_page_config(page_title="S.H.E.R.L.O.C.K. Learning Assistant", page_icon="🕵️", layout="wide")
+    st.sidebar.title("S.H.E.R.L.O.C.K.")
+    st.sidebar.markdown("""
+    **S**ystematic **H**olistic **E**ducational **R**esource for **L**earning and **O**ptimizing **C**ognitive **K**nowledge
+    
+    Enhance your cognitive abilities, memory techniques, and subject-specific knowledge with AI-powered personalized learning.
+    """)
+    
+    query = st.sidebar.text_input("What would you like to learn about?", "")
+    
+    min_duration = st.sidebar.selectbox(
+        "Minimum video duration",
+        ["Any", "5:00", "10:00", "15:00", "30:00", "45:00", "1:00:00"],
+        index=0
+    )
+    
+    search_button = st.sidebar.button("Search for Learning Resources")
+    
+    st.title("Learning Resources")
+    
+    if search_button and query:
+        with st.spinner("Searching for the best learning resources..."):
+            results = search_youtube(query)
+            
+        if results:
+            filtered_results = []
+            for item in results:
+                video_id = item['id']['videoId']
+                video_details = get_video_details(video_id)
+                
+                if video_details:
+                    duration = video_details['contentDetails']['duration']
+                    formatted_duration = format_duration(duration)
+                    views = int(video_details['statistics']['viewCount'])
+                    
+                    if min_duration == "Any" or parse_duration(formatted_duration) >= parse_duration(min_duration):
+                        filtered_results.append((item, formatted_duration, views))
+                
+            if filtered_results:
+                for item, duration, views in filtered_results:
+                    col1, col2 = st.columns([1, 3])
+                    with col1:
+                        st.image(item['snippet']['thumbnails']['medium']['url'], use_column_width=True)
+                    with col2:
+                        st.markdown(f"### [{item['snippet']['title']}](https://www.youtube.com/watch?v={item['id']['videoId']})")
+                        st.markdown(f"**Duration:** {duration} | **Views:** {views:,}")
+                        st.markdown(item['snippet']['description'])
+                    
+                    st.markdown("---")
+            else:
+                st.warning("No results found matching your duration criteria. Try adjusting the minimum duration or search query.")
+        else:
+            st.warning("No results found. Please try a different search query.")
+
+if __name__ == "__main__":
     main()