Spaces:
Running
Running
| import streamlit as st | |
| import io | |
| import PyPDF2 | |
| from app import disable_sidebar, initialize_models | |
| disable_sidebar("Drake | Upload File") | |
| processing, drake = initialize_models() | |
| st.title('Drake') | |
| st.subheader('Learn without the mess of making notes!') | |
| st.divider() | |
| if st.button("Youtube/Video URL"): | |
| st.switch_page("pages/upload_url.py") | |
| st.subheader('Upload the file') | |
| uploaded_file = st.file_uploader(label="Choose a file", type=['pdf', 'doc']) | |
| is_scanned = st.toggle("Is the file scanned?") | |
| allow_make_notes = st.toggle('Make Complete Notes!') | |
| st.caption("Note: Currently, Drake support Gemini, Llama support to be added soon!") | |
| if uploaded_file: | |
| if st.button("Upload to DB"): | |
| # Chunking the file | |
| with st.spinner('Please wait, file is chunking ...'): | |
| try: | |
| pdf_stream = io.BytesIO(uploaded_file.getvalue()) | |
| if is_scanned: | |
| text = processing.load_scanned_pdf(uploaded_file.getvalue()) | |
| else: | |
| pdf_reader = PyPDF2.PdfReader(pdf_stream) | |
| text = "" | |
| for page in pdf_reader.pages: | |
| text += page.extract_text() | |
| documents, metadata = processing.load_pdf(text) | |
| st.session_state["metadata"] = metadata | |
| except Exception as e: | |
| st.error("Error in chunking") | |
| # Uploading to DB | |
| with st.spinner('Please wait, documents uploading ...'): | |
| try: | |
| processing.upload_to_db(documents) | |
| st.success("Successfully uploaded the file") | |
| 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: | |
| st.error("Error in generating notes", e) | |
| else: | |
| st.switch_page("pages/chat.py") | |