adding inference

app.py

@@ -1,9 +1,15 @@
+import os
 import whisper
 from flask import Flask, jsonify, render_template, request, send_file
+from werkzeug.utils import secure_filename
+
 
 app = Flask(__name__)
 
 STARTING_SIZE = 'small'
+UPLOAD_FOLDER = 'uploads'
+ALLOWED_EXTENSIONS = {'ogg', 'mp3', 'mp4', 'wav',
+                      'flac', 'm4a', 'aac', 'wma', 'webm', 'opus'}
 
 
 current_size = STARTING_SIZE
@@ -11,11 +17,6 @@ model = whisper.load_model(current_size)
 model_en = whisper.load_model(f"{current_size}.en")
 
 
-@app.route("/")
-def index():
-    return "nothing yet to see here"
-
-
 def inference(audio_file):
     audio = whisper.load_audio(audio_file)
     audio = whisper.pad_or_trim(audio)
@@ -37,5 +38,30 @@ def inference(audio_file):
     return segmented_text
 
 
+def allowed_file(filename):
+    return '.' in filename and \
+           filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
+
+@app.route("/", methods=['GET', 'POST'])
+def index():
+    if request.method == 'POST':
+        # check if the post request has the file part
+        if 'file' not in request.files:
+            return "no file sent!"
+        uploaded_file = request.files['file']
+        if uploaded_file.filename == '':
+            return "no file sent!"
+        if uploaded_file and allowed_file(uploaded_file.filename):
+            filename = secure_filename(uploaded_file.filename)
+            uploaded_file.save(os.path.join(
+                app.config['UPLOAD_FOLDER'], filename))
+            results = inference(os.path.join(
+                app.config['UPLOAD_FOLDER'], filename))
+            return jsonify({"results": results})
+
+    return "nothing yet to see here"
+
+
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)