Commit Test SRT

app.py

@@ -1,10 +1,29 @@
 import streamlit as st
 from transformers import pipeline
 
-pipe = pipeline('summarization')
-# pipe = pipeline('sentiment-analysis')
-text = st.text_area('enter some text:')
+# Function to extract text from SRT file
+def extract_text_from_srt(srt_text):
+    lines = srt_text.strip().split("\n")
+    # Extract text content from SRT
+    texts = [line.split("\n")[2] for line in lines if not line.startswith("0")]
+    return " ".join(texts)
 
-if text:
-    out = pipe(text)
-    st.json(out)
+# Load summarization pipeline
+summarizer = pipeline("summarization")
+
+# Streamlit app
+st.title("SRT Summarization")
+
+# Text area for user to upload SRT file
+srt_file = st.file_uploader("Upload SRT file", type=["srt"])
+
+if srt_file is not None:
+    # Read uploaded SRT file
+    srt_text = srt_file.read().decode("utf-8")
+    # Extract text from SRT
+    text_to_summarize = extract_text_from_srt(srt_text)
+    # Summarize text
+    summary = summarizer(text_to_summarize, max_length=150, min_length=30, do_sample=False)
+    # Display summary
+    st.subheader("Summary:")
+    st.write(summary[0]["summary_text"])