Update app.py

app.py

@@ -41,7 +41,7 @@ def youtube_link_dataloader(video_link,max_video_length=1000):
 
 def youtube_chat(temperature=0.1,max_tokens=1096,API_key=None,llm_service='mistralai/Mistral-7B-v0.1',youtube_link=None,char_length=2000):
     
-    document  = youtube_link_dataloader(link=youtube_link,char_length=char_length)
+    video_document  = youtube_link_dataloader(link=youtube_link,char_length=char_length)
     print("docuemt:",document)
     
     if  llm_service== 'mistralai/Mistral-7B-v0.1':
@@ -53,11 +53,13 @@ def youtube_chat(temperature=0.1,max_tokens=1096,API_key=None,llm_service='mistr
                             )
     elif llm_service == 'OpenAI':
         llm = chatops.get_openai_chat_model(API_key=API_key)
-    elif llm_service == 'llama':
+    elif llm_service == 'llama3-8b-8192':
         os.environ["GROQ_API_KEY"] = API_key
         llm = ChatGroq(model="llama3-8b-8192")
 
-    return "Youtube link Processing completed ..."
+    summarize_chain = load_summarize_chain(llm=llm, chain_type='stuff', verbose = True )
+    results = summarize_chain.invoke({'input_documents':video_document})
+    return results['output_text']
 
 iface = gr.Interface(
     fn = youtube_chat,