Spaces:
Runtime error
Runtime error
| import os | |
| import cv2 | |
| from flask import Flask, app, request, jsonify | |
| from werkzeug.utils import secure_filename | |
| #----------------------------------------------------- | |
| # Tempat setting server | |
| UPLOAD_FOLDER = './upload' | |
| UPLOAD_MODEL = './models' | |
| ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg','zip','h5'} | |
| app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER | |
| app.config['UPLOAD_MODEL'] = UPLOAD_MODEL | |
| app.config['MAX_CONTENT_LENGTH'] = 500 * 1024 * 1024 | |
| #----------------------------------------------------- | |
| class FaceDetectionApp: | |
| def __init__(self, upload_folder="./static", public_url="/"): | |
| self.app = Flask(__name__) | |
| self.app.config['UPLOAD_FOLDER'] = upload_folder | |
| self.app.config['UPLOAD_MODEL'] = os.path.join(upload_folder, "model") | |
| self.public_url = public_url | |
| def delete_img(): | |
| for i in range (0,4): | |
| if os.path.exists(f"./static/result_upload{i}.jpg"): | |
| os.remove(f"./static/result_upload{i}.jpg") | |
| print("File terhapus") | |
| return jsonify({'message': 'Berhasil di hapus'}), 400 | |
| else: | |
| print("File tidak ditemukan.") | |
| return jsonify({'message': 'No file selected for uploading'}), 400 | |
| def upload_file(): | |
| if 'file' not in request.files: | |
| return jsonify({'message': 'No file part in the request'}), 400 | |
| file = request.files['file'] | |
| if file.filename == '': | |
| return jsonify({'message': 'No file selected for uploading'}), 400 | |
| filename = secure_filename(file.filename) | |
| filepath = os.path.join(self.app.config['UPLOAD_FOLDER'], filename) | |
| file.save(filepath) | |
| img = cv2.imread(filepath) | |
| # Detect Face | |
| try: | |
| self.preprocessing(img) | |
| return jsonify({'message': 'File successfully uploaded'}) | |
| except: | |
| path = "empty_image.png" | |
| return jsonify({'message': 'File failed to uploaded'}) | |
| def upload_data(): | |
| if 'file' not in request.files: | |
| return jsonify({'message': 'No file part in the request'}), 400 | |
| file = request.files['file'] | |
| if file.filename == '': | |
| return jsonify({'message': 'No file selected for uploading'}), 400 | |
| if file and self.allowed_file(file.filename): | |
| filename = secure_filename(file.filename) | |
| filepath = os.path.join(self.app.config['UPLOAD_FOLDER'], filename) | |
| file.save(filepath) | |
| self.extract_zip(filepath) | |
| return jsonify({'message': 'File successfully uploaded'}) | |
| return jsonify({'message': 'File failed to uploaded'}) | |
| def upload_model(): | |
| if 'file' not in request.files: | |
| return jsonify({'message': 'No file part in the request'}), 400 | |
| file = request.files['file'] | |
| if file.filename == '': | |
| return jsonify({'message': 'No file selected for uploading'}), 400 | |
| if file and self.allowed_file(file.filename): | |
| filename = secure_filename(file.filename) | |
| filepath = os.path.join(self.app.config['UPLOAD_MODEL'], filename) | |
| file.save(filepath) | |
| return jsonify({'message': 'File successfully uploaded'}) | |
| return jsonify({'message': 'File failed to uploaded'}) | |
| def get_total_files(): | |
| training_counts = self.get_training_file_counts() | |
| testing_counts = self.get_testing_file_counts() | |
| result = {} | |
| result['training'] = training_counts.json | |
| result['testing'] = testing_counts.json | |
| return jsonify(result) |