Spaces:
Sleeping
Sleeping
File size: 2,209 Bytes
a828a8b f8db355 a828a8b |
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 |
import streamlit as st
from app import disable_sidebar, initialize_models
from model import DrakeLM
from utilis import Processing
# Upload Template
disable_sidebar()
processing, drake = initialize_models()
st.title('Drake')
st.subheader('Learn without the mess of making notes!')
st.divider()
if st.button("PDF/Transcript"):
st.switch_page("pages/upload_file.py")
st.subheader('Enter the Video URL')
video_url = st.text_input(label="Enter the URL")
allow_make_notes = st.toggle('Make Complete Notes!')
if video_url:
# Upload to DB
if st.button("Upload to DB"):
# Chunking the file
with st.spinner('Please wait, file is chunking ...'):
try:
documents, metadata = processing.load_yt_transcript(video_url)
st.session_state["metadata"] = {"id": metadata["id"]}
st.success("Successfully chunked the file")
except Exception as e:
print(e)
st.error("Error in chunking")
# Uploading to DB
# with st.spinner('Please wait, documents uploading ...'):
# try:
# processing.upload_to_db(documents)
# except Exception as e:
# st.error("Error in uploading")
# Generating Notes
if allow_make_notes:
with st.spinner('Please wait, notes are being generated ...'):
try:
config = {"max_new_tokens": 4096, "context_length": 8192, "temperature": 0.3}
notes = drake.create_notes(documents)
encoded_text = notes.encode('utf-8')
st.success("Notes generated successfully")
if st.download_button(
label="Download your notes",
data=encoded_text,
file_name='your_notes.md',
mime='text/markdown',
):
st.switch_page("pages/chat.py")
except Exception as e:
print(e)
st.error("Error in generating notes")
else:
st.switch_page("pages/chat.py")
|