Update app.py

app.py

@@ -35,13 +35,13 @@ def main():
                         captions_list.append({"Timestamp": timestamp, "Language": lang, "Text": text})
         
         if translated_captions:
-            # Parse translated captions
-            translated_text = translated_captions.split("\n")
-            for i, line in enumerate(translated_text):
-                lang, timestamp, text = line.split(" ", 2)
-                if lang == "fr:" or lang == "en:":  # Considering lines starting with "fr:" or "en:"
-                    captions_list[i]["Translated"] = text
-                    
+            captions_list = []
+            for original_line, translated_line in zip(captions_list, translated_captions):
+                lang, timestamp, text = original_line.split(" ", 2)
+                lang_trans, timestamp_trans, text_trans = translated_line.split(" ", 2)
+                if lang == "en:" or lang == "fr:":  # Assuming lines starting with "fr:" or "en:"
+                    captions_list.append({"Timestamp": timestamp, "Language": lang, "Text": text, "Translated": text_trans})
+                
         df = pd.DataFrame(captions_list)
         
         start_index = st.session_state.get('start_index', 0)