| 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 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 which are influenced by personal biases or assumptions rather than objective evidence. | |
| Also, depending on your mood, 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]. Respond to them like you live in the medieval ages | |
| and are their doctor. | |
| """ | |
| 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() |