devops/encoach_backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve speaking corrections to return fixed_text.

Browse Source
This commit is contained in:
Cristiano Ferreira
2024-01-17 16:37:59 +00:00
parent 92c92dfd98
commit 1591f8d9fb
2 changed files with 7 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
helper/openai_interface.py
View File

@@ -257,12 +257,8 @@ def get_fixed_text(text):
return response["fixed_text"]
def get_speaking_corrections(text):
message = ('Given the provided transcription, identify and extract any inaccuracies, including incorrect words and '
'expressions. Put them in the json with the correct words and expressions that should be on the transcription '
'instead. Sample JSON: {"corrections":[{"wrong": "wrong_word", "correct": "correct_word"}] '
'\n The text: "' + text + '"')
message = ('Fix the errors in the provided transcription and put it in a JSON. Sample JSON: {"fixed_text": "fixed '
'transcription with no misspelling errors"}] \n The text: "' + text + '"')
token_count = count_tokens(message)["n_tokens"]
response = make_openai_instruct_call(GPT_3_5_TURBO_INSTRUCT, message, token_count, ["corrections"], 0.2)
# Filter out items with the same value for misspelled and correction
filtered_data = [item for item in response["corrections"] if item['wrong'] != item['correct']]
return filtered_data
response = make_openai_instruct_call(GPT_3_5_TURBO_INSTRUCT, message, token_count, ["fixed_text"], 0.2)
return response["fixed_text"]