Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import subprocess
-import pandas as pd
 
 def install_libraries():
     try:
@@ -13,15 +12,19 @@ def install_libraries():
 install_libraries()
 
 def main():
+    # Importing after ensuring dependencies are installed
     import Milestone5API
+    
+    # Title 
     st.title("Translate YouTube Video to French")
     
     url = st.text_input("Enter YouTube Video URL")
-    captions_list = []  
+    captions_list = []  # Initialize empty list for captions
     
     if url:
         captions = Milestone5API.download_video_transcript(url)
         if captions:
+            # Parse captions from the API response
             for line in captions.split("\n"):
                 if line.strip():
                     timestamp, text = line.split(" ", 1)
@@ -32,26 +35,36 @@ def main():
         start_index = st.session_state.get('start_index', 0)
         end_index = start_index + 6
 
-        df = pd.DataFrame(captions_list)
-
+        # 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(
+                f"<tr><td class='timestamp'>{captions_list[i]['Timestamp']}</td>"
+                f"<td class='original'>{captions_list[i]['Original']}</td></tr>",
+                unsafe_allow_html=True,
+            )
 
-        st.markdown(f"<style>{css}</style>", unsafe_allow_html=True)
-
-        st.table(df[['Timestamp', 'Original']].iloc[start_index:end_index])
+        st.write("</table>", unsafe_allow_html=True)
 
+        # Navigation buttons
         col1, col2, col3 = st.columns([1, 8, 1])
-        with col1:
-            st.button("Previous", key='prev_button')
-
-        with col3:
-            if col3.button("Next") and end_index < len(captions_list):
-                start_index += 6
-                end_index += 6
-                st.session_state['start_index'] = start_index
+        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 __name__ == "__main__":
     main()