File size: 861 Bytes
cdc6995 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | import openai
from dotenv import load_dotenv
load_dotenv()
POSTROCESSINGPROMPT = """Ti si specijalizovan za medicinsku komunikaciju. Tvoj zadatak je da gramatički ispraviš tekst bez menjanja originalnog teksta. Dodatno formatiraj lekarsku epikrizu tako da bude čitljiviji. Dodaj novi red gde je potrebno kako bi se poboljšala čitljivost. Lekarska epikriza sastoji se od stručnih medicinskih termina, tako da uzmi to u obzir prilikom leksičke analize i ispravljanja netačnih termina."""
def postprocessing(old_epikriza):
response = openai.chat.completions.create(
model="gpt-3.5-turbo-0125",
messages=[
{"role": "system", "content": POSTROCESSINGPROMPT},
{"role": "user", "content": old_epikriza},
],
)
epikriza = response.choices[0].message.content
print(epikriza)
return epikriza
|