Update app.py
Browse files
app.py
CHANGED
|
@@ -55,11 +55,9 @@ async def check_grammar(transcription: str = Form(...), language: str = Form(...
|
|
| 55 |
# Grammar check
|
| 56 |
grammar_prompt = (
|
| 57 |
f"Briefly check the grammar of the following text in {language}: {transcription}. "
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
"Provide a score from 1 to 10 based on the grammar accuracy, reducing points for incorrect words and make sure to output the score on a new line after two line breaks like 'SCORE='."
|
| 61 |
)
|
| 62 |
-
|
| 63 |
grammar_check_response = client.chat.completions.create(
|
| 64 |
model="llama3-groq-70b-8192-tool-use-preview",
|
| 65 |
messages=[{"role": "user", "content": grammar_prompt}]
|
|
@@ -69,9 +67,8 @@ async def check_grammar(transcription: str = Form(...), language: str = Form(...
|
|
| 69 |
# Vocabulary check
|
| 70 |
vocabulary_prompt = (
|
| 71 |
f"Check the vocabulary accuracy of the following text in {language}: {transcription}. "
|
| 72 |
-
"Identify any word that does not belong to the selected language and flag it. "
|
| 73 |
-
"
|
| 74 |
-
"Provide a score from 1 to 10 based on the vocabulary accuracy reducing points for incorrect words and make sure to output the score on a new line after two line breaks like 'SCORE='."
|
| 75 |
)
|
| 76 |
|
| 77 |
vocabulary_check_response = client.chat.completions.create(
|
|
@@ -96,22 +93,21 @@ async def check_grammar(transcription: str = Form(...), language: str = Form(...
|
|
| 96 |
|
| 97 |
|
| 98 |
def calculate_score(feedback: str) -> int:
|
| 99 |
-
|
| 100 |
-
Calculate score based on feedback content.
|
| 101 |
This function searches for the keyword 'SCORE=' or similar variations
|
| 102 |
(SCORE:, score:, etc.) and extracts the score value.
|
| 103 |
"""
|
|
|
|
| 104 |
import re
|
| 105 |
match = re.search(r'(SCORE=|score=|SCORE:|score:|SCORE = )\s*(\d+)', feedback)
|
| 106 |
|
| 107 |
if match:
|
|
|
|
| 108 |
return int(match.group(2))
|
| 109 |
|
| 110 |
-
|
| 111 |
-
|
| 112 |
-
|
| 113 |
-
# Run the FastAPI app (only needed for local development)
|
| 114 |
-
# Run the FastAPI app (needed for local development)
|
| 115 |
if __name__ == "__main__":
|
| 116 |
import uvicorn
|
| 117 |
uvicorn.run(app, host="0.0.0.0", port=7860)
|
|
|
|
| 55 |
# Grammar check
|
| 56 |
grammar_prompt = (
|
| 57 |
f"Briefly check the grammar of the following text in {language}: {transcription}. "
|
| 58 |
+
"Identify any word that does not belong to the selected language and flag it. Based on the number of incorrect words also check the grammer deeply and carefully "
|
| 59 |
+
"Provide a score from 1 to 10 based on the grammar accuracy, reducing points for incorrect words and make sure to output the score on a new line after two line break like ""SCORE=""."
|
|
|
|
| 60 |
)
|
|
|
|
| 61 |
grammar_check_response = client.chat.completions.create(
|
| 62 |
model="llama3-groq-70b-8192-tool-use-preview",
|
| 63 |
messages=[{"role": "user", "content": grammar_prompt}]
|
|
|
|
| 67 |
# Vocabulary check
|
| 68 |
vocabulary_prompt = (
|
| 69 |
f"Check the vocabulary accuracy of the following text in {language}: {transcription}. "
|
| 70 |
+
"Identify any word that does not belong to the selected language and flag it. Based on the number of incorrect words also check the grammer deeply and carefully "
|
| 71 |
+
"provide a score from 1 to 10,based on the vocabulary accuracy reducing points for incorrect words and make sure to output the score on a new line after two line break like ""SCORE=""."
|
|
|
|
| 72 |
)
|
| 73 |
|
| 74 |
vocabulary_check_response = client.chat.completions.create(
|
|
|
|
| 93 |
|
| 94 |
|
| 95 |
def calculate_score(feedback: str) -> int:
|
| 96 |
+
"""
|
| 97 |
+
Calculate score based on feedback content.
|
| 98 |
This function searches for the keyword 'SCORE=' or similar variations
|
| 99 |
(SCORE:, score:, etc.) and extracts the score value.
|
| 100 |
"""
|
| 101 |
+
# Look for 'SCORE=' or similar variations and extract the score using a regular expression
|
| 102 |
import re
|
| 103 |
match = re.search(r'(SCORE=|score=|SCORE:|score:|SCORE = )\s*(\d+)', feedback)
|
| 104 |
|
| 105 |
if match:
|
| 106 |
+
# Extract and return the score as an integer
|
| 107 |
return int(match.group(2))
|
| 108 |
|
| 109 |
+
# Return a default score of 0 if no score is found in the feedback
|
| 110 |
+
return 0
|
|
|
|
|
|
|
|
|
|
| 111 |
if __name__ == "__main__":
|
| 112 |
import uvicorn
|
| 113 |
uvicorn.run(app, host="0.0.0.0", port=7860)
|