diff --git a/app.py b/app.py index 9cdbe80..0740b42 100644 --- a/app.py +++ b/app.py @@ -3,7 +3,7 @@ from flask_jwt_extended import JWTManager, jwt_required from functools import reduce import firebase_admin from firebase_admin import credentials -from helper.api_messages import QuestionType, get_grading_messages, get_question_gen_messages +from helper.api_messages import QuestionType, get_grading_messages, get_question_gen_messages, get_question_tips from helper.file_helper import delete_files_older_than_one_day from helper.firebase_helper import download_firebase_file, upload_file_firebase from helper.speech_to_text_helper import speech_to_text, text_to_speech @@ -11,6 +11,7 @@ from helper.token_counter import count_tokens from helper.openai_interface import make_openai_call import os import uuid +import re from dotenv import load_dotenv @@ -140,6 +141,27 @@ def grade_writing_task_2(): except Exception as e: return str(e) +@app.route('/fetch_tips', methods=['POST']) +@jwt_required() +def fetch_answer_tips(): + try: + data = request.get_json() + context = data.get('context') + question = data.get('question') + answer = data.get('answer') + correct_answer = data.get('correct_answer') + messages = get_question_tips(question, answer, correct_answer, context) + token_count = reduce(lambda count, item: count + count_tokens(item)['n_tokens'], + map(lambda x: x["content"], filter(lambda x: "content" in x, messages)), 0) + response = make_openai_call(GPT_3_5_TURBO, messages, token_count, GRADING_FIELDS, GRADING_TEMPERATURE) + + if isinstance(response, str): + response = re.sub(r"^[a-zA-Z0-9_]+\:\s*", "", response) + + return response + except Exception as e: + return str(e) + @app.route('/writing_task2', methods=['GET']) @jwt_required() diff --git a/helper/api_messages.py b/helper/api_messages.py index b89b721..3548a15 100644 --- a/helper/api_messages.py +++ b/helper/api_messages.py @@ -344,3 +344,37 @@ def get_question_gen_messages(question_type: QuestionType): ] else: raise Exception("Question type not implemented: " + question_type.value) + + +def get_question_tips(question: str, answer: str, correct_answer: str, context: str = None): + messages = [ + { + "role": "system", + "content": "You are a IELTS exam program that analyzes incorrect answers to questions and gives tips to " + "help students understand why it was a wrong answer and explains how their thought process " + "should have been. The tip should refer to the context and question.", + } + ] + + if not (context is None or context == ""): + messages.append({ + "role": "system", + "content": f"This is the context for the question: {context}", + }) + + messages.extend([ + { + "role": "system", + "content": f"This is the question: {question}", + }, + { + "role": "system", + "content": f"This is the answer: {answer}", + }, + { + "role": "system", + "content": f"This is the correct answer: {correct_answer}", + } + ]) + + return messages diff --git a/postman/ielts.postman_collection.json b/postman/ielts.postman_collection.json index 94f4b70..6824066 100644 --- a/postman/ielts.postman_collection.json +++ b/postman/ielts.postman_collection.json @@ -1,9 +1,9 @@ { "info": { - "_postman_id": "2e0eed9d-6a6f-4785-9972-087d51ac0265", + "_postman_id": "c3a09737-c624-4b32-9e9a-af8ee8084764", "name": "ielts", "schema": "https://schema.getpostman.com/json/collection/v2.1.0/collection.json", - "_exporter_id": "26107457" + "_exporter_id": "29491168" }, "item": [ { @@ -257,6 +257,53 @@ } }, "response": [] + }, + { + "name": "Fetch Answer Tips", + "request": { + "auth": { + "type": "bearer", + "bearer": [ + { + "key": "token", + "value": "{{jwt_token}}", + "type": "string" + } + ] + }, + "method": "POST", + "header": [ + { + "key": "Content-Type", + "value": "application/json", + "type": "text" + } + ], + "body": { + "mode": "raw", + "raw": "{\n \"question\": \"When did Kendrick Lamar sign for TDE?\",\n \"answer\": \"Hello GPT.\",\n\t\t\"correct_answer\": \"2005\",\n \"context\": \"Kendrick Lamar Duckworth (born June 17, 1987) is an American rapper and songwriter. Known for his progressive musical styles and socially conscious songwriting, he is often considered one of the most influential hip hop artists of his generation. Born and raised in Compton, California, Lamar began his career as a teenager performing under the stage name K.Dot. He quickly garnered local attention which led to him signing a recording contract with Top Dawg Entertainment (TDE) in 2005.\"\n}\n", + "options": { + "raw": { + "language": "json" + } + } + }, + "url": { + "raw": "http://127.0.0.1:5000/fetch_tips", + "protocol": "http", + "host": [ + "127", + "0", + "0", + "1" + ], + "port": "5000", + "path": [ + "fetch_tips" + ] + } + }, + "response": [] } ] } \ No newline at end of file