Spaces:
Sleeping
Sleeping
File size: 4,906 Bytes
5347681 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 |
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__":
main() |