Update app.py

app.py

@@ -191,13 +191,13 @@ RESPUESTA: """
             st.markdown(prompt)
         st.session_state.messages.append({"role": "user", "content": prompt})
         with st.chat_message("assistant"):
-            llm_response = qa_chain(prompt)
+            llm_response, cleaned_prompt = qa_chain(prompt)
             llm_response = utils.process_llm_response(llm_response, nlp)
             st.markdown(llm_response)
             start_time_str_list = []; start_time_seconds_list = []; end_time_seconds_list = []
             for response in llm_response.split('\n'):
                 if re.search(r'(\d{2}:\d{2}:\d{2}(.\d{6})?)', response) != None:
-                    start_time_str, start_time_seconds, _, end_time_seconds = utils.add_hyperlink_and_convert_to_seconds(response)
+                    start_time_str, start_time_seconds, _, end_time_seconds = utils.add_hyperlink_and_convert_to_seconds(response, cleaned_prompt)
                     start_time_str_list.append(start_time_str)
                     start_time_seconds_list.append(start_time_seconds)
                     end_time_seconds_list.append(end_time_seconds)