Update app.py

app.py

@@ -1,4 +1,3 @@
-import streamlit as st
 import whisper
 import os
 import ffmpeg
@@ -120,35 +119,14 @@ def translate_text(input_file:str, output_path:str, lang: str):
     in_file.close()
     out_file.close()
 
-        
-
-app = Flask(__name__)
-
-
-@app.route("/")
-
-def main():
-    audio_dir_name = 'audio'
-    video_dir_name = 'videos'
-    whisper_dir_name = 'whisper_transcripts'
-    current_dir = os.getcwd()
-    audio_dir_path = os.path.join(current_dir, audio_dir_name)
-    video_dir_path = os.path.join(current_dir, video_dir_name)
-    whisper_dir_path = os.path.join(current_dir, whisper_dir_name)
 
-    if(not(os.path.exists(audio_dir_path))):
-        os.makedirs(audio_dir_path)
-    if(not(os.path.exists(whisper_dir_path))):
-        os.makedirs(whisper_dir_path)
-    
-
-    files = os.listdir(video_dir_path) # accessing all files in audio dir
-    in_file = os.path.join(video_dir_path, files[0]) #grabs first file in the directory 
-
-
-    translate_text(transcribe_audio(sep_audio(in_file,audio_dir_path),whisper_dir_path),current_dir,'german')
+### FRONT END ###
+import streamlit as st
+from transformers import pipeline
 
-    return 200, "audio has been transcribed successfully"
+pipe = pipeline('video-translation')
+text = st.text_area('enter a video url!')
 
-if __name__ == "__main__":
-    app.run(debug=True)
+if text:
+    out = pipe(text)
+    st.json(out)