from flask import Flask, request, jsonify
from moviepy.editor import *
from vid2sum_model import Vid2SumModel

app = Flask(__name__)
model = Vid2SumModel()

@app.route('/predict', methods=['POST'])
def predict():
    file = request.files['video']
    file.save('uploaded_video.mp4')  # Save the uploaded video file
    summary_text = model.predict('uploaded_video.mp4')  # Call your model's predict method
    return jsonify({'summary_text': summary_text})

if __name__ == '__main__':
    app.run()