feat: choose llms model

app.py

@@ -22,8 +22,8 @@ def initialize_models():
     return processing, drake
 
 
-def disable_sidebar():
-    st.set_page_config(page_title="Upload",
+def disable_sidebar(page_title: str):
+    st.set_page_config(page_title=page_title,
                        page_icon=None,
                        layout="centered",
                        initial_sidebar_state="collapsed",
@@ -37,7 +37,7 @@ def disable_sidebar():
 
 
 def main():
-    disable_sidebar()
+    disable_sidebar("Drake")
     initialize_models()
     st.switch_page(initial_page)
 

pages/chat.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from app import disable_sidebar, initialize_models
 
-disable_sidebar()
+disable_sidebar("Drake | Chat")
 col1, col2= st.columns([3, 1.6])
 
 if "messages" not in st.session_state:

pages/upload_file.py

@@ -2,10 +2,10 @@ import streamlit as st
 import io
 import PyPDF2
 from app import disable_sidebar, initialize_models
-from model import DrakeLM
-from utilis import Processing
 
-disable_sidebar()
+
+disable_sidebar("Drake | Upload File")
+processing, drake = initialize_models()
 st.title('Drake')
 st.subheader('Learn without the mess of making notes!')
 st.divider()
@@ -16,6 +16,8 @@ if st.button("Youtube/Video URL"):
 st.subheader('Upload the file')
 uploaded_file = st.file_uploader(label="Choose a file", type=['pdf', 'doc'])
 allow_make_notes = st.toggle('Make Complete Notes!')
+llm_model = st.selectbox('Choose LLM Model', ('gemini-pro', 'llama', 'Mobile phone'))
+drake.llm_model = llm_model
 
 
 if uploaded_file:
@@ -31,7 +33,6 @@ if uploaded_file:
                 for page in pdf_reader.pages:
                     text += page.extract_text()
 
-                processing, drake = initialize_models()
                 documents, metadata = processing.load_pdf("hello world", text)
                 st.session_state["metadata"] = metadata
                 st.success("Successfully chunked the file")

pages/upload_url.py

@@ -1,11 +1,8 @@
 import streamlit as st
-
 from app import disable_sidebar, initialize_models
-from model import DrakeLM
-from utilis import Processing
 
 # Upload Template
-disable_sidebar()
+disable_sidebar("Drake | Upload URL")
 processing, drake = initialize_models()
 st.title('Drake')
 st.subheader('Learn without the mess of making notes!')
@@ -16,6 +13,9 @@ if st.button("PDF/Transcript"):
 
 st.subheader('Enter the Video URL')
 video_url = st.text_input(label="Enter the URL")
+llm_model = st.selectbox('Choose LLM Model', ('gemini-pro', 'llama', 'Mobile phone'))
+drake.llm_model = llm_model
+
 allow_make_notes = st.toggle('Make Complete Notes!')
 
 
@@ -34,11 +34,11 @@ if video_url:
                 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")
+            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: