Update app.py

app.py

@@ -27,25 +27,19 @@ def main():
     if url:
         captions, translated_captions, original_file, translated_file = Milestone5API.download_video_transcript(url)
      
-        if captions:
+        if captions and translated_captions:
             captions_list = []
-            for line in captions.split("\n"):
-                if line.strip():
-                    lang, timestamp, text = line.split(" ", 2)
-                    if lang == "en:" or lang == "fr:":
-                        captions_list.append({"Timestamp": timestamp, "Language": lang, "Text": text})
-
-            if translated_captions:
-                for i, translated_line in enumerate(translated_captions):
-                    if i < len(captions_list):  # Ensure captions_list has enough elements
-                        lang_trans, timestamp_trans, text_trans = translated_line.split(" ", 2)
-                        if lang_trans == "fr:" or lang_trans == "en:":
-                            captions_list[i]["Translated"] = text_trans
-                    else:
-                        break  # Stop if captions_list ends before translated_captions
+            for line, translated_line in zip(captions.split("\n"), translated_captions):
+                lang, timestamp, text = line.split(" ", 2)
+                lang_trans, _, text_trans = translated_line.split(" ", 2)
+                if lang in ["en:", "fr:"] and lang_trans in ["en:", "fr:"]:
+                    captions_list.append({"Timestamp": timestamp, "Original": text, "Translated": text_trans})
 
             df = pd.DataFrame(captions_list)
 
+            # Display DataFrame
+            st.write(df)
+
         
         start_index = st.session_state.get('start_index', 0)
         end_index = start_index + 6