Implemented validation into app.py

MLH-Fellowship · Feb 3, 2025 · 758380c · 758380c
1 parent b97c166
commit 758380c
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/app.py b/app.py
@@ -5,6 +5,7 @@
 from models import Experience, Education, Skill
 from gpt_connection import get_improvement
 from spell_check import spell_check
+from validation import validate_experience, validate_education, validate_skill
 app = Flask(__name__)
 
 data = {
@@ -51,9 +52,14 @@ def experience():
 
     if request.method == 'POST':
         json_data = request.json
+
         if json_data.get('spell_check') and isinstance(json_data.get('description'), str):
             json_data['description'] = spell_check(json_data['description'])
-
+
+        try:
+            validate_experience(json_data)
+        except ValueError as e:
+            return jsonify({"error": str(e)}), 400
         return jsonify(json_data)
 
     return jsonify({})
@@ -70,6 +76,10 @@ def education():
         json_data = request.json
         if json_data.get('spell_check') and isinstance(json_data.get('description'), str):
             json_data['description'] = spell_check(json_data['description'])
+        try:
+            validate_education(json_data)
+        except ValueError as e:
+            return jsonify({"error": str(e)}), 400
         return jsonify(json_data)
 
     return jsonify({})
@@ -104,6 +114,10 @@ def skill():
         json_data = request.json
         if json_data.get('spell_check') and isinstance(json_data.get('description'), str):
             json_data['description'] = spell_check(json_data['description'])
+        try:
+            validate_skill(json_data)
+        except ValueError as e:
+            return jsonify({"error": str(e)}), 400
 
         return jsonify(json_data)