Update app.py

app.py

@@ -151,7 +151,16 @@ if prompt := st.chat_input("Tell me about YSA"):
                 "distances": results["distances"][0]
             }
         )
-        ref_from_db_search = ref["answers"]
+        special_threshold = 0.3
+        filtered_ref = ref[ref["distances"] < special_threshold]
+        if filtered_ref.shape[0] > 0:
+            st.success("There are highly relevant information in our database.")
+            ref_from_db_search = filtered_ref["answers"]
+            final_ref = filtered_ref
+        else:
+            st.warning("The database may not have relevant information to help your question so please be aware of hallucinations.")
+            ref_from_db_search = ref["answers"]
+            final_ref = ref
 
         engineered_prompt = f"""
             Based on the context: {ref_from_db_search},