From fc879b9fe102d4bbb97e034b686f760710c80d81 Mon Sep 17 00:00:00 2001
From: craigrbarnes <craig.barnes@intelligentlight.com>
Date: Tue, 23 Apr 2024 13:44:14 -0500
Subject: [PATCH] Add Banner to AISearch (#7)

Co-authored-by: craigrbarnes <craigrbarnes@uchicago.edu>
---
 src/pages/AISearch.tsx | 40 ++++++++++++++++++++++++++++++++++++++--
 1 file changed, 38 insertions(+), 2 deletions(-)

diff --git a/src/pages/AISearch.tsx b/src/pages/AISearch.tsx
index 178d3d4..3945a8b 100644
--- a/src/pages/AISearch.tsx
+++ b/src/pages/AISearch.tsx
@@ -1,13 +1,49 @@
 import {
-  DiscoveryPage,
+  NavPageLayout,
+  NavPageLayoutProps,
+  Discovery,
   DiscoveryPageGetServerSideProps as getServerSideProps,
 } from '@gen3/frontend';
+
+import { Center, Text } from "@mantine/core";
+
 import { registerDiscoveryCustomCellRenderers } from '@/lib/Discovery/CustomCellRenderers';
 import { registerDiscoveryStudyPreviewRenderers } from '@/lib/Discovery/CustomRowRenderers';
 
 registerDiscoveryCustomCellRenderers();
 registerDiscoveryStudyPreviewRenderers();
 
-export default DiscoveryPage;
+const DiscoveryPageCustom = ({
+                         headerProps,
+                         footerProps,
+                         discoveryConfig,
+                       }: {
+                          headerProps: any,
+                         footerProps: any,
+                         discoveryConfig:any,
+}): JSX.Element => {
+
+  if (discoveryConfig === undefined) {
+    return (
+        <Center maw={400} h={100} mx="auto">
+          <div>Discovery config is not defined. Page disabled</div>
+        </Center>
+    );
+  }
+
+  return (
+      <NavPageLayout {...{ headerProps, footerProps }}>
+        <div className="flex flex-col w-full">
+        <div className="flex bg-base-lightest rounded-4 justify-center align-middle px-2 pt-2">
+          <Text size="xl">AI Search is a Beta Feature. Please note that this is still in testing and we appreciate your feedback. You must login to use it. </Text>
+        </div>
+
+        <Discovery discoveryConfig={discoveryConfig} />
+        </div>
+      </NavPageLayout>
+  );
+};
+
+export default DiscoveryPageCustom;
 
 export { getServerSideProps };