app.py

import os
from flask import Flask,render_template,request,send_from_directory
from src1.components.video_to_subtitle import SubtitleGenerator,Video2SubConfig
from src1.components.summarize import Summarizer
from src1.pipeline import run_training_pipeline
import threading
from gtts import gTTS
from src1.components.video_downloader import VideoDownloader

app=Flask(__name__)
app.config['upload_dir']=os.path.join('uploads/')
os.makedirs(app.config['upload_dir'],exist_ok=True)

print("Load summarizer Model")
summarizer=Summarizer()
summarizer.load_model()

@app.route('/',methods=['GET'])
def index():
    """
    It renders the index.html file in the templates folder

    Returns:
        The index.html file is being returned
    """
    return render_template('index.html')

@app.route('/transcribe',methods=['POST'])
def upload_file():
    """
    The function upload_file() takes in a video file or a video link,transcribes the video,
    and summarizes the transcript
    
    Returns:
        the transcript and summary text.
    """
    
    if request.method =='POST':
        video_file=request.files["video_file"]
        subtitle_generator=SubtitleGenerator(config=Video2SubConfig)
        if video_file:
            filename=video_file.filename
            video_path=os.path.join((os.path.join(app.config['upload_dir'],filename)))
            print(video_path)
            video_file.save(video_path)
            task='transcribe'
            if len(request.form.getlist('translate-btn')) >0:
                task='translate'
                print("translate task is selected:",task)
            subtitle=subtitle_generator.get_subtitles(video_paths=video_path,model_path=tiny,
                                                      task=task,verbose=False)

        else:
            video_link=request.form["link-input"]

            downloader=VideoDownloader(
                url=video_link,save_path=app.config['upload_dir']
            )
            video_path=downloader.download()
            task='transcribe'
            option=request.form.get('options')
            if option=='translate':
                task='translate'
                print("translate task is selected:",task)
            print("task selected:",task)
            subtitle=subtitle_generator.get_subtitles(video_paths=[video_path],
                                                      model_path='tiny',task=task,verbose=False)
        transcript_text=subtitle[1]['text']
        summary_text=summarizer.summarize_text(transcript_text)
        tts=gTTS(summary_text,tld="co.uk")
        audio_path=os.path.join(app.config['upload_dir'],'summary.mp3')
        tts.save(audio_path)

        return render_template('index.html',transcript=transcript_text,summary=summary_text,
                               audio_path=audio_path)


@app.route('/audio/<path:filename>')
def serve_audio(filename):
    return send_from_directory(app.config['upload_dir'],filename,mimetype='audio/mpeg')

@app.route('/train',methods=['GET','POST'])
def start_train():
    def run_training():
        run_training_pipeline()
    train_thread=threading.Thread(target=run_training)
    train_thread.start()
    return "Training has started"

if __name__=="__main__":
    app.run(host='0.0.0.0',port=5000,debug=True)