Update app.py

app.py

@@ -24,26 +24,25 @@ def main():
     url = st.text_input("Enter YouTube Video URL")
     captions_list = []  # Initialize empty list for captions
     
-    if url:
+     if url:
         captions, translated_captions, original_file, translated_file = Milestone5API.download_video_transcript(url)
      
         if captions:
-            # Parse captions from the API response for original 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:
-            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)
+
+            if translated_captions:
+                for i, translated_line in enumerate(translated_captions):
+                    lang_trans, timestamp_trans, text_trans = translated_line.split(" ", 2)
+                    if lang_trans == "fr:" or lang_trans == "en:":
+                        captions_list[i]["Translated"] = text_trans
+
+            df = pd.DataFrame(captions_list)
+
         
         start_index = st.session_state.get('start_index', 0)
         end_index = start_index + 6