Update app.py

app.py

@@ -30,23 +30,23 @@ def main():
   
         if captions and translated_captions:
             captions_list = []
-            translated_timestamps = []
+            translated_list = []
             for line in captions.split("\n"):
                 if line.strip():
                     timestamp, text = line.split(" ", 1)
-                    captions_list.append({"Original Timestamp": timestamp, "Original": text})
+                    captions_list.append({"Timestamp": timestamp, "Original": text})
 
             for line in translated_captions.split("\n"):
                 if line.strip():
                     timestamp, text = line.split(" ", 1)
-                    translated_timestamps.append(timestamp)
+                    translated_list.append(text)
 
             df = pd.DataFrame(captions_list)
-            df['Translated Timestamp'] = translated_timestamps
-            df['Translated'] = translated_captions.split("\n")
+            df['Translated'] = translated_list
+            df['Translated Timestamp'] = df['Timestamp']  # Assuming the timestamps align
 
             st.write("## Caption Table")
-            st.write(df)
+            st.write(df[['Timestamp', 'Original', 'Translated Timestamp', 'Translated']])
 
         else:
             st.write("Provided URL did not contain captions or translations")