Commit Test SRT

app.py

@@ -1,10 +1,11 @@
 import streamlit as st
 from transformers import pipeline
 
+# Function to extract text from SRT-formatted text
 # Function to extract text from SRT-formatted text
 def extract_text_from_srt_text(srt_text):
-    lines = srt_text.strip().split("\n")
-    texts = [line.split("\n")[2] for line in lines if not line.startswith("0")]
+    lines = srt_text.strip().split("\n\n")  # Split by empty lines to separate subtitles
+    texts = [subtitle.split("\n")[2] for subtitle in lines if subtitle.strip()]  # Extract text from the third line of each subtitle
     return " ".join(texts)
 
 # Load summarization pipeline