Commit Test SRT

app.py

@@ -1,25 +1,13 @@
 from transformers import pipeline
 
-def summarize_srt(srt_text, max_length=100):
-    # Extract text from SRT format
-    lines = srt_text.split('\n\n')
-    text = ' '.join(line.split('\n')[2] for line in lines if line.strip())
-
-    # Initialize the summarization pipeline
+def summarize_text(input_text, max_length=100):
     summarizer = pipeline("summarization")
-
-    # Summarize the text
-    summary = summarizer(text, max_length=max_length)
-
+    summary = summarizer(input_text, max_length=max_length)
     return summary[0]['summary_text']
 
 def main():
-    # Take SRT content as input
-    srt_content = input("Enter the SRT content:\n")
-
-    max_length = 100  # Maximum length of the summary
-
-    summary = summarize_srt(srt_content, max_length)
+    input_text = input("Enter the text to summarize: ")
+    summary = summarize_text(input_text)
     print("Summary:")
     print(summary)