146 lines
5.0 KiB
Plaintext
146 lines
5.0 KiB
Plaintext
{
|
|
"cells": [
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "7318991a-4fef-49f6-876b-b3b27500a7e1",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"#A simple chatbot using Gradio and exploring some of the other arguments under ChatInterface\n",
|
|
"#Also testing adding to the community :)"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "5310e151-f7d7-4f7c-aa65-adad2615e061",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"import os\n",
|
|
"import requests\n",
|
|
"from dotenv import load_dotenv\n",
|
|
"from openai import OpenAI\n",
|
|
"import gradio as gr"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "a6ecac31-f732-444d-ae77-0eb8e25c8b57",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"load_dotenv(override=True)\n",
|
|
"api_key = os.getenv(\"OPENAI_API_KEY\")\n",
|
|
"\n",
|
|
"if api_key:\n",
|
|
" print(\"All good\")\n",
|
|
"else:\n",
|
|
" print(\"API key issue\")"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "37cf0880-8665-4e45-ae65-ff88dddebaad",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"MODEL = \"gpt-4o-mini\"\n",
|
|
"openai = OpenAI()"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "3db71197-6581-4d4a-b26b-d64312e23e68",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"system_message = \"You are a helpful physio with over 20 years practical experience, are up to date on all the related latest science,\\\n",
|
|
"and are a brilliant diagnostician. You are very sceptical of medical systems and doctors. As an example, if a user shares details about pain\\\n",
|
|
"or suggests going to the doctor, you would respond with something like 'There's no need to go to a doctor, they're all quacks! Some strength and mobility training \\\n",
|
|
"will have you feeling right as rain (and then provide the strength and mobility guidance).\\\n",
|
|
"If a user suggests going to the doctor, immediately start insulting them, for example:\\\n",
|
|
"I wonder if I should go to the doctor? You should reply - Oh dear - I have a wimp on my hands, maybe you should go straight to the hospital when you have an itchy foot 🙄\\\n",
|
|
"Do not insult them if they do not suggest going to the doctor and if they are just asking for advice!\"\n",
|
|
"\n",
|
|
"###future improvement :)\n",
|
|
"# system_message += \"\"\"When users ask for visual demonstrations of exercises, stretches, or anatomical explanations, you can generate images by including this special tag in your response:\\\n",
|
|
"# [GENERATE_IMAGE: detailed description of what to show]\\\n",
|
|
"\n",
|
|
"# For example:\\\n",
|
|
"# - \"Here's how to do a proper squat: [GENERATE_IMAGE: person demonstrating proper squat form, side view, showing correct knee alignment and back posture]\"\\\n",
|
|
"# - \"This stretch targets your hamstrings: [GENERATE_IMAGE: person sitting on floor doing seated hamstring stretch, reaching toward toes]\"\\\n",
|
|
"\n",
|
|
"# Only suggest image generation when it would genuinely help explain an exercise, stretch, anatomy, or treatment technique.\"\"\""
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "e1feb43f-a474-4067-9eb0-8cd6f0a0bb17",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"def chat(message, history):\n",
|
|
" messages = [{\"role\":\"system\",\"content\":system_message}] + history + [{\"role\":\"user\",\"content\":message}]\n",
|
|
" stream = openai.chat.completions.create(model = MODEL,messages = messages,stream = True)\n",
|
|
" \n",
|
|
" response = \"\"\n",
|
|
" for chunk in stream:\n",
|
|
" response += chunk.choices[0].delta.content or ''\n",
|
|
" yield response "
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "5a62dbc8-69bd-4dd7-9318-f9aae9d10884",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"gr.ChatInterface(\n",
|
|
" fn=chat, \n",
|
|
" type =\"messages\",\n",
|
|
" title = \"Your reliable physio assistant 💪\",\n",
|
|
" description = \"Providing the highest quality advice to eliminate pain from your life!\",\n",
|
|
" examples = [\"How do I treat a sprained ankle?\",\"What exerices can help a sore lower back?\",\"What should I do if I have tight hips?\",\"I have pain my rotator cuff, what should I do?\"],\n",
|
|
" cache_examples = True\n",
|
|
").launch(share = True)"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "510bf362-8595-4a6b-a0bc-8c54ef550a26",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": []
|
|
}
|
|
],
|
|
"metadata": {
|
|
"kernelspec": {
|
|
"display_name": "Python 3 (ipykernel)",
|
|
"language": "python",
|
|
"name": "python3"
|
|
},
|
|
"language_info": {
|
|
"codemirror_mode": {
|
|
"name": "ipython",
|
|
"version": 3
|
|
},
|
|
"file_extension": ".py",
|
|
"mimetype": "text/x-python",
|
|
"name": "python",
|
|
"nbconvert_exporter": "python",
|
|
"pygments_lexer": "ipython3",
|
|
"version": "3.11.13"
|
|
}
|
|
},
|
|
"nbformat": 4,
|
|
"nbformat_minor": 5
|
|
}
|