Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import subprocess
 import sys
+import pandas as pd
 
 def install_libraries():
     try:
@@ -23,54 +24,32 @@ def main():
     st.title("Translate YouTube Video to French")
     
     url = st.text_input("Enter YouTube Video URL")
-    captions_list = []  # Initialize empty list for captions
     
     if url:
         captions, translated_captions, original_file, translated_file = Milestone5API.download_video_transcript(url)
   
-        if captions:
-            # Parse captions from the API response
+        if captions and translated_captions:
+            captions_list = []
+            translated_timestamps = []
             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
+                    captions_list.append({"Original Timestamp": timestamp, "Original": text})
 
-        # 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><th class='translated'>Translated</th></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'>{translated_captions[i]}</td></tr>",
-                unsafe_allow_html=True,
-            )
+            for line in translated_captions.split("\n"):
+                if line.strip():
+                    timestamp, text = line.split(" ", 1)
+                    translated_timestamps.append(timestamp)
 
-        st.write("</table>", unsafe_allow_html=True)
+            df = pd.DataFrame(captions_list)
+            df['Translated Timestamp'] = translated_timestamps
+            df['Translated'] = translated_captions.split("\n")
 
-        # 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
+            st.write("## Caption Table")
+            st.write(df)
 
-        if col3.button("Next") and end_index < len(captions_list):
-            start_index += 6
-            end_index += 6
-            st.session_state['start_index'] = start_index
+        else:
+            st.write("Provided URL did not contain captions or translations")
 
 if __name__ == "__main__":
     main()