117 lines
5.1 KiB
Python
117 lines
5.1 KiB
Python
|
import os
|
||
|
|
from flask import Blueprint, request, jsonify, current_app
|
||
|
|
from werkzeug.utils import secure_filename
|
||
|
|
|
||
|
|
from src.services.sop_generator import SopGenerator
|
||
|
|
from src.utils.utils import delete_all_files_in_directory
|
||
|
|
from src.utils.document_loader import load_document
|
||
|
|
import json
|
||
|
|
# Initialize the Blueprint
|
||
|
|
sops_bp = Blueprint('sops', __name__)
|
||
|
|
|
||
|
|
# Initialize SopGenerator
|
||
|
|
sop_generator = SopGenerator()
|
||
|
|
|
||
|
|
ALLOWED_EXTENSIONS = {'pdf', 'doc', 'docx'}
|
||
|
|
|
||
|
|
def allowed_file(filename):
|
||
|
|
"""Check if the file has an allowed extension."""
|
||
|
|
return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
|
||
|
|
|
||
|
|
@sops_bp.route('/get_roles', methods=['POST'])
|
||
|
|
def get_roles():
|
||
|
|
# Check if the post request has the file part
|
||
|
|
if 'document' not in request.files:
|
||
|
|
return jsonify({"error": "No file part", "message": "Please upload a file with the key 'document'."}), 400
|
||
|
|
|
||
|
|
file = request.files['document']
|
||
|
|
|
||
|
|
# If the user does not select a file, the browser may also submit an empty part without filename
|
||
|
|
if file.filename == '':
|
||
|
|
return jsonify({"error": "No selected file", "message": "A file was not selected for upload. Please select a valid file."}), 400
|
||
|
|
|
||
|
|
if file and allowed_file(file.filename):
|
||
|
|
filename = secure_filename(file.filename)
|
||
|
|
upload_folder = current_app.config['UPLOAD_FOLDER']
|
||
|
|
file_path = os.path.join(upload_folder, filename)
|
||
|
|
|
||
|
|
# Save the file to the upload folder
|
||
|
|
file.save(file_path)
|
||
|
|
|
||
|
|
try:
|
||
|
|
# Use the utility function to generate docs from the file
|
||
|
|
docs = load_document(file_path)
|
||
|
|
|
||
|
|
# Generate roles from the docs
|
||
|
|
roles = sop_generator.get_roles(docs)["roles"]
|
||
|
|
|
||
|
|
# Cleanup: Delete all files in the upload directory after processing
|
||
|
|
delete_all_files_in_directory(upload_folder)
|
||
|
|
|
||
|
|
return jsonify({"roles": roles, "message": "Roles successfully extracted from the document."}), 200
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
# Cleanup: Delete all files in the upload directory if an error occurs
|
||
|
|
delete_all_files_in_directory(upload_folder)
|
||
|
|
return jsonify({"error": "Processing error", "message": f"An error occurred while processing the document: {str(e)}"}), 500
|
||
|
|
|
||
|
|
return jsonify({"error": "File type not allowed", "message": "The uploaded file type is not allowed. Please upload a PDF, DOC, or DOCX file."}), 400
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
@sops_bp.route('/generate_sops', methods=['POST'])
|
||
|
|
def generate_sops():
|
||
|
|
# Check if the POST request has the file part
|
||
|
|
if 'document' not in request.files:
|
||
|
|
return jsonify({"error": "No file part", "message": "Please upload a file with the key 'document'."}), 400
|
||
|
|
|
||
|
|
print("Running................")
|
||
|
|
|
||
|
|
file = request.files['document']
|
||
|
|
roles_json = request.form.get('roles') # Get the roles as a JSON string
|
||
|
|
if not roles_json:
|
||
|
|
return jsonify({"error": "No roles provided", "message": "Please provide a list of roles in the 'roles' field."}), 400
|
||
|
|
|
||
|
|
try:
|
||
|
|
roles = json.loads(roles_json) # Parse the roles from JSON string to a list
|
||
|
|
print(f"Roles are:{roles}")
|
||
|
|
except json.JSONDecodeError:
|
||
|
|
return jsonify({"error": "Invalid JSON", "message": "The 'roles' field contains invalid JSON."}), 400
|
||
|
|
|
||
|
|
# If the user does not select a file, the browser may also submit an empty part without a filename
|
||
|
|
if file.filename == '':
|
||
|
|
return jsonify({"error": "No selected file", "message": "A file was not selected for upload. Please select a valid file."}), 400
|
||
|
|
|
||
|
|
if file and allowed_file(file.filename):
|
||
|
|
filename = secure_filename(file.filename)
|
||
|
|
upload_folder = current_app.config['UPLOAD_FOLDER']
|
||
|
|
file_path = os.path.join(upload_folder, filename)
|
||
|
|
|
||
|
|
# Save the file to the upload folder
|
||
|
|
file.save(file_path)
|
||
|
|
|
||
|
|
try:
|
||
|
|
# Use the utility function to generate docs from the file
|
||
|
|
docs = load_document(file_path)
|
||
|
|
|
||
|
|
# Check if the document can generate SOPs for the roles
|
||
|
|
status_check = sop_generator.check_role_sop(roles=roles, docs=docs)
|
||
|
|
|
||
|
|
if not status_check["status"]:
|
||
|
|
return jsonify({"error": "Document cannot extract SOPs", "message": status_check["message"]}), 400
|
||
|
|
|
||
|
|
# Generate SOPs based on the roles provided
|
||
|
|
sops = sop_generator.generate_sops(roles, docs)
|
||
|
|
|
||
|
|
# Cleanup: Delete all files in the upload directory after processing
|
||
|
|
delete_all_files_in_directory(upload_folder)
|
||
|
|
|
||
|
|
return jsonify({"sops": sops, "message": "SOPs successfully generated for the roles from the document."}), 200
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
# Cleanup: Delete all files in the upload directory if an error occurs
|
||
|
|
delete_all_files_in_directory(upload_folder)
|
||
|
|
return jsonify({"error": "Processing error", "message": f"An error occurred while processing the document: {str(e)}"}), 500
|
||
|
|
|
||
|
|
return jsonify({"error": "File type not allowed", "message": "The uploaded file type is not allowed. Please upload a PDF, DOC, or DOCX file."}), 400
|