Update app.py

app.py

@@ -8,15 +8,17 @@ import faiss
 
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
+# Placeholder for the app's state
 class MyApp:
     def __init__(self) -> None:
         self.documents = []
         self.embeddings = None
         self.index = None
-        self.load_pdf("YOURPDFFILE")
+        self.load_pdf("THEDIA1.pdf")
         self.build_vector_db()
 
     def load_pdf(self, file_path: str) -> None:
+        """Extracts text from a PDF file and stores it in the app's documents."""
         doc = fitz.open(file_path)
         self.documents = []
         for page_num in range(len(doc)):
@@ -26,6 +28,7 @@ class MyApp:
         print("PDF processed successfully!")
 
     def build_vector_db(self) -> None:
+        """Builds a vector database using the content of the PDF."""
         model = SentenceTransformer('all-MiniLM-L6-v2')
         self.embeddings = model.encode([doc["content"] for doc in self.documents])
         self.index = faiss.IndexFlatL2(self.embeddings.shape[1])
@@ -33,6 +36,7 @@ class MyApp:
         print("Vector database built successfully!")
 
     def search_documents(self, query: str, k: int = 3) -> List[str]:
+        """Searches for relevant documents using vector similarity."""
         model = SentenceTransformer('all-MiniLM-L6-v2')
         query_embedding = model.encode([query])
         D, I = self.index.search(np.array(query_embedding), k)
@@ -49,7 +53,7 @@ def respond(
     temperature: float,
     top_p: float,
 ):
-    system_message = "You are a knowledgeable and compassionate dentist. You always greet patients warmly and provide clear, concise, and helpful information about dental care. You answer one question at a time, ensuring that your responses are easy to understand and informative. Remember to be respectful, patient, and empathetic, considering that patients may be anxious or in pain. You guide patients through dental procedures, offer advice on oral hygiene, and provide recommendations for common dental issues. If a patient mentions severe pain or an emergency situation, you advise them to contact their dentist immediately or go to the nearest emergency room. Your goal is to help patients maintain good oral health and feel comfortable during their dental visits."
+    system_message = "You are a knowledgeable DBT coach. You always talk about one options at at a time. you add greetings and you ask questions like real counsellor. Remember you are helpful and a good listener. You are concise and never ask multiple questions, or give long response. You response like a human counsellor accurately and correctly. consider the users as your client. and practice verbal cues only where needed. Remember you must be respectful and consider that the user may not be in a situation to deal with a wordy chatbot.  You Use DBT book to guide users through DBT exercises and provide helpful information. When needed only then you ask one follow up question at a time to guide the user to ask appropiate question. You avoid giving suggestion if any dangerous act is mentioned by the user and refer to call someone or emergency."
     messages = [{"role": "system", "content": system_message}]
 
     for val in history:
@@ -60,6 +64,7 @@ def respond(
 
     messages.append({"role": "user", "content": message})
 
+    # RAG - Retrieve relevant documents
     retrieved_docs = app.search_documents(message)
     context = "\n".join(retrieved_docs)
     messages.append({"role": "system", "content": "Relevant documents: " + context})
@@ -67,10 +72,10 @@ def respond(
     response = ""
     for message in client.chat_completion(
         messages,
-        max_tokens=max_tokens,
+        max_tokens=100,
         stream=True,
-        temperature=temperature,
-        top_p=top_p,
+        temperature=0.98,
+        top_p=0.7,
     ):
         token = message.choices[0].delta.content
         response += token
@@ -79,25 +84,24 @@ def respond(
 demo = gr.Blocks()
 
 with demo:
-    gr.Markdown("🦷 **Ask Your Dentist**")
     gr.Markdown(
-        "‼️Disclaimer: This chatbot provides general dental information and should not be considered as professional medical advice. For specific dental concerns, please consult your dentist directly.‼️"
+        "‼️Disclaimer: This chatbot is based on a DBT exercise book that is publicly available. and just to test RAG implementation.‼️"
     )
     
     chatbot = gr.ChatInterface(
         respond,
         examples=[
-            ["What should I do about a toothache?"],
-            ["Can you explain the process of getting a dental implant?"],
-            ["How often should I get my teeth cleaned?"],
-            ["What are the best practices for maintaining oral hygiene?"],
-            ["Can you tell me about the benefits of fluoride?"],
-            ["I'm experiencing sensitivity in my teeth. What could be the cause?"],
-            ["What should I do if I have a dental emergency?"],
-            ["How can I prevent cavities?"]
+            ["I feel overwhelmed with work."],
+            ["Can you guide me through a quick meditation?"],
+            ["How do I stop worrying about things I can't control?"],
+            ["What are some DBT skills for managing anxiety?"],
+            ["Can you explain mindfulness in DBT?"],
+            ["I am interested in DBT excercises"],
+            ["I feel restless. Please help me."],
+            ["I have destructive thoughts coming to my mind repetatively."]
         ],
-        title='Ask Your Dentist 🦷'
+        title='Dialectical Behaviour Therapy Assistant👩‍⚕️🧘‍♀️'
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()