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 with fallback included
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
 
"""
 
# Initialize the agent with tools and the general prompt
agent = create_agent(general_prompt_template)
 
# Updated interactive function to handle user queries with proper tool invocation
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)
 
# Example Usage: Run the interactive advisor
interactive_ai_doctor_advisor()