Update server.py

server.py

@@ -1,16 +1,20 @@
 from flask import Flask, request, jsonify
-from moviepy.editor import *
-from vid2sum_model import Vid2SumModel
+from summarize import Vid2Sum
 
 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})
+  # Check if video file is included in the request
+    if 'video' not in request.files:
+        return jsonify({'error': 'No video file found'}), 400
+
+    video_file = request.files['video']
+    # Perform video-to-text conversion using your existing code
+    text_output = video_to_text(video_file)
+
+    return jsonify({'text': text_output})
 
 if __name__ == '__main__':
     app.run()