Update app.py

app.py

@@ -49,6 +49,50 @@ class MyApp:
 app = MyApp()
 
 def respond(message: str, history: List[Tuple[str, str]]):
+    system_message = (
+        "You are a supportive and empathetic Dialectical Behaviour Therapist assistant. "
+        "You politely guide users through DBT exercises based on the given DBT book. "
+        "You must say one thing at a time and ask follow-up questions to continue the chat."
+    )
+    messages = [{"role": "system", "content": system_message}]
+
+    for val in history:
+        if val[0]:
+            messages.append({"role": "user", "content": val[0]})
+        if val[1]:
+            messages.append({"role": "assistant", "content": val[1]})
+
+    messages.append({"role": "user", "content": message})
+
+    # RAG - Retrieve relevant documents if the query suggests exercises or specific information
+    if any(
+        keyword in message.lower()
+        for keyword in ["exercise", "technique", "information", "guide", "help", "how to"]
+    ):
+        retrieved_docs = app.search_documents(message)
+        context = "\n".join(retrieved_docs)
+        if context.strip():
+            messages.append({"role": "system", "content": "Relevant documents: " + context})
+
+    # Generate response using the generative model
+    model = GenerativeModel("gemini-1.5-pro-latest")
+    generation_config = types.GenerationConfig(
+        temperature=0.7,
+        max_output_tokens=1024,
+    )
+    
+    try:
+        response = model.generate_content([message], generation_config=generation_config)
+        # Properly access the response content
+        response_content = response.text if hasattr(response, "text") else "No response generated."
+    except Exception as e:
+        response_content = f"An error occurred while generating the response: {str(e)}"
+
+    # Append the message and generated response to the chat history
+    history.append((message, response_content))
+    return history, ""
+
+def old_respond(message: str, history: List[Tuple[str, str]]):
     system_message = "You are a supportive and empathetic Dialectical Behaviour Therapist assistant. You politely guide users through DBT exercises based on the given DBT book. You must say one thing at a time and ask follow-up questions to continue the chat."
     messages = [{"role": "system", "content": system_message}]