Fix mistake on grade speaking task 1.

app.py

@@ -5,7 +5,8 @@ import firebase_admin
 from firebase_admin import credentials
 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.firebase_helper import download_firebase_file, upload_file_firebase, upload_file_firebase_get_url, \
+    save_to_db
 from helper.speech_to_text_helper import speech_to_text, text_to_speech, has_words
 from helper.token_counter import count_tokens
 from helper.openai_interface import make_openai_call
@@ -15,6 +16,8 @@ import re
 
 from dotenv import load_dotenv
 
+from templates.question_templates import getListening1Template, getListening2Template
+
 load_dotenv()
 
 app = Flask(__name__)
@@ -49,10 +52,37 @@ def get_listening_section_1_question():
                              map(lambda x: x["content"], filter(lambda x: "content" in x, messages)), 0)
         response = make_openai_call(GPT_3_5_TURBO_16K, messages, token_count, LISTENING_GEN_FIELDS,
                                     GEN_QUESTION_TEMPERATURE)
+        # file_name = str(uuid.uuid4()) + ".mp3"
+        # sound_file_path = AUDIO_FILES_PATH + file_name
+        # firebase_file_path = FIREBASE_LISTENING_AUDIO_FILES_PATH + file_name
+        # text_to_speech(response["transcript"], sound_file_path)
+        # upload_file_firebase(FIREBASE_BUCKET, firebase_file_path, sound_file_path)
+        # response["audio_file"] = firebase_file_path
         return response
     except Exception as e:
         return str(e)
 
+@app.route('/save_listening_section_1', methods=['POST'])
+@jwt_required()
+def save_listening_section_1_question():
+    try:
+        # data = request.get_json()
+        # question = data.get('question')
+        question = getListening1Template()
+        file_name = str(uuid.uuid4()) + ".mp3"
+        sound_file_path = AUDIO_FILES_PATH + file_name
+        firebase_file_path = FIREBASE_LISTENING_AUDIO_FILES_PATH + file_name
+        # TODO it's the conversation audio, still work to do on text-to-speech
+        text_to_speech(question["audio"]["conversation"], sound_file_path)
+        file_url = upload_file_firebase_get_url(FIREBASE_BUCKET, firebase_file_path, sound_file_path)
+        question["audio"]["source"] = file_url
+        if save_to_db("listening", question):
+            return question
+        else:
+            raise Exception("Failed to save question: " + question)
+    except Exception as e:
+        return str(e)
+
 
 @app.route('/listening_section_2', methods=['GET'])
 @jwt_required()
@@ -75,6 +105,26 @@ def get_listening_section_2_question():
     except Exception as e:
         return str(e)
 
+@app.route('/save_listening_section_2', methods=['POST'])
+@jwt_required()
+def save_listening_section_2_question():
+    try:
+        # data = request.get_json()
+        # question = data.get('question')
+        question = getListening2Template()
+        file_name = str(uuid.uuid4()) + ".mp3"
+        sound_file_path = AUDIO_FILES_PATH + file_name
+        firebase_file_path = FIREBASE_LISTENING_AUDIO_FILES_PATH + file_name
+        text_to_speech(question["audio"]["text"], sound_file_path)
+        file_url = upload_file_firebase_get_url(FIREBASE_BUCKET, firebase_file_path, sound_file_path)
+        question["audio"]["source"] = file_url
+        if save_to_db("listening", question):
+            return question
+        else:
+            raise Exception("Failed to save question: " + question)
+    except Exception as e:
+        return str(e)
+
 
 @app.route('/listening_section_3', methods=['GET'])
 @jwt_required()
@@ -97,6 +147,26 @@ def get_listening_section_3_question():
     except Exception as e:
         return str(e)
 
+@app.route('/save_listening_section_3', methods=['POST'])
+@jwt_required()
+def save_listening_section_3_question():
+    try:
+        # data = request.get_json()
+        # question = data.get('question')
+        question = getListening2Template()
+        file_name = str(uuid.uuid4()) + ".mp3"
+        sound_file_path = AUDIO_FILES_PATH + file_name
+        firebase_file_path = FIREBASE_LISTENING_AUDIO_FILES_PATH + file_name
+        text_to_speech(question["audio"]["text"], sound_file_path)
+        file_url = upload_file_firebase_get_url(FIREBASE_BUCKET, firebase_file_path, sound_file_path)
+        question["audio"]["source"] = file_url
+        if save_to_db("listening", question):
+            return question
+        else:
+            raise Exception("Failed to save question: " + question)
+    except Exception as e:
+        return str(e)
+
 
 @app.route('/listening_section_4', methods=['GET'])
 @jwt_required()
@@ -119,6 +189,26 @@ def get_listening_section_4_question():
     except Exception as e:
         return str(e)
 
+@app.route('/save_listening_section_4', methods=['POST'])
+@jwt_required()
+def save_listening_section_4_question():
+    try:
+        # data = request.get_json()
+        # question = data.get('question')
+        question = getListening2Template()
+        file_name = str(uuid.uuid4()) + ".mp3"
+        sound_file_path = AUDIO_FILES_PATH + file_name
+        firebase_file_path = FIREBASE_LISTENING_AUDIO_FILES_PATH + file_name
+        text_to_speech(question["audio"]["text"], sound_file_path)
+        file_url = upload_file_firebase_get_url(FIREBASE_BUCKET, firebase_file_path, sound_file_path)
+        question["audio"]["source"] = file_url
+        if save_to_db("listening", question):
+            return question
+        else:
+            raise Exception("Failed to save question: " + question)
+    except Exception as e:
+        return str(e)
+
 
 @app.route('/writing_task1', methods=['POST'])
 @jwt_required()
@@ -148,6 +238,21 @@ def grade_writing_task_1():
     except Exception as e:
         return str(e)
 
+@app.route('/save_writing_task_1', methods=['POST'])
+@jwt_required()
+def save_writing_task_1_question():
+    try:
+        # data = request.get_json()
+        # question = data.get('question')
+        # TODO ADD SAVE IMAGE TO DB
+        question = getListening2Template()
+        if save_to_db("writing", question):
+            return question
+        else:
+            raise Exception("Failed to save question: " + question)
+    except Exception as e:
+        return str(e)
+
 
 @app.route('/writing_task2', methods=['POST'])
 @jwt_required()
@@ -211,6 +316,20 @@ def get_writing_task_2_question():
     except Exception as e:
         return str(e)
 
+@app.route('/save_writing_task_2', methods=['POST'])
+@jwt_required()
+def save_writing_task_2_question():
+    try:
+        # data = request.get_json()
+        # question = data.get('question')
+        question = getListening2Template()
+        if save_to_db("writing", question):
+            return question
+        else:
+            raise Exception("Failed to save question: " + question)
+    except Exception as e:
+        return str(e)
+
 
 @app.route('/speaking_task_1', methods=['POST'])
 @jwt_required()
@@ -224,7 +343,7 @@ def grade_speaking_task_1():
 
         download_firebase_file(FIREBASE_BUCKET, answer_firebase_path, sound_file_name)
         answer = speech_to_text(sound_file_name)
-        if has_words("ajajajajd"):
+        if has_words(answer):
             messages = get_grading_messages(QuestionType.SPEAKING_1, question, answer)
             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)