Update app.py

app.py

@@ -107,7 +107,32 @@ def yt_transcribe(yt_url, task):
     text= convert_to_srt(text)
     return html_embed_str, text
 
-
+# SRT prepare
+# Assuming srt format is a sequence of subtitles with index, time range and text
+def convert_to_srt(input):
+  output = ""
+  index = 1
+  for chunk in input["chunks"]:
+    start, end = chunk["timestamp"]
+    text = chunk["text"]
+    if end is None:
+      end = "None"
+    # Convert seconds to hours:minutes:seconds,milliseconds format
+    start = format_time(start)
+    end = format_time(end)
+    output += f"{index}\n{start} --> {end}\n{text}\n\n"
+    index += 1
+  return output
+
+# Helper function to format time
+def format_time(seconds):
+  if seconds == "None":
+    return seconds
+  hours = int(seconds // 3600)
+  minutes = int((seconds % 3600) // 60)
+  seconds = int(seconds % 60)
+  milliseconds = int((seconds % 1) * 1000)
+  return f"{hours:02}:{minutes:02}:{seconds:02},{milliseconds:03}"
 demo = gr.Blocks()
 mf_transcribe = gr.Interface(
     fn=transcribe,