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)