api creation

ProficiencyLevelDetection/app.py

+from flask import Flask, request, jsonify
+import pickle
+import pandas as pd
+
+
+from flask_cors import CORS  # Import CORS
+
+
+
+app = Flask(__name__)
+
+CORS(app)
+
+# Load the proficiency model
+model_proficiency_path = 'weights/proficiency-rf.pickle'
+with open(model_proficiency_path, 'rb') as f:
+    model_proficiency = pickle.load(f)
+
+class_dict_rev = {
+    0: 'Basic',
+    1: 'Intermediate',
+    2: 'Advanced'
+}
+
+@app.route('/predict_proficiency', methods=['POST'])
+def predict_proficiency():
+    try:
+        data = request.get_json()
+        sample_json = pd.DataFrame(data, index=[0]).values
+        pred = model_proficiency.predict(sample_json)[0]
+        predicted_class = class_dict_rev[pred]
+        result = {'predicted_class': predicted_class}
+        return jsonify(result)
+    except Exception as e:
+        return jsonify({'error': str(e)})
+
+if __name__ == '__main__':
+    app.run(debug=True)