Update app.py

app.py

@@ -24,22 +24,21 @@ 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 translated_captions:
             # Parse translated captions from the API response
-            translated_lines = translated_captions.split('\n')
-            for line in translated_lines:
-                if line.strip():
+            for line in translated_captions:
+                if line.strip().startswith("fr:"):
                     _, timestamp, text = line.partition(": ")[2].split(" ", 2)
-                    captions_list.append({"Timestamp": timestamp, "Original": text})
+                    captions_list.append({"Timestamp": timestamp, "Translated": text})
         
         if captions_list:
             st.write("## Translated Captions")
             st.write("Timestamp - Translated Text")
             for caption in captions_list:
-                st.write(f"{caption['Timestamp']} - {caption['Original']}")
+                st.write(f"{caption['Timestamp']} - {caption['Translated']}")
         else:
             st.write("No translated captions available")