|
import requests |
|
import util |
|
import glob |
|
import json |
|
import os.path |
|
import pickle |
|
|
|
from flask import Flask, request, jsonify, send_file, send_from_directory |
|
app = Flask(__name__) |
|
|
|
white = ['http://127.0.0.1:5500'] |
|
@app.after_request |
|
def add_cors_headers(response): |
|
if(request.referrer): |
|
r = request.referrer[:-1] |
|
if r in white: |
|
response.headers.add('Access-Control-Allow-Origin', r) |
|
response.headers.add('Access-Control-Allow-Credentials', 'true') |
|
response.headers.add('Access-Control-Allow-Headers', 'Content-Type') |
|
response.headers.add('Access-Control-Allow-Headers', 'Cache-Control') |
|
response.headers.add('Access-Control-Allow-Headers', 'X-Requested-With') |
|
response.headers.add('Access-Control-Allow-Headers', 'Authorization') |
|
response.headers.add('Access-Control-Allow-Methods', 'GET, POST, OPTIONS, PUT, DELETE') |
|
return response |
|
|
|
|
|
@app.route('/h') |
|
def hello(): |
|
return "hi" |
|
|
|
|
|
@app.route('/files') |
|
def get_file_names(): |
|
folder_path = util.__folder_path__ + "Actor_*\\*.wav" |
|
file_names = [] |
|
for file in glob.glob(folder_path): |
|
file_name = os.path.basename(file) |
|
file_path = os.path.dirname(file).split('\\')[-1] |
|
rel_file_name = file_path+"/"+file_name |
|
file_names.append(rel_file_name) |
|
return file_names |
|
|
|
|
|
@app.route('/get_file', methods=['POST']) |
|
def get_file(): |
|
file_name = request.json['file_name'] |
|
file_path = util.__folder_path__+file_name |
|
return send_file(file_path) |
|
|
|
|
|
@app.route('/get_emotion', methods=['POST']) |
|
def get_emotion(): |
|
file_name = request.form['file_name'] |
|
file_path = util.__folder_path__+file_name |
|
util.load_model() |
|
emotion = util.predict_emotion(file_path) |
|
return emotion |
|
|
|
|
|
if __name__ == "__main__": |
|
print("starting") |
|
|
|
app.run(port=8000,debug=True) |
|
|
|
|
|
|
