Update app.py

app.py

@@ -1,3 +1,4 @@
+import streamlit as st
 import subprocess
 import sys
 
@@ -7,53 +8,62 @@ def install_libraries():
         subprocess.check_call(['pip', 'install', 'youtube_transcript_api'])
         subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'torch'])
         subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'transformers'])
-        subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'sentencepiece'])
+        subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'sentencepiece']) 
         print("Libraries installed successfully!")
     except subprocess.CalledProcessError as e:
         print("Error during installation:", e)
 
-def install_and_download(url):
-    install_libraries()
+install_libraries()
 
-    import streamlit as st
-    import pandas as pd
+def main():
+    # Importing after ensuring dependencies are installed
     import Milestone5API
-
-    @st.experimental_memo
-    def download_transcripts(url):
-        return Milestone5API.download_video_transcript(url)
-
+    
+    # Title 
     st.title("Translate YouTube Video to French")
+    
     url = st.text_input("Enter YouTube Video URL")
-
+    captions_list = []  # Initialize empty list for captions
+    
     if url:
-        cached_data = download_transcripts(url)
-
-        captions, translated_captions, _, _ = cached_data
-
+        captions, translated_captions, original_file, translated_file = Milestone5API.download_video_transcript(url)
+     
         if captions and translated_captions:
-            captions_list = []
-            translated_text = []
-
-            for line, translated_line in zip(captions, translated_captions):
-                lang, timestamp, text = line.split(" ", 2)
+            # 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.append(text_trans)
+                    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
 
+            # Display caption table with new columns
             st.write("## Caption Table")
-            df = pd.DataFrame(captions_list[start_index:end_index], columns=["Timestamp", "Original"])
-            df['Translated'] = translated_text[start_index:end_index]
-
             with open("styles.css", "r") as f:
                 css = f.read()
             st.markdown(f"<style>{css}</style>", unsafe_allow_html=True)
-            st.write(df)
+            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'>{captions_list[i]['Translated']}</td></tr>",
+                    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
@@ -66,4 +76,4 @@ def install_and_download(url):
                 st.session_state['start_index'] = start_index
 
 if __name__ == "__main__":
-    install_and_download(None)  # Call the function to start the app
+    main()