Update app.py

app.py

@@ -7,92 +7,7 @@ hide_streamlit_style = """
             </style>
             """
 st.markdown(hide_streamlit_style, unsafe_allow_html=True)
-def paid_version():
-    import os
-    import argparse
-    import shutil
-    from langchain.document_loaders import YoutubeLoader
-    from langchain.text_splitter import RecursiveCharacterTextSplitter
-    from langchain.vectorstores import Chroma
-    from langchain.embeddings import OpenAIEmbeddings
-    from langchain.chains import RetrievalQA
-    from langchain.llms import OpenAI
-    import streamlit as st
-    from langchain.chat_models import ChatOpenAI
-    from urllib.parse import urlparse, parse_qs
-    def extract_video_id(youtube_url):
-        try:
-            parsed_url = urlparse(youtube_url)
-            query_params = parse_qs(parsed_url.query)
-            video_id = query_params.get('v', [None])[0]
-
-            return video_id
-        except Exception as e:
-            print(f"Error extracting video ID: {e}")
-            return None
-    def set_openAi_api_key(api_key: str):
-                st.session_state["OPENAI_API_KEY"] = api_key
-                os.environ['OPENAI_API_KEY'] = api_key
-    def openai_api_insert_component():
-                with st.sidebar:
-                    st.markdown(
-                        """
-                        ## Quick Guide 🚀
-                        1. Get started by adding your [OpenAI API key](https://platform.openai.com/account/api-keys) below🔑
-                        2. Easily input the video url
-                        3. Engage with the content - ask questions, seek answers💬
-                        """
-                    )
-
-                    api_key_input = st.text_input("Input your OpenAI API Key",
-                                                type="password",
-                                                placeholder="Format: sk-...",
-                                                help="You can get your API key from https://platform.openai.com/account/api-keys.")
-
-
-                    if api_key_input == "" or api_key_input is None:
-                            st.sidebar.caption("👆 :red[Please set your OpenAI API Key here]")
-
-
-                    st.caption(":green[Your API is not stored anywhere. It is only used to generate answers to your questions.]")
-
-                    set_openAi_api_key(api_key_input)
 
-    def launchpaidversion():
-            openai_api_insert_component()
-            os.environ['OPENAI_API_KEY'] = st.session_state['OPENAI_API_KEY']
-            st.title('MKG: Your Chat with Youtube Assistant')
-            
-
-            videourl = st.text_input("Insert The video URL")
-            query = st.text_input("Ask any question about the video")
-            if st.button("Submit Question", type="primary"):
-                video_id = extract_video_id(videourl)
-                loader = YoutubeLoader(video_id)
-                documents = loader.load()
-
-                text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
-                documents = text_splitter.split_documents(documents)
-
-                shutil.rmtree('./data')
-                vectordb = Chroma.from_documents(
-                    documents,
-                    embedding=OpenAIEmbeddings(),
-                    persist_directory='./data'
-            )
-                vectordb.persist()
-
-                qa_chain = RetrievalQA.from_chain_type(
-                    llm=ChatOpenAI(model_name='gpt-3.5-turbo'),
-                    retriever=vectordb.as_retriever(),
-                    return_source_documents=True,
-                    verbose=False
-                )
-                response = qa_chain(query)
-                st.write(response)
-
-
-    launchpaidversion()
 def free_version():
           import torch
           import os
@@ -249,9 +164,7 @@ def intro():
             """)
 page_names_to_funcs = {
     "Main Page": intro,
-    "Open Source Edition (Free version)": free_version,
-    "Premium edition (Requires Open AI API Key )": paid_version
-
+    "Open Source Edition (Free version)": free_version
 }