Update app.py

app.py

@@ -83,17 +83,28 @@ def respond(message, history: list[tuple[str, str]], system_message, max_tokens,
     print(datetime.now())
     context = retriever_tool(message)
     
-    prompt = f"""Given the question and supporting documents below, give a comprehensive answer to the question.
-    Respond only to the question asked, response should be relevant to the question and in the same language as the question.
-    Provide the number of the source document when relevant, as well as the link to the document.
-    If you cannot find information, do not give up and try calling your retriever again with different arguments!
-    Always give url of the sources at the end and only answer in the language the question is asked.
+    is_law = client.text_generation(f"""Given the user question below, classify it as either being about "Law" or "Other".
+Do NOT respond with more than one word.
+Question:
+{message}""")
+
+    print(is_law)
+
+    if is_law.lower() != "other":    
+        prompt = f"""Given the question and supporting documents below, give a comprehensive answer to the question.
+Respond only to the question asked, response should be relevant to the question and in the same language as the question.
+Provide the number of the source document when relevant, as well as the link to the document.
+If you cannot find information, do not give up and try calling your retriever again with different arguments!
+Always give url of the sources at the end and only answer in the language the question is asked.
     
-    Question:
-    {message}
+Question:
+{message}
     
-    {context}
-    """
+{context}
+"""
+    else:
+        prompt = f"""A user wrote the following message, please answer him to best of your knowledge in the language of his message:
+{message}"""
     
     messages = [{"role": "system", "content": system_message}]