Update app.py

app.py

@@ -25,10 +25,9 @@ QA_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"], temp
 YOUTUBE_DIR = "docs/youtube/" 
 CHROMA_DIR = "docs/chroma/"
 
-shutil.rmtree(CHROMA_DIR)
-
-def invoke(openai_api_key, youtube_url, prompt):
+def invoke(openai_api_key, youtube_url, prompt, process_video):
     openai.api_key = openai_api_key
+    print(process_video)
     if (os.path.isdir(CHROMA_DIR) == False):
         print(111)
         loader = GenericLoader(YoutubeAudioLoader([youtube_url], YOUTUBE_DIR), OpenAIWhisperParser())
@@ -53,7 +52,7 @@ description = """The app demonstrates how to use a <strong>Large Language Model<
 
 gr.close_all()
 demo = gr.Interface(fn=invoke, 
-                    inputs = [gr.Textbox(label = "OpenAI API Key", value = "sk-", lines = 1), gr.Textbox(label = "YouTube URL", value = "https://www.youtube.com/watch?v=--khbXchTeE", lines = 1), gr.Textbox(label = "Prompt", value = "GPT-4 human level performance", lines = 1)],
+                    inputs = [gr.Textbox(label = "OpenAI API Key", value = "sk-", lines = 1), gr.Textbox(label = "YouTube URL", value = "https://www.youtube.com/watch?v=--khbXchTeE", lines = 1), gr.Textbox(label = "Prompt", value = "GPT-4 human level performance", lines = 1), gr.Radio(["Yes", "No"], label="Process Video")],
                     outputs = [gr.Textbox(label = "Completion", lines = 1)],
                     title = "Generative AI - LLM & RAG",
                     description = description)