Update app.py

app.py

@@ -36,7 +36,7 @@ st.markdown(DESCRIPTION)
 def get_video_title(youtube_url: str) -> str:
     yt = YouTube(youtube_url)
     embed_url = f"https://www.youtube.com/embed/{yt.video_id}"
-    embed_html = f'<iframe width="560" height="315" src="{embed_url}" frameborder="0" allowfullscreen></iframe>'
+    embed_html = f'<iframe  src="{embed_url}" frameborder="0" allowfullscreen></iframe>'
     return yt.title, embed_html
 
 
@@ -152,7 +152,7 @@ if st.session_state.youtube_url and not st.session_state.setup_done:
       retriever.search_kwargs['k'] = 4
     with st.status("Running RetrievalQA..."):
       llama_instance = LlamaLLM()
-      st.session_state.qa = RetrievalQA.from_chain_type(llm=llama_instance, chain_type="stuff", retriever=retriever,return_source_documents=True,chain_type_kwargs={"prompt": prompt})
+      st.session_state.qa = RetrievalQA.from_chain_type(llm=llama_instance, chain_type="stuff", retriever=retriever,chain_type_kwargs={"prompt": prompt})
         
     st.session_state.doneYoutubeurl = st.session_state.youtube_url
     st.session_state.setup_done = True  # Mark the setup as done for this URL
@@ -170,9 +170,8 @@ if prompt := textinput:
   st.chat_message("human",avatar = "🧑‍💻").markdown(prompt)
   st.session_state.messages.append({"role": "human", "content": prompt})
   with st.status("Requesting Client..."):
-      response = st.session_state.qa(
-                {"query": prompt})
+      response = st.session_state.qa.run(prompt)
   with st.chat_message("assistant", avatar='🦙'):
-      st.markdown(str(response["result"])+'\n'+str(response["source_documents"]))
+      st.markdown(response)
   # Add assistant response to chat history
   st.session_state.messages.append({"role": "assistant", "content": response})