Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
+from datasets import load_dataset  # Import the datasets library
 
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
@@ -7,6 +8,11 @@ For more information on `huggingface_hub` Inference API support, please check th
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 
+# Load the dataset from Hugging Face
+dataset = load_dataset("samhog/psychology-10k", split="train")  # Load the training split
+
+
+
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -15,6 +21,18 @@ def respond(
     temperature,
     top_p,
 ):
+    # First, attempt to find a response in the dataset
+    closest_match = None
+    for entry in dataset:
+        if message.lower() in entry["question"].lower():  # Adjust "question" to the actual column name
+            closest_match = entry["answer"]  # Adjust "answer" to the actual column name
+            break
+
+    # If a match is found, return the dataset answer
+    if closest_match:
+        return closest_match
+
+    # If no match is found, use the chatbot model
     messages = [{"role": "system", "content": system_message}]
 
     for val in history: