Update utils.py

utils.py

@@ -106,7 +106,7 @@ PREGUNTA:""", cleaned_prompt, re.DOTALL)
                                           )
         text = output['output']['choices'][0]['text']
         text = self.clean_duplicates(text)
-        return text
+        return text, new_cleaned_prompt
 
 # -- Python function to setup basic features: translator, SpaCy pipeline and LLM model
 @st.cache_resource
@@ -216,11 +216,16 @@ def time_to_seconds(time_str):
     return int((hours * 3600) + (minutes * 60) + seconds)
 
 # -- Extract seconds from transcription
-def add_hyperlink_and_convert_to_seconds(text):
+def add_hyperlink_and_convert_to_seconds(text, original_prompt):
     time_pattern = r'(\d{2}:\d{2}:\d{2}(?:.\d{6})?)'
     
     def get_seconds(match):
-        start_time_str, end_time_str = match[0], match[1]
+        if len(match) == 2:
+            start_time_str, end_time_str = match[0], match[1]
+        else:
+            start_time_str = match[0]
+            end_time_str   = re.findall(r"Desde el instante {} hasta {}".format(start_time_str, time_pattern))[0].split('hasta ')[-1]
+            
         start_time_seconds = time_to_seconds(start_time_str)
         end_time_seconds   = time_to_seconds(end_time_str)
         return start_time_str, start_time_seconds, end_time_str, end_time_seconds