d_transcript / app.py
johann22's picture
Create app.py
4e9f8f3 verified
raw
history blame
1.86 kB
import os
import gradio as gr
import yt_dlp
def dl(inp,img=None):
uid=uuid.uuid4()
fps="Error"
out = None
out_file=[]
if img == None and inp !="":
try:
inp_out=inp.replace("https://","")
inp_out=inp_out.replace("/","_").replace(".","_").replace("=","_").replace("?","_")
#os.system(f'yt-dlp "{inp}" --trim-filenames 160 -o "{uid}/{inp_out}.mp4" -S res,mp4 --recode mp4')
#os.system(f'yt-dlp --skip-download --write-subs --write-auto-subs --sub-lang en --sub-format ttml --convert-subs srt "{inp}" -o "{uid}/{inp_out}"')
os.system(f'yt-dlp -x "{inp}" -o "{uid}/{inp_out}"')
#f = open(f"{uid}/{inp_out}.en.srt")
ffmpeg -i f"{uid}/{inp_out}.opus" "output.mp3"
##ft=f.readlines()
##line_fin=""
##line_out=""
##for line in ft:
## if "<" in line:
## line_out = line.split(">",1)[1].split("<",1)[0]
## else:
## line_out = line
## if not line.strip("\n").isnumeric():
## line_fin+=line_out
#print(ft)
#out = f"{uid}/{inp_out}.mp4"
#capture = cv2.VideoCapture(out)
#fps = capture.get(cv2.CAP_PROP_FPS)
#capture.release()
out="output.mp3"
##out=f'{line_fin}'
except Exception as e:
print(e)
out = None
elif img !=None and inp == "":
capture = cv2.VideoCapture(img)
fps = capture.get(cv2.CAP_PROP_FPS)
capture.release()
out = f"{img}"
return out
os.system(f"python diarize.py -a {AUDIO_FILE_NAME}")
with gr.Blocks() as app:
inp_url=gr.Textbox()
btn=gr.Button()
outp_aud=gr.Audio()
btn.click(dl,inp_url,outp_aud)
app.launch()