role extracion and sop generation added

src/api/app.py

@@ -0,0 +1,21 @@
+import os
+from flask import Flask
+from src.api.routes.sops import sops_bp
+
+def create_app():
+    app = Flask(__name__)
+
+    # Register the blueprint with the desired prefix
+    app.register_blueprint(sops_bp, url_prefix='/api/v1/sop')
+
+    # Set up the upload folder configuration inside the src directory
+    UPLOAD_FOLDER = os.path.join(os.path.dirname(os.path.abspath(__file__)), '../../uploads')
+    UPLOAD_FOLDER = os.path.abspath(UPLOAD_FOLDER)
+
+    # Make sure the upload folder exists
+    os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+
+    # Assign the upload folder path to Flask config
+    app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+
+    return app

src/api/routes/sops.py

@@ -0,0 +1,116 @@
+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