Feature/add description improvements

.gitignore

@@ -2,4 +2,5 @@
 .venv
 .pytest_cache/
 __pycache__
-.vscode
+.vscode
+.env

app.py

@@ -3,7 +3,7 @@
 '''
 from flask import Flask, jsonify, request
 from models import Experience, Education, Skill
-
+from gpt_connection import get_improvement
 app = Flask(__name__)
 
 data = {
@@ -21,7 +21,8 @@
                   "September 2019",
                   "July 2022",
                   "80%",
-                  "example-logo.png")
+                  "example-logo.png",
+                  "I was head of the debate team at university")
     ],
     "skill": [
         Skill("Python",
@@ -65,6 +66,25 @@ def education():
 
     return jsonify({})
 
+@app.route('/resume/reword_description', methods=['GET'])
+def reword_description():
+    '''
+    Rewords the description using GPT
+    '''
+    model = None
+    try:
+        model = Experience(**request.json)
+    except:
+        model = Education(**request.json)
+
+    if model is None:
+        return jsonify({"error": "Invalid request"}), 400
+
+    response = get_improvement(model)
+    if response is None:
+        return jsonify({"error": "Failed to get improvement"}), 500
+
+    return jsonify({"response": response})
 
 @app.route('/resume/skill', methods=['GET', 'POST'])
 def skill():

gpt_connection.py

@@ -0,0 +1,31 @@
+
+
+from openai import OpenAI
+import os
+from dotenv import load_dotenv
+from models import Experience, Education
+load_dotenv()
+
+client = OpenAI(
+    api_key=os.getenv("OPENAI_API_KEY")
+)
+
+def get_improvement(model : Experience | Education):
+    prompt = f"Improve the following description for the {model.__class__.__name__}: {model.description}"
+
+    response = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=[
+            {"role": "system", "content": "You are a helpful assistant that improves descriptions for resumes."},
+            {"role": "assistant", "content": "Here is the improved description:"},
+            {"role": "system", "content": f"Consider the following information about the {model.__class__.__name__}: {model.model_dump_json()}"},
+            {"role": "user", "content": prompt},
+        ]
+    )
+    try:
+        return response.choices[0].message.content
+    except:
+        return None
+
+
+

models.py

@@ -5,6 +5,7 @@
 '''
 
 from dataclasses import dataclass
+import json
 
 
 @dataclass
@@ -19,6 +20,11 @@ class Experience:
     description: str
     logo: str
 
+    def model_dump_json(self):
+        '''
+        Dumps the class into a JSON string
+        '''
+        return json.dumps(self, default=lambda o: o.__dict__)
 
 @dataclass
 class Education:
@@ -31,6 +37,13 @@ class Education:
     end_date: str
     grade: str
     logo: str
+    description: str
+
+    def model_dump_json(self):
+        '''
+        Dumps the class into a JSON string
+        '''
+        return json.dumps(self, default=lambda o: o.__dict__)
 
 
 @dataclass