Update app.py

app.py

@@ -30,22 +30,20 @@ RAG_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"],
 CHROMA_DIR  = "/data/chroma"
 YOUTUBE_DIR = "/data/youtube"
 
-YOUTUBE_URL = "https://www.youtube.com/watch?v=--khbXchTeE"
-#YOUTUBE_URL = "https://www.youtube.com/watch?v=RfvL_423a-I&list=PL2yQDdvlhXf_hIzmfHCdbcXj2hS52oP9r&index=2"
+#YOUTUBE_URL = "https://www.youtube.com/watch?v=--khbXchTeE"
+YOUTUBE_URL = "https://www.youtube.com/watch?v=Iy1IpvcJH7I&list=PL2yQDdvlhXf9XsB2W76_seM6dJxcE2Pdc&index=2"
 
 MODEL_NAME  = "gpt-4"
 
 def invoke(openai_api_key, use_rag, prompt):
-    print(os.listdir("../"))
-    print(os.listdir("../app/"))
     llm = ChatOpenAI(model_name = MODEL_NAME, 
                      openai_api_key = openai_api_key, 
                      temperature = 0)
     if (use_rag):
-        if (os.path.isdir(CHROMA_DIR)):
-            vector_db = Chroma(embedding_function = OpenAIEmbeddings(),
-                               persist_directory = CHROMA_DIR)
-            print("Load DB")
+#        if (os.path.isdir(CHROMA_DIR)):
+#            vector_db = Chroma(embedding_function = OpenAIEmbeddings(),
+#                               persist_directory = CHROMA_DIR)
+#            print("Load DB")
         else:
             loader = GenericLoader(YoutubeAudioLoader([YOUTUBE_URL], YOUTUBE_DIR), 
                                    OpenAIWhisperParser())
@@ -71,8 +69,8 @@ def invoke(openai_api_key, use_rag, prompt):
     return result
 
 description = """<strong>Overview:</strong> The app demonstrates how to use a Large Language Model (LLM) with Retrieval Augmented Generation (RAG) on external data 
-                 (in this case a YouTube video, but it could be PDFs, URLs, or other 
-                 <a href='https://raw.githubusercontent.com/bstraehle/ai-ml-dl/c38b224c196fc984aab6b6cc6bdc666f8f4fbcff/langchain/document-loaders.png'>data sources</a>).\n\n
+                 (in this case the <a href='https://www.youtube.com/playlist?list=PL2yQDdvlhXf9XsB2W76_seM6dJxcE2Pdc'>AWS re:Invent 2022 - AI/ML YouTube playlist</a>, 
+                 but it could be PDFs, URLs, or other <a href='https://raw.githubusercontent.com/bstraehle/ai-ml-dl/c38b224c196fc984aab6b6cc6bdc666f8f4fbcff/langchain/document-loaders.png'>data sources</a>).\n\n
                  <strong>Instructions:</strong> Enter an OpenAI API key and perform LLM use cases (semantic search, sentiment analysis, summarization, translation, etc.) on 
                  a <a href='https://www.youtube.com/watch?v=--khbXchTeE'>short video of GPT-4</a>.
                  <ul style="list-style-type:square;">