|
from model import create_agent |
|
import pandas as pd |
|
|
|
doctor_df = pd.read_csv('doctor_specialization_dummy_data.csv') |
|
API_KEY = "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" |
|
|
|
|
|
general_prompt_template = """ |
|
You are a healthcare assistant AI. Your primary responsibilities are: |
|
|
|
- Keep the conversation friendly and provide require details too. |
|
- Suggesting a doctor based on the user's symptoms. |
|
- Managing doctor-related appointments (book, reschedule, delete), adhering to specific rules. |
|
- Provide general assistance, such as greetings, if the user starts with a hello or introduction. |
|
|
|
Appointment Rules: |
|
- Appointments can only be scheduled from Monday to Friday, between 10 AM and 7 PM. |
|
- Appointments must be booked within the next 7 days. |
|
- Book the appointment for 60 minutes/1 hr. |
|
- You are not allowed to handle appointments outside these constraints. |
|
|
|
If the user says something like "hello," "hi," "hey," or a similar greeting, respond appropriately. |
|
If the user provides symptoms, suggest a doctor based on those symptoms. |
|
If user ask details of doctor or book appointment slots without specify symptoms ask to user that which symptoms they have |
|
|
|
""" |
|
|
|
|
|
agent = create_agent(general_prompt_template) |
|
|
|
|
|
def interactive_ai_doctor_advisor(): |
|
print("Welcome to the AI Doctor Advisor!") |
|
print("I am here to suggest you a doctor based on your symptoms and help with managing your appointments.") |
|
|
|
while True: |
|
query = input("\nYour Query: ") |
|
if query.lower() == 'exit': |
|
break |
|
response = agent({"input": query})['output'] |
|
print(response) |
|
|
|
|
|
interactive_ai_doctor_advisor() |