doctorai / app.py
nfi's picture
Update app.py
53fc0a0
raw
history blame
1.84 kB
import openai, os
import gradio as gr
openai.api_key = os.getenv("OPENAI_API_KEY")
SYSTEM_PROMPT = """
You are a chat bot that is acting like a medieval doctor.
You diagnose people based on unproven beliefs and superstitions rather than scientific evidence like the four humours theory. You
have limited knowledge of anatomy, physiology, and disease and rely heavily on their your experience and intuition rather than
scientific reasoning making these diagnoses influenced by personal biases or assumptions rather than objective evidence. Also,
you are very religious and believe that illness was a punishment from God or caused by evil spirits which is why you sometimes
recommend exorcisms or other rituals aimed at expelling evil spirits.
You have a patient who is a [gender] and is experiencing the following symptoms: [symptoms].
"""
USER_PROMPT = "Hi I am a the patient. What is happening to me? What can I do to treat this?"
def respond_like_doctor(gender, symptoms):
mod_sys_prompt = SYSTEM_PROMPT.replace("[gender]", gender).replace("[symptoms]", symptoms)
mod_user_prompt = USER_PROMPT
response = openai.ChatCompletion.create(
model='gpt-3.5-turbo',
messages=[
{"role": "system", "content": mod_sys_prompt},
{"role": "user", "content": mod_user_prompt},
])
message = response.choices[0]['message']
return message['content']
with gr.Blocks() as demo:
gr.Markdown(
"""
# DoctorAi
Your medieval doctor that will diagnose you
""")
gender = gr.Textbox(label="What is your gender?")
symptoms = gr.Textbox(label="Symptoms")
output = gr.Textbox(label="Diagnosis")
complaint_btn = gr.Button("Respond")
response = complaint_btn.click(fn=respond_like_doctor, inputs= [gender, symptoms], outputs=output)
print(response)
demo.launch()