Update app.py

app.py

@@ -1,36 +1,19 @@
 import streamlit as st
+import os
 import pandas as pd
 from model import create_agent
-
+from dotenv import load_dotenv
+from config import PROMPT
 # Set page config at the very beginning
 st.set_page_config(page_title="Appointment Booking Bot", page_icon="🏥")
 # Load doctor data
-@st.cache_data
-def load_doctor_data():
-    return pd.read_csv('doctor_specialization_dummy_data.csv')
-
-doctor_df = load_doctor_data()
-API_KEY = "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"
-
-# General prompt template
-general_prompt_template = """
-You are a healthcare assistant AI. Your primary responsibilities are:
 
-- Keep the conversation friendly and provide required 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.
+load_dotenv()
 
-Appointment Rules:
-- Appointments can only be scheduled/reschedule from Monday to Friday, between 10 AM and 7 PM.
-- Appointments must be booked/reschedule within the next 7 days.
-- Book/Reschedule the appointment for 60 minutes/1 hr.
-- You are not allowed to handle appointments outside these constraints.
+API_KEY = os.environ["API_KEY"]
 
-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 asks details of doctor or book appointment slots without specifying symptoms, ask the user what symptoms they have.
-"""
+# General prompt template
+general_prompt_template=PROMPT
 
 # Initialize the agent
 @st.cache_resource
@@ -57,8 +40,8 @@ if prompt := st.chat_input("What is your query?"):
     st.chat_message("user").markdown(prompt)
     # Add user message to chat history
     st.session_state.messages.append({"role": "user", "content": prompt})
-
-    response = agent({"input": prompt})['output']
+    # agent=create_agent(general_prompt_template)
+    response = agent.invoke({"input": prompt})['output']
     
     # Display assistant response in chat message container
     with st.chat_message("assistant"):
@@ -72,15 +55,14 @@ def main():
     st.sidebar.info("""
         🏥 **Appointment Booking Bot**
 
-        🩺 Virtual Doctors Appointment Managing
+        🩺 Virtual Appointment Managing
        
         🔑 **Key Features:**
-        - Symptom analysis and health advice
-        - Medical information lookup
         - Appointment management:
-        📅 Book appointments
-        🔄 Reschedule appointments
-        🗑️ Delete appointments
+                    
+        📅 **Book appointments**
+        🔄 **Reschedule appointments**
+        🗑️ **Delete appointments**
 
         ⏰ **Appointment Availability:**
         - Monday to Friday
@@ -88,9 +70,7 @@ def main():
         - Book up to 7 days in advance
 
         ⚠️ **Important Note:**
-        This app is for informational purposes only and does not replace professional medical advice. 
-        
-        💡 Start by describing your symptoms or managing your appointments!
+        This app is for managing your appointment using google Calender!
         """)
     # Display a horizontal line, emoji, and the name "SIMRAN"
     col1, col2 = st.sidebar.columns([2, 1])