Update app.py

app.py

@@ -17,8 +17,7 @@ install_libraries()
 def main():
     # Importing after ensuring dependencies are installed
     import Milestone5API
-    
-    # Title 
+     # Title 
     st.title("Translate YouTube Video to French")
     
     url = st.text_input("Enter YouTube Video URL")
@@ -26,22 +25,25 @@ def main():
     
     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
+            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:
-            # Parse translated captions from the API response
-            for line in translated_captions:
-                if line.strip().startswith("fr:"):
-                    _, timestamp, text = line.partition(": ")[2].split(" ", 2)
-                    captions_list.append({"Timestamp": timestamp, "Translated": text})
+            # 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
+                    
+        df = pd.DataFrame(captions_list)
         
-        if captions_list:
-            st.write("## Translated Captions")
-            st.write("Timestamp - Translated Text")
-            for caption in captions_list:
-                st.write(f"{caption['Timestamp']} - {caption['Translated']}")
-        else:
-            st.write("No translated captions available")
-    
         start_index = st.session_state.get('start_index', 0)
         end_index = start_index + 6
 
@@ -51,14 +53,15 @@ def main():
             css = f.read()
         st.markdown(f"<style>{css}</style>", unsafe_allow_html=True)
         st.write(
-            "<table><tr><th class='timestamp'>Timestamp</th><th class='original'>Original</th></tr>",
+            "<table><tr><th class='timestamp'>Timestamp</th><th class='original'>Original</th><th class='translated'>Translated</th></tr>",
             unsafe_allow_html=True,
         )
         
-        for i in range(start_index, min(end_index, len(captions_list))):
+        for i in range(start_index, min(end_index, len(df))):
             st.write(
-                f"<tr><td class='timestamp'>{captions_list[i]['Timestamp']}</td>"
-                f"<td class='original'>{captions_list[i]['Original']}</td></tr>",
+                f"<tr><td class='timestamp'>{df['Timestamp'][i]}</td>"
+                f"<td class='{df['Language'][i]}'>{df['Text'][i]}</td>"
+                f"<td class='{df['Language'][i]}'>{df['Translated'][i]}</td></tr>",
                 unsafe_allow_html=True,
             )
 
@@ -71,10 +74,10 @@ def main():
             end_index -= 6
             st.session_state['start_index'] = start_index
 
-        if col3.button("Next") and end_index < len(captions_list):
+        if col3.button("Next") and end_index < len(df):
             start_index += 6
             end_index += 6
             st.session_state['start_index'] = start_index
 
 if __name__ == "__main__":
-    main()
+    main()