Epikriza / src /postprocessing.py
LukaVidakovic's picture
Upload postprocessing.py
cdc6995 verified
Raw History Blame Contribute Delete
861 Bytes
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