Update app.py

app.py

@@ -194,26 +194,26 @@ RESPUESTA: """
             else:
                 if prompt.lower() == 'resume':
                     llm_response = utils.summarise_doc(TRANSCRIPTION_PATH)
+                    st.markdown(llm_response)
                 else:
                     llm_response = utils.get_gpt_response(TRANSCRIPTION_PATH, prompt, logger)
-            
-            llm_response = utils.process_llm_response(llm_response)
-            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_list.append(start_time_str)
-                    start_time_seconds_list.append(start_time_seconds)
-                    end_time_seconds_list.append(end_time_seconds)
-
-            if start_time_str_list:
-                for start_time_seconds, start_time_str, end_time_seconds in zip(start_time_seconds_list, start_time_str_list, end_time_seconds_list):
-                    st.markdown("__Fragmento: " + start_time_str + "__")
-                    _, container, _ = st.columns([SIDE, WIDTH, SIDE])
-                    with container:
-                        st_player(youtube_video_url.replace("?enablejsapi=1", "") + f'?start={start_time_seconds}&end={end_time_seconds}')
+                    llm_response = utils.process_llm_response(llm_response)
+                    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_list.append(start_time_str)
+                            start_time_seconds_list.append(start_time_seconds)
+                            end_time_seconds_list.append(end_time_seconds)
+        
+                    if start_time_str_list:
+                        for start_time_seconds, start_time_str, end_time_seconds in zip(start_time_seconds_list, start_time_str_list, end_time_seconds_list):
+                            st.markdown("__Fragmento: " + start_time_str + "__")
+                            _, container, _ = st.columns([SIDE, WIDTH, SIDE])
+                            with container:
+                                st_player(youtube_video_url.replace("?enablejsapi=1", "") + f'?start={start_time_seconds}&end={end_time_seconds}')
                         
         st.session_state.messages.append({"role": "assistant", "content": llm_response})
 # -- Sample: streamlit run app.py -- --DEFAULT_SYSTEM_PROMPT_LINK=https://raw.githubusercontent.com/AlbertoUAH/Castena/main/prompts/default_system_prompt.txt --PODCAST_URL_VIDEO_PATH=https://raw.githubusercontent.com/AlbertoUAH/Castena/main/data/podcast_youtube_video.csv --TRANSCRIPTION=worldcast_roberto_vaquero --MODEL=togethercomputer/llama-2-7b-chat --EMB_MODEL=BAAI/bge-base-en-v1.5