add react app frontend

frontend/.gitignore

@@ -0,0 +1,23 @@
+# See https://help.github.com/articles/ignoring-files/ for more about ignoring files.
+
+# dependencies
+/node_modules
+/.pnp
+.pnp.js
+
+# testing
+/coverage
+
+# production
+/build
+
+# misc
+.DS_Store
+.env.local
+.env.development.local
+.env.test.local
+.env.production.local
+
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*

frontend/README.md

@@ -0,0 +1,70 @@
+# Getting Started with Create React App
+
+This project was bootstrapped with [Create React App](https://github.com/facebook/create-react-app).
+
+## Available Scripts
+
+In the project directory, you can run:
+
+### `npm start`
+
+Runs the app in the development mode.\
+Open [http://localhost:3000](http://localhost:3000) to view it in your browser.
+
+The page will reload when you make changes.\
+You may also see any lint errors in the console.
+
+### `npm test`
+
+Launches the test runner in the interactive watch mode.\
+See the section about [running tests](https://facebook.github.io/create-react-app/docs/running-tests) for more information.
+
+### `npm run build`
+
+Builds the app for production to the `build` folder.\
+It correctly bundles React in production mode and optimizes the build for the best performance.
+
+The build is minified and the filenames include the hashes.\
+Your app is ready to be deployed!
+
+See the section about [deployment](https://facebook.github.io/create-react-app/docs/deployment) for more information.
+
+### `npm run eject`
+
+**Note: this is a one-way operation. Once you `eject`, you can't go back!**
+
+If you aren't satisfied with the build tool and configuration choices, you can `eject` at any time. This command will remove the single build dependency from your project.
+
+Instead, it will copy all the configuration files and the transitive dependencies (webpack, Babel, ESLint, etc) right into your project so you have full control over them. All of the commands except `eject` will still work, but they will point to the copied scripts so you can tweak them. At this point you're on your own.
+
+You don't have to ever use `eject`. The curated feature set is suitable for small and middle deployments, and you shouldn't feel obligated to use this feature. However we understand that this tool wouldn't be useful if you couldn't customize it when you are ready for it.
+
+## Learn More
+
+You can learn more in the [Create React App documentation](https://facebook.github.io/create-react-app/docs/getting-started).
+
+To learn React, check out the [React documentation](https://reactjs.org/).
+
+### Code Splitting
+
+This section has moved here: [https://facebook.github.io/create-react-app/docs/code-splitting](https://facebook.github.io/create-react-app/docs/code-splitting)
+
+### Analyzing the Bundle Size
+
+This section has moved here: [https://facebook.github.io/create-react-app/docs/analyzing-the-bundle-size](https://facebook.github.io/create-react-app/docs/analyzing-the-bundle-size)
+
+### Making a Progressive Web App
+
+This section has moved here: [https://facebook.github.io/create-react-app/docs/making-a-progressive-web-app](https://facebook.github.io/create-react-app/docs/making-a-progressive-web-app)
+
+### Advanced Configuration
+
+This section has moved here: [https://facebook.github.io/create-react-app/docs/advanced-configuration](https://facebook.github.io/create-react-app/docs/advanced-configuration)
+
+### Deployment
+
+This section has moved here: [https://facebook.github.io/create-react-app/docs/deployment](https://facebook.github.io/create-react-app/docs/deployment)
+
+### `npm run build` fails to minify
+
+This section has moved here: [https://facebook.github.io/create-react-app/docs/troubleshooting#npm-run-build-fails-to-minify](https://facebook.github.io/create-react-app/docs/troubleshooting#npm-run-build-fails-to-minify)

frontend/package.json

@@ -0,0 +1,44 @@
+{
+  "name": "semantic-search",
+  "version": "0.1.0",
+  "private": true,
+  "dependencies": {
+    "@emotion/react": "^11.13.0",
+    "@emotion/styled": "^11.13.0",
+    "@mui/icons-material": "^5.16.7",
+    "@mui/material": "^5.16.7",
+    "@testing-library/jest-dom": "^5.17.0",
+    "@testing-library/react": "^13.4.0",
+    "@testing-library/user-event": "^13.5.0",
+    "axios": "^1.7.4",
+    "react": "^18.3.1",
+    "react-dom": "^18.3.1",
+    "react-router-dom": "^6.26.1",
+    "react-scripts": "5.0.1",
+    "web-vitals": "^2.1.4"
+  },
+  "scripts": {
+    "start": "react-scripts start",
+    "build": "react-scripts build",
+    "test": "react-scripts test",
+    "eject": "react-scripts eject"
+  },
+  "eslintConfig": {
+    "extends": [
+      "react-app",
+      "react-app/jest"
+    ]
+  },
+  "browserslist": {
+    "production": [
+      ">0.2%",
+      "not dead",
+      "not op_mini all"
+    ],
+    "development": [
+      "last 1 chrome version",
+      "last 1 firefox version",
+      "last 1 safari version"
+    ]
+  }
+}

frontend/public/index.html

@@ -0,0 +1,43 @@
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="utf-8" />
+    <link rel="icon" href="%PUBLIC_URL%/favicon.ico" />
+    <meta name="viewport" content="width=device-width, initial-scale=1" />
+    <meta name="theme-color" content="#000000" />
+    <meta
+      name="description"
+      content="Web site created using create-react-app"
+    />
+    <link rel="apple-touch-icon" href="%PUBLIC_URL%/logo192.png" />
+    <!--
+      manifest.json provides metadata used when your web app is installed on a
+      user's mobile device or desktop. See https://developers.google.com/web/fundamentals/web-app-manifest/
+    -->
+    <link rel="manifest" href="%PUBLIC_URL%/manifest.json" />
+    <!--
+      Notice the use of %PUBLIC_URL% in the tags above.
+      It will be replaced with the URL of the `public` folder during the build.
+      Only files inside the `public` folder can be referenced from the HTML.
+
+      Unlike "/favicon.ico" or "favicon.ico", "%PUBLIC_URL%/favicon.ico" will
+      work correctly both with client-side routing and a non-root public URL.
+      Learn how to configure a non-root public URL by running `npm run build`.
+    -->
+    <title>React App</title>
+  </head>
+  <body>
+    <noscript>You need to enable JavaScript to run this app.</noscript>
+    <div id="root"></div>
+    <!--
+      This HTML file is a template.
+      If you open it directly in the browser, you will see an empty page.
+
+      You can add webfonts, meta tags, or analytics to this file.
+      The build step will place the bundled scripts into the <body> tag.
+
+      To begin the development, run `npm start` or `yarn start`.
+      To create a production bundle, use `npm run build` or `yarn build`.
+    -->
+  </body>
+</html>

frontend/public/manifest.json

@@ -0,0 +1,25 @@
+{
+  "short_name": "React App",
+  "name": "Create React App Sample",
+  "icons": [
+    {
+      "src": "favicon.ico",
+      "sizes": "64x64 32x32 24x24 16x16",
+      "type": "image/x-icon"
+    },
+    {
+      "src": "logo192.png",
+      "type": "image/png",
+      "sizes": "192x192"
+    },
+    {
+      "src": "logo512.png",
+      "type": "image/png",
+      "sizes": "512x512"
+    }
+  ],
+  "start_url": ".",
+  "display": "standalone",
+  "theme_color": "#000000",
+  "background_color": "#ffffff"
+}

frontend/public/robots.txt

@@ -0,0 +1,3 @@
+# https://www.robotstxt.org/robotstxt.html
+User-agent: *
+Disallow:

frontend/src/App.css

@@ -0,0 +1,28 @@
+.top-bar {
+  display: flex;
+  flex-direction: column;
+  gap: 16px;
+  margin-bottom: 16px;
+}
+
+.source-logo {
+  width: 50px; /* Adjust the size of the logos */
+}
+
+.result {
+  border-radius: 8px;
+  padding: 16px;
+  margin-bottom: 16px;
+  background: transparent; /* Use theme background instead */
+}
+
+h2 {
+  font-size: 1.2em;
+  margin: 0;
+  color: white; /* Ensure headings are visible in dark mode */
+}
+
+p {
+  margin: 8px 0;
+  color: #b0bec5; /* Light grey for paragraph text */
+}

frontend/src/App.js

@@ -0,0 +1,94 @@
+import React, { useState } from "react";
+import { Box, Typography, CircularProgress, Container } from "@mui/material";
+import SearchBar from "./components/SearchBar";
+import ResultsGrid from "./components/ResultsGrid";
+import CombinedResults from "./components/CombinedResults";
+import axios from "axios";
+
+const App = () => {
+  const [results, setResults] = useState({});
+  const [threadId, setThreadId] = useState(null); // State for threadId
+  const [hasSearched, setHasSearched] = useState(false);
+  const [loading, setLoading] = useState(false);
+
+  const handleSearch = async (query) => {
+    setLoading(true);
+    setHasSearched(true);
+
+    try {
+      const response = await axios.post(
+        `http://localhost:8000/query?q=${encodeURIComponent(query)}`,
+      );
+      const { documents, thread_id } = response.data; // Destructure thread_id from the response
+
+      // Preserve original index
+      const documentsWithIndex = documents.map((doc, index) => ({
+        ...doc,
+        originalIndex: index,
+      }));
+
+      const groupedResults = documentsWithIndex.reduce((acc, doc) => {
+        const { source } = doc.metadata;
+        if (!acc[source]) {
+          acc[source] = [];
+        }
+        acc[source].push(doc);
+        return acc;
+      }, {});
+
+      setResults(groupedResults);
+      setThreadId(thread_id); // Set the threadId
+    } catch (error) {
+      console.error("Error fetching query results:", error);
+    } finally {
+      setLoading(false);
+    }
+  };
+
+  const handleClear = () => {
+    setResults({});
+    setThreadId(null);
+    setHasSearched(false);
+  };
+
+  const combinedResults = Object.values(results).flat();
+
+  return (
+    <Container maxWidth="lg">
+      <Box sx={{ py: 4 }}>
+        <Typography
+          variant="h2"
+          align="center"
+          gutterBottom
+          sx={{ cursor: "pointer" }}
+          onClick={handleClear}
+        >
+          Semantic Catalogue Search
+        </Typography>
+        <SearchBar onSearch={handleSearch} />
+        {loading && (
+          <Box sx={{ display: "flex", justifyContent: "center", mt: 4 }}>
+            <CircularProgress />
+          </Box>
+        )}
+        {!loading && hasSearched && (
+          <>
+            <ResultsGrid results={results} threadId={threadId} />
+            <CombinedResults results={combinedResults} threadId={threadId} />
+          </>
+        )}
+        {!hasSearched && !loading && (
+          <Typography
+            variant="h6"
+            sx={{ mt: 4, textAlign: "center", color: "text.secondary" }}
+          >
+            Please perform a search to see results.
+          </Typography>
+        )}
+      </Box>
+    </Container>
+  );
+};
+
+export default App;
+

frontend/src/App.test.js

@@ -0,0 +1,8 @@
+import { render, screen } from '@testing-library/react';
+import App from './App';
+
+test('renders learn react link', () => {
+  render(<App />);
+  const linkElement = screen.getByText(/learn react/i);
+  expect(linkElement).toBeInTheDocument();
+});

frontend/src/components/CombinedResults.js

@@ -0,0 +1,333 @@
+import React, { useState } from "react";
+import {
+  Box,
+  Card,
+  CardContent,
+  Typography,
+  IconButton,
+  Link,
+  Divider,
+  Button,
+  CircularProgress,
+} from "@mui/material";
+import ExpandMoreIcon from "@mui/icons-material/ExpandMore";
+import LinkIcon from "@mui/icons-material/Link"; // For document URL link
+import HelpOutlineIcon from "@mui/icons-material/HelpOutline"; // For explain feature
+import CloseIcon from "@mui/icons-material/Close"; // For collapsing explanation
+import axios from "axios";
+
+const CombinedResults = ({ results = [], threadId }) => {
+  const [expandedDocs, setExpandedDocs] = useState({});
+  const [visibleCount, setVisibleCount] = useState(5); // Number of results to show initially
+  const [explanations, setExplanations] = useState({});
+  const [loadingExplanation, setLoadingExplanation] = useState({});
+  const [showExplanation, setShowExplanation] = useState({});
+
+  // Toggle expand state for a specific document
+  const handleToggleExpand = (index) => {
+    setExpandedDocs((prev) => ({
+      ...prev,
+      [index]: !prev[index],
+    }));
+  };
+
+  const handleLoadMore = () => {
+    setVisibleCount((prevCount) => prevCount + 5); // Show 10 more results
+  };
+
+  // Fetch explanation for a document
+  const handleFetchExplanation = async (originalIndex) => {
+    if (!threadId) {
+      return;
+    }
+
+    // Set loading state immediately
+    setLoadingExplanation((prev) => ({
+      ...prev,
+      [originalIndex]: true,
+    }));
+
+    // Check if the explanation is already available
+    if (explanations[`${threadId}-${originalIndex}`]) {
+      setLoadingExplanation((prev) => ({
+        ...prev,
+        [originalIndex]: false,
+      }));
+      return;
+    }
+
+    try {
+      const response = await axios.get(
+        `http://localhost:8000/explain/${threadId}?docid=${originalIndex}`,
+      );
+      setExplanations((prev) => ({
+        ...prev,
+        [`${threadId}-${originalIndex}`]: response.data,
+      }));
+    } catch (error) {
+      console.error("Error fetching explanation:", error);
+    } finally {
+      setLoadingExplanation((prev) => ({
+        ...prev,
+        [originalIndex]: false,
+      }));
+    }
+  };
+
+  // Toggle explanation visibility
+  const handleToggleExplanation = (originalIndex) => {
+    setShowExplanation((prev) => ({
+      ...prev,
+      [originalIndex]: !prev[originalIndex],
+    }));
+
+    // Fetch explanation if not already available
+    if (!explanations[`${threadId}-${originalIndex}`]) {
+      handleFetchExplanation(originalIndex);
+    }
+  };
+
+  // Format date for display
+  const formatDate = (dateString) => {
+    const date = new Date(dateString);
+    return date.toLocaleDateString(undefined, {
+      year: "numeric",
+      month: "long",
+      day: "numeric",
+    });
+  };
+
+  if (!Array.isArray(results)) {
+    console.warn("Results prop is not an array:", results);
+    return <Typography variant="h6">No results to display.</Typography>;
+  }
+
+  // Slice results to only show the number of visible results
+  const displayedResults = results.slice(0, visibleCount);
+
+  return (
+    <Box sx={{ mt: 4, p: 2 }}>
+      {results.length === 0 ? (
+        <Typography variant="h6" sx={{ textAlign: "center" }}>
+          No results found.
+        </Typography>
+      ) : (
+        <Card
+          sx={{
+            borderRadius: 2,
+            boxShadow: 3,
+            overflow: "hidden",
+            width: "100%",
+            backgroundColor: "background.default",
+          }}
+        >
+          <CardContent
+            sx={{
+              display: "flex",
+              flexDirection: "column",
+              p: 3,
+            }}
+          >
+            {displayedResults.map((doc, index) => (
+              <Box key={index} sx={{ mb: 3 }}>
+                <Box sx={{ display: "flex", alignItems: "center", mb: 1 }}>
+                  <Link
+                    href={doc.metadata.url}
+                    target="_blank"
+                    rel="noopener"
+                    sx={{
+                      textDecoration: "none",
+                      color: "text.primary",
+                      flex: 1,
+                      "&:hover": {
+                        textDecoration: "underline",
+                      },
+                    }}
+                  >
+                    <Typography
+                      variant="h6"
+                      sx={{
+                        fontSize: "1.125rem",
+                        fontWeight: "bold",
+                        overflow: "hidden",
+                        textOverflow: "ellipsis",
+                      }}
+                    >
+                      {index + 1 + ". " + doc.metadata.title || "Untitled"}
+                    </Typography>
+                  </Link>
+                  <IconButton
+                    href={doc.metadata.url}
+                    target="_blank"
+                    rel="noopener"
+                    sx={{
+                      ml: 1,
+                      color: "primary.main",
+                      transition: "color 0.3s",
+                      "&:hover": {
+                        color: "primary.dark",
+                      },
+                    }}
+                  >
+                    <LinkIcon />
+                  </IconButton>
+                  <IconButton
+                    onClick={() => handleToggleExpand(index)}
+                    sx={{
+                      ml: 2,
+                      color: expandedDocs[index]
+                        ? "text.primary"
+                        : "primary.main",
+                    }}
+                  >
+                    <ExpandMoreIcon
+                      sx={{
+                        color: "secondary.light",
+                        transform: expandedDocs[index]
+                          ? "rotate(180deg)"
+                          : "rotate(0deg)",
+                        transition: "transform 0.3s",
+                      }}
+                    />
+                  </IconButton>
+                </Box>
+
+                {/* Date Created */}
+                <Typography
+                  variant="caption"
+                  sx={{
+                    color: "text.secondary",
+                    mb: 1,
+                    fontStyle: "italic",
+                  }}
+                >
+                  {doc.metadata.date_created
+                    ? `Created on: ${formatDate(doc.metadata.date_created)}`
+                    : "Date not available"}
+                </Typography>
+
+                <Typography
+                  variant="body2"
+                  sx={{
+                    fontSize: "0.875rem",
+                    color: expandedDocs[index]
+                      ? "text.secondary"
+                      : "text.disabled", // Change color based on expansion
+                    overflow: "hidden",
+                    textOverflow: "none",
+                    WebkitBoxOrient: "vertical",
+                    WebkitLineClamp: expandedDocs[index] ? "none" : 3, // Limit lines when not expanded
+                    backgroundColor: "background.default",
+                    // display: "-webkit-box",
+                    p: 2,
+                    borderRadius: 1,
+                    lineHeight: "1.5",
+                    maxHeight: expandedDocs[index] ? "100em" : "5em", // Control max height for transition
+                    transition: "color 0.5s ease, max-height 0.5s ease", // Add transition for color and max-height
+                  }}
+  dangerouslySetInnerHTML={{
+    __html: doc.page_content?.startsWith("Dataset Title:")
+      ? doc.page_content.slice(doc.metadata.title.length + 15, 5000).replace(/\n/g, '<p></p>')
+      : (doc.page_content || "No preview available.").replace(/\n/g, '<p></p>')
+  }}
+                >
+                </Typography>
+
+                {/* Explain Feature */}
+                <Box sx={{ mt: 2 }}>
+                  <Button
+                    variant="contained"
+                    color="primary"
+                    startIcon={
+                      showExplanation[doc.originalIndex] ? (
+                        <CloseIcon />
+                      ) : (
+                        <HelpOutlineIcon />
+                      )
+                    }
+                    onClick={() => handleToggleExplanation(doc.originalIndex)}
+                    sx={{
+                      fontWeight: "bold",
+                      borderRadius: 1,
+                      textTransform: "none",
+                      mb: 1,
+                      transition: "background-color 0.3s",
+                      "&:hover": {
+                        backgroundColor: "info.dark",
+                      },
+                    }}
+                  >
+                    {showExplanation[doc.originalIndex]
+                      ? "Collapse Explanation"
+                      : "Ask AI"}
+                  </Button>
+
+                  {showExplanation[doc.originalIndex] && (
+                    <Box
+                      sx={{
+                        mt: 2,
+                        p: 2,
+                        borderRadius: 2,
+                        backgroundColor: "info.light",
+                        color: "info.contrastText",
+                        border: "1px solid",
+                        borderColor: "info.dark",
+                        maxHeight: "12em",
+                        overflowY: "auto",
+                        textAlign: "justify",
+                        scrollbarWidth: "none",
+                        msOverflowStyle: "none",
+                        "&::-webkit-scrollbar": {
+                          display: "none",
+                        },
+                      }}
+                    >
+                      <Typography
+                        variant="subtitle2"
+                        sx={{ mb: 1, fontWeight: "bold" }}
+                      >
+                        AI Explanation
+                      </Typography>
+                      {loadingExplanation[doc.originalIndex] ? (
+                        <Box
+                          sx={{
+                            display: "flex",
+                            justifyContent: "center",
+                            alignItems: "center",
+                          }}
+                        >
+                          <CircularProgress size={20} color="inherit" />
+                        </Box>
+                      ) : (
+                        <Typography
+                          variant="body2"
+                          sx={{ fontSize: "0.875rem", lineHeight: "1.6" }}
+                        >
+                          {explanations[`${threadId}-${doc.originalIndex}`]
+                            ?.generation || "No explanation available."}
+                        </Typography>
+                      )}
+                    </Box>
+                  )}
+                </Box>
+                <Divider sx={{ my: 2 }} />
+              </Box>
+            ))}
+            {visibleCount < results.length && (
+              <Button
+                onClick={handleLoadMore}
+                variant="contained"
+                color="secondary"
+                sx={{ mt: 2, alignSelf: "center" }}
+              >
+                Load More
+              </Button>
+            )}
+          </CardContent>
+        </Card>
+      )}
+    </Box>
+  );
+};
+
+export default CombinedResults;

frontend/src/components/DocumentItem.js

@@ -0,0 +1,129 @@
+import React, { useState } from "react";
+import {
+  Typography,
+  Box,
+  IconButton,
+  Link,
+  Collapse,
+  Tooltip,
+  Chip,
+  Paper,
+} from "@mui/material";
+import ExpandMoreIcon from "@mui/icons-material/ExpandMore";
+import HelpOutlineIcon from "@mui/icons-material/HelpOutline";
+import CloseIcon from "@mui/icons-material/Close";
+import ExplanationBox from "./ExplanationBox";
+
+function DocumentItem({ document, threadId }) {
+  const [expanded, setExpanded] = useState(false);
+  const [explanationVisible, setExplanationVisible] = useState(false);
+
+  const handleToggleExpand = () => {
+    setExpanded(!expanded);
+  };
+
+  const handleToggleExplanation = () => {
+    setExplanationVisible(!explanationVisible);
+  };
+
+  return (
+    <Paper sx={{ mb: 0, overflow: "hidden" }}>
+      <Box
+        sx={{
+          display: "flex",
+          alignItems: "center",
+          p: 0,
+          bgcolor: "background.paper",
+        }}
+      >
+        <Box sx={{ flex: 1, minWidth: 0 }}>
+          <Link
+            href={document.metadata.url}
+            target="_blank"
+            rel="noopener"
+            sx={{
+              textDecoration: "none",
+              color: "primary.main",
+            }}
+          >
+            <Typography
+              variant="subtitle2"
+              sx={{
+                fontWeight: "bold",
+                fontSize: "0.75rem",
+                overflow: "hidden",
+                textOverflow: "ellipsis",
+                whiteSpace: "nowrap",
+                color: "text.secondary",
+                "&:hover": {
+                  color: "primary.main",
+                },
+              }}
+            >
+              {document.metadata.title || "Untitled"}
+            </Typography>
+          </Link>
+        </Box>
+        <IconButton onClick={handleToggleExpand} size="small">
+          <ExpandMoreIcon
+            sx={{
+              color: "secondary.light",
+              transform: expanded ? "rotate(180deg)" : "rotate(0deg)",
+              transition: "transform 0.3s",
+            }}
+          />
+        </IconButton>
+        <Tooltip
+          title={explanationVisible ? "Hide Explanation" : "Show Explanation"}
+        >
+          <IconButton onClick={handleToggleExplanation} size="small">
+            {explanationVisible ? (
+              <CloseIcon sx={{ color: "error.main" }} fontSize="small" />
+            ) : (
+              <HelpOutlineIcon sx={{ color: "info.dark" }} fontSize="small" />
+            )}
+          </IconButton>
+        </Tooltip>
+      </Box>
+      <Collapse in={expanded}>
+        <Box sx={{ p: 0.5, bgcolor: "background.paper" }}>
+          <Typography
+            variant="body2"
+            sx={{
+              color: "text.secondary",
+              fontSize: "0.75rem",
+              mb: 0.5,
+              overflow: "hidden",
+              textOverflow: "ellipsis",
+              display: "-webkit-box",
+              WebkitLineClamp: 10,
+              WebkitBoxOrient: "vertical",
+            }}
+          >
+            {document.page_content || "No preview available."}
+          </Typography>
+          <Typography
+            variant="caption"
+            sx={{
+              mb: 0.5,
+              color: "text.disabled",
+              fontStyle: "italic",
+            }}
+          >
+            {`Created: ${new Date(document.metadata.date_created).toLocaleDateString()}`}
+          </Typography>
+        </Box>
+      </Collapse>
+      {explanationVisible && (
+        <Box sx={{ p: 0.5, bgcolor: "background.paper" }}>
+          <ExplanationBox
+            documentId={document.originalIndex}
+            threadId={threadId}
+          />
+        </Box>
+      )}
+    </Paper>
+  );
+}
+
+export default DocumentItem;

frontend/src/components/ExplanationBox.js

@@ -0,0 +1,118 @@
+import React, { useState, useEffect } from "react";
+import {
+  Box,
+  Typography,
+  CircularProgress,
+  Paper,
+  Tooltip,
+} from "@mui/material";
+import axios from "axios";
+
+// Create a cache object to store explanations
+const explanationCache = {};
+
+function ExplanationBox({ documentId, threadId }) {
+  const [explanation, setExplanation] = useState(null);
+  const [loading, setLoading] = useState(true);
+  const [error, setError] = useState(null);
+
+  useEffect(() => {
+    const fetchExplanation = async () => {
+      setLoading(true);
+      setError(null);
+      try {
+        // Check the cache before making a request
+        if (explanationCache[`${documentId}-${threadId}`]) {
+          setExplanation(explanationCache[`${documentId}-${threadId}`]);
+          setLoading(false);
+          return;
+        }
+
+        const response = await axios.get(
+          `http://localhost:8000/explain/${threadId}?docid=${documentId}`,
+        );
+        setExplanation(response.data);
+        // Store the fetched explanation in the cache
+        explanationCache[`${documentId}-${threadId}`] = response.data;
+      } catch (error) {
+        console.error("Error fetching explanation:", error);
+        setError("Failed to fetch explanation");
+      } finally {
+        setLoading(false);
+      }
+    };
+
+    fetchExplanation();
+  }, [documentId, threadId]);
+
+  if (loading) {
+    return (
+      <Box sx={{ display: "flex", justifyContent: "center", py: 2 }}>
+        <CircularProgress size={20} color="primary" />
+      </Box>
+    );
+  }
+
+  if (error) {
+    return (
+      <Typography variant="body2" color="error" sx={{ py: 1 }}>
+        {error}
+      </Typography>
+    );
+  }
+
+  const renderExplanationWithLinks = (text, chunks) => {
+    const parts = text.split(/(\[\d+\])/);
+    return parts.map((part, index) => {
+      const match = part.match(/\[(\d+)\]/);
+      if (match) {
+        const chunkIndex = parseInt(match[1]);
+        return (
+          <Tooltip
+            key={index + 1}
+            title={chunks[chunkIndex]?.page_content || "No content available"}
+            arrow
+            placement="top"
+          >
+            <Typography
+              component="span"
+              sx={{
+                fontSize: "0.75em",
+                color: "text.secondary",
+                cursor: "pointer",
+                "&:hover": {
+                  textDecoration: "underline",
+                  color: "primary.main",
+                },
+              }}
+            >
+              {`[${chunkIndex + 1}]`}
+            </Typography>
+          </Tooltip>
+        );
+      }
+      return <span key={index + 1}>{part}</span>;
+    });
+  };
+
+  return (
+    <Paper
+      elevation={0}
+      sx={{ p: 2, bgcolor: "primary.main", borderRadius: 1 }}
+    >
+      <Typography
+        variant="body2"
+        sx={{ color: "secondary.contrastText", fontSize: "0.75em" }}
+      >
+        {explanation?.generation
+          ? renderExplanationWithLinks(
+              explanation.generation,
+              explanation.chunks,
+            )
+          : "No explanation available."}
+      </Typography>
+    </Paper>
+  );
+}
+
+export default ExplanationBox;

frontend/src/components/ResultsGrid.js

@@ -0,0 +1,62 @@
+import React from "react";
+import {
+  Grid,
+  Typography,
+  Box,
+  Card,
+  CardContent,
+  Button,
+} from "@mui/material";
+import SourceCard from "./SourceCard";
+
+function ResultsGrid({ results, threadId }) {
+  if (!results || Object.keys(results).length === 0) {
+    return (
+      <Typography
+        variant="h6"
+        sx={{ p: 2, textAlign: "center", color: "text.secondary" }}
+      >
+        No results found.
+      </Typography>
+    );
+  }
+
+  return (
+    <Box sx={{ p: 2 }}>
+      <Typography
+        variant="h4"
+        sx={{ mb: 4, color: "text.primary", textAlign: "center" }}
+      ></Typography>
+      <Grid container spacing={3}>
+        {Object.keys(results).map((source, idx) => (
+          <Grid item xs={12} sm={6} md={6} key={idx}>
+            <Card
+              sx={{
+                borderRadius: 2,
+                boxShadow: 3,
+                overflow: "hidden",
+                bgcolor: "background.paper",
+              }}
+            >
+              <CardContent sx={{ p: 2, bgcolor: "background.paper" }}>
+                <Typography
+                  variant="subtitle1"
+                  sx={{ fontWeight: "bold", mb: 2 }}
+                >
+                  {source}
+                </Typography>
+                <SourceCard
+                  source={source}
+                  documents={results[source]}
+                  threadId={threadId}
+                />
+              </CardContent>
+            </Card>
+          </Grid>
+        ))}
+      </Grid>
+    </Box>
+  );
+}
+
+export default ResultsGrid;

frontend/src/components/ResultsGrid_tmp.js

@@ -0,0 +1,356 @@
+import React, { useState } from "react";
+import {
+  Grid,
+  Card,
+  CardContent,
+  Typography,
+  Box,
+  Divider,
+  IconButton,
+  Link,
+  CircularProgress,
+} from "@mui/material";
+import ExpandMoreIcon from "@mui/icons-material/ExpandMore";
+import ChevronLeftIcon from "@mui/icons-material/ChevronLeft";
+import ChevronRightIcon from "@mui/icons-material/ChevronRight";
+import HelpOutlineIcon from "@mui/icons-material/HelpOutline"; // Icon for explanations
+import CloseIcon from "@mui/icons-material/Close"; // Icon to close explanation
+import axios from "axios";
+
+function ResultsGrid({ results, threadId }) {
+  const [expandedDocs, setExpandedDocs] = useState({});
+  const [pageIndices, setPageIndices] = useState({});
+  const [explanations, setExplanations] = useState({});
+  const [loadingExplanation, setLoadingExplanation] = useState({});
+  const [explanationVisible, setExplanationVisible] = useState({}); // Track visibility of explanations
+
+  const handleToggleExpand = (source, index) => {
+    setExpandedDocs((prev) => ({
+      ...prev,
+      [`${source}-${index}`]: !prev[`${source}-${index}`],
+    }));
+  };
+
+  const handlePageChange = (source, direction) => {
+    setPageIndices((prev) => ({
+      ...prev,
+      [source]: Math.max(
+        0,
+        Math.min(
+          results[source].length - 4,
+          (prev[source] || 0) + direction * 4,
+        ),
+      ),
+    }));
+  };
+
+  const handleFetchExplanation = async (originalIndex) => {
+    if (explanationVisible[originalIndex]) {
+      setExplanationVisible((prev) => ({
+        ...prev,
+        [originalIndex]: false,
+      }));
+      return;
+    }
+
+    setExplanationVisible((prev) => ({
+      ...prev,
+      [originalIndex]: true,
+    }));
+
+    if (explanations[`${threadId}-${originalIndex}`]) return;
+
+    setLoadingExplanation((prev) => ({
+      ...prev,
+      [originalIndex]: true,
+    }));
+
+    try {
+      const response = await axios.get(
+        `http://localhost:8000/explain/${threadId}?docid=${originalIndex}`,
+      );
+      setExplanations((prev) => ({
+        ...prev,
+        [`${threadId}-${originalIndex}`]: response.data,
+      }));
+    } catch (error) {
+      console.error("Error fetching explanation:", error);
+    } finally {
+      setLoadingExplanation((prev) => ({
+        ...prev,
+        [originalIndex]: false,
+      }));
+    }
+  };
+
+  if (!results || Object.keys(results).length === 0) {
+    return (
+      <Typography variant="h6" sx={{ p: 2 }}>
+        No results found.
+      </Typography>
+    );
+  }
+
+  return (
+    <Box sx={{ p: 2 }}>
+      <Grid container spacing={3}>
+        {Object.keys(results).map((source, idx) => (
+          <Grid item xs={12} sm={6} md={6} key={idx}>
+            <Card
+              sx={{
+                borderRadius: 2,
+                boxShadow: 3,
+                overflow: "hidden",
+                display: "flex",
+                flexDirection: "column",
+                height: "auto",
+                bgcolor: "background.paper",
+                transition: "transform 0.2s, box-shadow 0.2s", // Smooth transition on hover
+                "&:hover": {
+                  transform: "translateY(-4px)", // Lift card slightly on hover
+                  boxShadow: 6,
+                },
+              }}
+            >
+              <CardContent
+                sx={{
+                  display: "flex",
+                  flexDirection: "column",
+                  height: "100%",
+                }}
+              >
+                <Box
+                  sx={{
+                    display: "flex",
+                    justifyContent: "space-between",
+                    p: 1,
+                  }}
+                >
+                  <IconButton
+                    onClick={() => handlePageChange(source, -1)}
+                    disabled={(pageIndices[source] || 0) === 0}
+                    sx={{
+                      color: "text.secondary",
+                      "&:hover": {
+                        color: "text.primary",
+                      },
+                    }}
+                  >
+                    <ChevronLeftIcon />
+                  </IconButton>
+                  <Typography
+                    variant="h6"
+                    gutterBottom
+                    sx={{
+                      textAlign: "center",
+                      mb: 0,
+                      fontWeight: "bold",
+                      fontSize: "1.2rem",
+                      color: "text.secondary",
+                    }}
+                  >
+                    {source}
+                  </Typography>
+                  <IconButton
+                    onClick={() => handlePageChange(source, 1)}
+                    disabled={
+                      (pageIndices[source] || 0) + 4 >= results[source].length
+                    }
+                    sx={{
+                      color: "text.secondary",
+                      "&:hover": {
+                        color: "text.primary",
+                      },
+                    }}
+                  >
+                    <ChevronRightIcon />
+                  </IconButton>
+                </Box>
+                {results[source]
+                  .slice(
+                    pageIndices[source] || 0,
+                    (pageIndices[source] || 0) + 4,
+                  )
+                  .map((document, index) => (
+                    <Box key={index} sx={{ mb: 2 }}>
+                      <Box
+                        sx={{ display: "flex", alignItems: "center", mb: 1 }}
+                      >
+                        <Link
+                          href={document.metadata.url}
+                          target="_blank"
+                          rel="noopener"
+                          sx={{
+                            textDecoration: "none",
+                            flex: 1,
+                            "&:hover": {
+                              textDecoration: "underline",
+                            },
+                          }}
+                        >
+                          <Typography
+                            variant="body1"
+                            sx={{
+                              fontSize: "0.9rem",
+                              fontWeight: "bold",
+                              overflow: "hidden",
+                              textOverflow: "ellipsis",
+                              color: "text.primary",
+                            }}
+                          >
+                            {document.metadata.title || "Untitled"}
+                          </Typography>
+                        </Link>
+                        <IconButton
+                          onClick={() => handleToggleExpand(source, index)}
+                          sx={{
+                            ml: 1,
+                            color: expandedDocs[`${source}-${index}`]
+                              ? "primary.main"
+                              : "text.primary",
+                            fontSize: "1rem",
+                            transition: "color 0.3s",
+                            "&:hover": {
+                              color: "primary.dark",
+                            },
+                          }}
+                        >
+                          <ExpandMoreIcon
+                            sx={{
+                              transform: expandedDocs[`${source}-${index}`]
+                                ? "rotate(180deg)"
+                                : "rotate(0deg)",
+                              transition: "transform 0.3s",
+                            }}
+                          />
+                        </IconButton>
+                        <IconButton
+                          onClick={() =>
+                            handleFetchExplanation(document.originalIndex)
+                          }
+                          sx={{
+                            ml: 1,
+                            color: explanationVisible[document.originalIndex]
+                              ? "error.main"
+                              : "info.main",
+                            fontSize: "1.2rem",
+                            transition: "color 0.3s",
+                            "&:hover": {
+                              color: explanationVisible[document.originalIndex]
+                                ? "error.dark"
+                                : "info.dark",
+                            },
+                          }}
+                        >
+                          {explanationVisible[document.originalIndex] ? (
+                            <CloseIcon />
+                          ) : (
+                            <HelpOutlineIcon />
+                          )}
+                        </IconButton>
+                      </Box>
+                      <Box
+                        sx={{
+                          maxHeight: expandedDocs[`${source}-${index}`]
+                            ? "none"
+                            : "5em",
+                          overflow: "hidden",
+                          transition: "max-height 0.3s ease",
+                          position: "relative",
+                        }}
+                      >
+                        <Typography
+                          variant="body2"
+                          sx={{
+                            fontSize: "0.85rem",
+                            color: "text.secondary",
+                            backgroundColor: "background.default",
+                            p: 2,
+                            borderRadius: 1,
+                            maxHeight: expandedDocs[`${source}-${index}`]
+                              ? "none"
+                              : "calc(1.25em * 10)",
+                            overflow: "hidden",
+                            textOverflow: "ellipsis",
+                            whiteSpace: "normal",
+                            scrollbarWidth: "none",
+                            msOverflowStyle: "none",
+                            "&::-webkit-scrollbar": {
+                              display: "none",
+                            },
+                            textAlign: "justify",
+                            lineHeight: "1.6",
+                          }}
+                        >
+                          {/* Conditionally render page_content based on its starting text */}
+                          {document.page_content?.startsWith("Dataset Title:")
+                            ? document.page_content.slice(
+                                document.metadata.title.length + 15,
+                                50000,
+                              )
+                            : document.page_content || "No preview available."}
+                        </Typography>
+                      </Box>
+                      {/* Explanation Box */}
+                      {explanationVisible[document.originalIndex] && (
+                        <Box
+                          sx={{
+                            mt: 2,
+                            p: 2,
+                            borderRadius: 1,
+                            bgcolor: "info.light",
+                            color: "info.contrastText",
+                            border: "1px solid",
+                            borderColor: "info.dark",
+                            overflowY: "auto",
+                            textAlign: "justify",
+                            scrollbarWidth: "none",
+                            msOverflowStyle: "none",
+                            "&::-webkit-scrollbar": {
+                              display: "none",
+                            },
+                            position: "relative",
+                          }}
+                        >
+                          <Typography
+                            variant="subtitle2"
+                            sx={{ mb: 1, fontWeight: "bold" }}
+                          >
+                            Explanation
+                          </Typography>
+                          {loadingExplanation[document.originalIndex] ? (
+                            <Box
+                              sx={{
+                                display: "flex",
+                                justifyContent: "center",
+                                alignItems: "center",
+                                height: "100%",
+                              }}
+                            >
+                              <CircularProgress size={20} color="inherit" />
+                            </Box>
+                          ) : (
+                            <Typography
+                              variant="body2"
+                              sx={{ fontSize: "0.85rem", lineHeight: "1.6" }}
+                            >
+                              {explanations[
+                                `${threadId}-${document.originalIndex}`
+                              ]?.generation || "No explanation available."}
+                            </Typography>
+                          )}
+                        </Box>
+                      )}
+                      <Divider sx={{ my: 1 }} />
+                    </Box>
+                  ))}
+              </CardContent>
+            </Card>
+          </Grid>
+        ))}
+      </Grid>
+    </Box>
+  );
+}
+
+export default ResultsGrid;

frontend/src/components/SearchBar.js

@@ -0,0 +1,40 @@
+import React, { useState } from 'react';
+import { TextField, Button, Box } from '@mui/material';
+
+const SearchBar = ({ onSearch }) => {
+  const [query, setQuery] = useState('');
+
+  const handleSearchClick = () => {
+    onSearch(query);
+  };
+
+  const handleKeyDown = (e) => {
+    if (e.key === 'Enter') {
+      handleSearchClick();
+    }
+  };
+
+  return (
+    <Box sx={{ display: 'flex', justifyContent: 'center', my: 4 }}>
+      <TextField
+        variant="outlined"
+        label="Search"
+        fullWidth
+        sx={{ maxWidth: '800px' }}
+        value={query}
+        onChange={(e) => setQuery(e.target.value)}
+        onKeyDown={handleKeyDown} // Add this line to handle Enter key press
+      />
+      <Button
+        variant="contained"
+        color="primary"
+        onClick={handleSearchClick}
+        sx={{ ml: 2 }}
+      >
+        Search
+      </Button>
+    </Box>
+  );
+};
+
+export default SearchBar;

frontend/src/components/SourceCard.js

@@ -0,0 +1,102 @@
+import React, { useState } from "react";
+import {
+  Card,
+  CardContent,
+  Typography,
+  Box,
+  IconButton,
+  Divider,
+} from "@mui/material";
+import ChevronLeftIcon from "@mui/icons-material/ChevronLeft";
+import ChevronRightIcon from "@mui/icons-material/ChevronRight";
+import DocumentItem from "./DocumentItem";
+
+function SourceCard({ source, documents, threadId }) {
+  const [pageIndex, setPageIndex] = useState(0);
+  const itemsPerPage = 8;
+
+  const handlePageChange = (direction) => {
+    setPageIndex((prev) =>
+      Math.max(
+        0,
+        Math.min(
+          documents.length - itemsPerPage,
+          prev + direction * itemsPerPage,
+        ),
+      ),
+    );
+  };
+
+  return (
+    <Card
+      sx={{
+        borderRadius: 0,
+        boxShadow: 0,
+        overflow: "hidden",
+        display: "flex",
+        flexDirection: "column",
+        background: "primary.main",
+      }}
+    >
+      <Divider />
+      <CardContent sx={{ p: 0 }}>
+        <Box
+          sx={{
+            display: "flex",
+            flexDirection: "column",
+            bgcolor: "background.paper",
+          }}
+        >
+          {documents
+            .slice(pageIndex, pageIndex + itemsPerPage)
+            .map((document, index) => (
+              <DocumentItem
+                key={`${document.metadata.id}-${pageIndex + index}`}
+                document={document}
+                threadId={threadId}
+              />
+            ))}
+        </Box>
+      </CardContent>
+      <Divider />
+      <Box
+        sx={{
+          p: 0,
+          bgcolor: "background.paper",
+          borderRadius: 0,
+          boxShadow: 0,
+        }}
+      >
+        <Box
+          sx={{
+            display: "flex",
+            justifyContent: "space-between",
+            alignItems: "center",
+          }}
+        >
+          <IconButton
+            onClick={() => handlePageChange(-1)}
+            disabled={pageIndex === 0}
+            size="small"
+            sx={{ color: "primary.main" }}
+          >
+            <ChevronLeftIcon />
+          </IconButton>
+          <Typography variant="caption" sx={{ color: "text.secondary" }}>
+            {`${pageIndex + 1} - ${Math.min(pageIndex + itemsPerPage, documents.length)} of ${documents.length}`}
+          </Typography>
+          <IconButton
+            onClick={() => handlePageChange(1)}
+            disabled={pageIndex + itemsPerPage >= documents.length}
+            size="small"
+            sx={{ color: "primary.main" }}
+          >
+            <ChevronRightIcon />
+          </IconButton>
+        </Box>
+      </Box>
+    </Card>
+  );
+}
+
+export default SourceCard;

frontend/src/index.css

@@ -0,0 +1,13 @@
+body {
+  margin: 0;
+  font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'Roboto', 'Oxygen',
+    'Ubuntu', 'Cantarell', 'Fira Sans', 'Droid Sans', 'Helvetica Neue',
+    sans-serif;
+  -webkit-font-smoothing: antialiased;
+  -moz-osx-font-smoothing: grayscale;
+}
+
+code {
+  font-family: source-code-pro, Menlo, Monaco, Consolas, 'Courier New',
+    monospace;
+}

frontend/src/index.js

@@ -0,0 +1,54 @@
+import React from 'react';
+import ReactDOM from 'react-dom';
+import App from './App';
+import { ThemeProvider, createTheme, CssBaseline } from '@mui/material';
+
+const theme = createTheme({
+  palette: {
+    mode: 'dark', // Set to dark mode
+    primary: {
+      main: '#1976d2', // Default MUI primary color
+    },
+    secondary: {
+      main: '#dc004e', // Default MUI secondary color
+    },
+    background: {
+      default: '#121212', // Dark background
+      paper: '#1e1e1e', // Slightly lighter background for cards
+    },
+    text: {
+      primary: '#ffffff', // White text for primary
+      secondary: '#b0bec5', // Light grey for secondary text
+    },
+  },
+  typography: {
+    fontFamily: '"Roboto", "Helvetica", "Arial", sans-serif',
+    h1: {
+      fontSize: '2rem',
+      fontWeight: 700,
+    },
+    h2: {
+      fontSize: '1.5rem',
+      fontWeight: 600,
+    },
+    body1: {
+      fontSize: '1rem',
+      lineHeight: 1.5,
+    },
+    body2: {
+      fontSize: '0.875rem',
+      lineHeight: 1.4,
+    },
+  },
+  shape: {
+    borderRadius: 8, // Rounded corners for cards and buttons
+  },
+});
+
+ReactDOM.render(
+  <ThemeProvider theme={theme}>
+    <CssBaseline />
+    <App />
+  </ThemeProvider>,
+  document.getElementById('root')
+);

frontend/src/reportWebVitals.js

@@ -0,0 +1,13 @@
+const reportWebVitals = onPerfEntry => {
+  if (onPerfEntry && onPerfEntry instanceof Function) {
+    import('web-vitals').then(({ getCLS, getFID, getFCP, getLCP, getTTFB }) => {
+      getCLS(onPerfEntry);
+      getFID(onPerfEntry);
+      getFCP(onPerfEntry);
+      getLCP(onPerfEntry);
+      getTTFB(onPerfEntry);
+    });
+  }
+};
+
+export default reportWebVitals;

frontend/src/setupTests.js

@@ -0,0 +1,5 @@
+// jest-dom adds custom jest matchers for asserting on DOM nodes.
+// allows you to do things like:
+// expect(element).toHaveTextContent(/react/i)
+// learn more: https://github.com/testing-library/jest-dom
+import '@testing-library/jest-dom';

frontend/src/theme.js

@@ -0,0 +1,49 @@
+import { createTheme } from '@mui/material/styles';
+
+const theme = createTheme({
+  palette: {
+    mode: 'dark',
+    primary: {
+      main: '#BB86FC', // Light purple for primary elements
+    },
+    secondary: {
+      main: '#03DAC6', // Teal for secondary elements
+    },
+    background: {
+      default: '#121212', // Dark background
+      paper: '#1E1E1E', // Slightly lighter background for cards
+    },
+    text: {
+      primary: '#FFFFFF', // White text for primary
+      secondary: '#B0BEC5', // Light grey for secondary text
+    },
+  },
+  typography: {
+    fontFamily: '"Roboto", "Helvetica", "Arial", sans-serif',
+    h1: {
+      fontSize: '2rem',
+      fontWeight: 700,
+    },
+    h2: {
+      fontSize: '1.5rem',
+      fontWeight: 600,
+    },
+    body1: {
+      fontSize: '1rem',
+      lineHeight: 1.5,
+    },
+    body2: {
+      fontSize: '0.875rem',
+      lineHeight: 1.4,
+    },
+    caption: {
+      fontSize: '0.75rem',
+      lineHeight: 1.2,
+    },
+  },
+  shape: {
+    borderRadius: 8, // Rounded corners for cards and buttons
+  },
+});
+
+export default theme;