Update app.py

app.py

@@ -1,7 +1,7 @@
+User
 import streamlit as st
 import subprocess
 import sys
-
 def install_libraries():
     try:
         subprocess.check_call(['pip', 'install', 'pytube'])
@@ -27,53 +27,49 @@ def main():
     
     if url:
         captions, translated_captions, original_file, translated_file = Milestone5API.download_video_transcript(url)
-     
-        if captions and translated_captions:
-            # Parse captions and translated captions
-            for original_line, translated_line in zip(captions, translated_captions):
-                lang, timestamp, text = original_line.split(" ", 2)
-                lang_trans, _, text_trans = translated_line.split(" ", 2)
-                if lang.startswith("en:") or lang.startswith("fr:") and lang_trans.startswith("en:") or lang_trans.startswith("fr:"):
-                    captions_list.append({
-                        "Timestamp": timestamp[4:], 
-                        "Original": text, 
-                        "Translated": text_trans
-                    })
-
-            start_index = st.session_state.get('start_index', 0)
-            end_index = start_index + 6
+        print (translated_text)
+        if captions:
+            # Parse captions from the API response
+            for line in captions.split("\n"):
+                if line.strip():
+                    timestamp, text = line.split(" ", 1)
+                    captions_list.append({"Timestamp": timestamp, "Original": text})
+        else:
+            st.write("Provided URL did not contain captions")
+    
+        start_index = st.session_state.get('start_index', 0)
+        end_index = start_index + 6
 
-            # Display caption table with new columns
-            st.write("## Caption Table")
-            with open("styles.css", "r") as f:
-                css = f.read()
-            st.markdown(f"<style>{css}</style>", unsafe_allow_html=True)
+        # Display caption table with CSS class names
+        st.write("## Caption Table")
+        with open("styles.css", "r") as f:
+            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>",
+            unsafe_allow_html=True,
+        )
+        
+        for i in range(start_index, min(end_index, len(captions_list))):
             st.write(
-                "<table><tr><th class='timestamp'>Timestamp</th><th class='original'>Original</th><th class='translated'>Translated</th></tr>",
+                f"<tr><td class='timestamp'>{captions_list[i]['Timestamp']}</td>"
+                f"<td class='original'>{captions_list[i]['Original']}</td></tr>",
                 unsafe_allow_html=True,
             )
-            
-            for i in range(start_index, min(end_index, len(captions_list))):
-                st.write(
-                    f"<tr><td class='timestamp'>{captions_list[i]['Timestamp']}</td>"
-                    f"<td class='original'>{captions_list[i]['Original']}</td>"
-                    f"<td class='translated'>{captions_list[i]['Translated']}</td></tr>",
-                    unsafe_allow_html=True,
-                )
 
-            st.write("</table>", unsafe_allow_html=True)
+        st.write("</table>", unsafe_allow_html=True)
 
-            # Navigation buttons
-            col1, col2, col3 = st.columns([1, 8, 1])
-            if col1.button("Previous") and start_index > 0:
-                start_index -= 6
-                end_index -= 6
-                st.session_state['start_index'] = start_index
+        # Navigation buttons
+        col1, col2, col3 = st.columns([1, 8, 1])
+        if col1.button("Previous") and start_index > 0:
+            start_index -= 6
+            end_index -= 6
+            st.session_state['start_index'] = start_index
 
-            if col3.button("Next") and end_index < len(captions_list):
-                start_index += 6
-                end_index += 6
-                st.session_state['start_index'] = start_index
+        if col3.button("Next") and end_index < len(captions_list):
+            start_index += 6
+            end_index += 6
+            st.session_state['start_index'] = start_index
 
 if __name__ == "__main__":
-    main()
+    main()