github-actions[bot] commited on
Commit
d00f0c4
β€’
0 Parent(s):

Sync to HuggingFace Spaces

Browse files
This view is limited to 50 files because it contains too many changes. Β  See raw diff
Files changed (50) hide show
  1. .dockerignore +25 -0
  2. .editorconfig +7 -0
  3. .env.example +32 -0
  4. .github/workflows/ai-pr-summarizer/model.txt +1 -0
  5. .github/workflows/ai-pr-summarizer/ollama-version.txt +1 -0
  6. .github/workflows/ai-pr-summarizer/prompt.txt +3 -0
  7. .github/workflows/deploy.yml +54 -0
  8. .github/workflows/on-pull-request-to-main.yml +29 -0
  9. .github/workflows/on-push-to-main.yml +7 -0
  10. .github/workflows/reusable-test-lint-ping.yml +25 -0
  11. .github/workflows/update-searxng-docker-image.yml +44 -0
  12. .gitignore +7 -0
  13. .npmrc +1 -0
  14. Dockerfile +82 -0
  15. README.md +131 -0
  16. biome.json +30 -0
  17. client/components/AiResponse/AiModelDownloadAllowanceContent.tsx +62 -0
  18. client/components/AiResponse/AiResponseContent.tsx +164 -0
  19. client/components/AiResponse/AiResponseSection.tsx +92 -0
  20. client/components/AiResponse/ChatInterface.tsx +195 -0
  21. client/components/AiResponse/CopyIconButton.tsx +32 -0
  22. client/components/AiResponse/FormattedMarkdown.tsx +38 -0
  23. client/components/AiResponse/LoadingModelContent.tsx +40 -0
  24. client/components/AiResponse/PreparingContent.tsx +29 -0
  25. client/components/AiResponse/WebLlmModelSelect.tsx +81 -0
  26. client/components/AiResponse/WllamaModelSelect.tsx +42 -0
  27. client/components/App/App.tsx +97 -0
  28. client/components/Logs/LogsModal.tsx +101 -0
  29. client/components/Logs/ShowLogsButton.tsx +42 -0
  30. client/components/Pages/AccessPage.tsx +61 -0
  31. client/components/Pages/Main/MainPage.tsx +65 -0
  32. client/components/Pages/Main/Menu/AISettingsForm.tsx +368 -0
  33. client/components/Pages/Main/Menu/ActionsForm.tsx +18 -0
  34. client/components/Pages/Main/Menu/ClearDataButton.tsx +59 -0
  35. client/components/Pages/Main/Menu/InterfaceSettingsForm.tsx +45 -0
  36. client/components/Pages/Main/Menu/MenuButton.tsx +53 -0
  37. client/components/Pages/Main/Menu/MenuDrawer.tsx +111 -0
  38. client/components/Pages/Main/Menu/SearchSettingsForm.tsx +43 -0
  39. client/components/Search/Form/SearchForm.tsx +140 -0
  40. client/components/Search/Results/Graphical/ImageResultsList.tsx +120 -0
  41. client/components/Search/Results/SearchResultsSection.tsx +132 -0
  42. client/components/Search/Results/Textual/SearchResultsList.tsx +87 -0
  43. client/index.html +36 -0
  44. client/index.tsx +9 -0
  45. client/modules/accessKey.ts +95 -0
  46. client/modules/logEntries.ts +20 -0
  47. client/modules/openai.ts +16 -0
  48. client/modules/parentWindow.ts +5 -0
  49. client/modules/pubSub.ts +106 -0
  50. client/modules/querySuggestions.ts +31 -0
.dockerignore ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Logs
2
+ logs
3
+ *.log
4
+ npm-debug.log*
5
+ yarn-debug.log*
6
+ yarn-error.log*
7
+ pnpm-debug.log*
8
+ lerna-debug.log*
9
+
10
+ node_modules
11
+ dist
12
+ dist-ssr
13
+ *.local
14
+
15
+ # Editor directories and files
16
+ .vscode/*
17
+ !.vscode/extensions.json
18
+ .idea
19
+ .DS_Store
20
+ *.suo
21
+ *.ntvs*
22
+ *.njsproj
23
+ *.sln
24
+ *.sw?
25
+
.editorconfig ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ [*]
2
+ charset = utf-8
3
+ insert_final_newline = true
4
+ end_of_line = lf
5
+ indent_style = space
6
+ indent_size = 2
7
+ max_line_length = 80
.env.example ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # A comma-separated list of access keys. Example: `ACCESS_KEYS="ABC123,JUD71F,HUWE3"`. Leave blank for unrestricted access.
2
+ ACCESS_KEYS=""
3
+
4
+ # The timeout in hours for access key validation. Set to 0 to require validation on every page load.
5
+ ACCESS_KEY_TIMEOUT_HOURS="24"
6
+
7
+ # The default model ID for WebLLM with F16 shaders.
8
+ WEBLLM_DEFAULT_F16_MODEL_ID="SmolLM2-360M-Instruct-q0f16-MLC"
9
+
10
+ # The default model ID for WebLLM with F32 shaders.
11
+ WEBLLM_DEFAULT_F32_MODEL_ID="SmolLM2-360M-Instruct-q0f32-MLC"
12
+
13
+ # The default model ID for Wllama.
14
+ WLLAMA_DEFAULT_MODEL_ID="smollm2-360m"
15
+
16
+ # The base URL for the internal OpenAI compatible API. Example: `INTERNAL_OPENAI_COMPATIBLE_API_BASE_URL="https://api.openai.com/v1"`. Leave blank to disable internal OpenAI compatible API.
17
+ INTERNAL_OPENAI_COMPATIBLE_API_BASE_URL=""
18
+
19
+ # The access key for the internal OpenAI compatible API.
20
+ INTERNAL_OPENAI_COMPATIBLE_API_KEY=""
21
+
22
+ # The model for the internal OpenAI compatible API.
23
+ INTERNAL_OPENAI_COMPATIBLE_API_MODEL=""
24
+
25
+ # The name of the internal OpenAI compatible API, displayed in the UI.
26
+ INTERNAL_OPENAI_COMPATIBLE_API_NAME="Internal API"
27
+
28
+ # The type of inference to use by default. The possible values are:
29
+ # "browser" -> In the browser (Private)
30
+ # "openai" -> Remote Server (API)
31
+ # "internal" -> $INTERNAL_OPENAI_COMPATIBLE_API_NAME
32
+ DEFAULT_INFERENCE_TYPE="browser"
.github/workflows/ai-pr-summarizer/model.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ qwen2.5-coder:7b-instruct-q8_0
.github/workflows/ai-pr-summarizer/ollama-version.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ 0.3.14
.github/workflows/ai-pr-summarizer/prompt.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ Your task is to write a detailed code review of the following changes, focusing on the "what" and "why" rather than the "how".
2
+
3
+ After that, suggest a commit message summarizing the changes in one sentence and using the present tense.
.github/workflows/deploy.yml ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: Deploy
2
+
3
+ on:
4
+ workflow_dispatch:
5
+
6
+ jobs:
7
+ build-and-push-image:
8
+ name: Publish Docker image to GitHub Packages
9
+ runs-on: ubuntu-latest
10
+ env:
11
+ REGISTRY: ghcr.io
12
+ IMAGE_NAME: ${{ github.repository }}
13
+ permissions:
14
+ contents: read
15
+ packages: write
16
+ steps:
17
+ - name: Checkout repository
18
+ uses: actions/checkout@v4
19
+ - name: Log in to the Container registry
20
+ uses: docker/login-action@v3
21
+ with:
22
+ registry: ${{ env.REGISTRY }}
23
+ username: ${{ github.actor }}
24
+ password: ${{ secrets.GITHUB_TOKEN }}
25
+ - name: Extract metadata (tags, labels) for Docker
26
+ id: meta
27
+ uses: docker/metadata-action@v5
28
+ with:
29
+ images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
30
+ - name: Set up Docker Buildx
31
+ uses: docker/setup-buildx-action@v3
32
+ - name: Build and push Docker image
33
+ uses: docker/build-push-action@v6
34
+ with:
35
+ context: .
36
+ push: true
37
+ tags: ${{ steps.meta.outputs.tags }}
38
+ labels: ${{ steps.meta.outputs.labels }}
39
+ platforms: linux/amd64,linux/arm64
40
+
41
+ sync-to-hf:
42
+ name: Sync to HuggingFace Spaces
43
+ runs-on: ubuntu-latest
44
+ steps:
45
+ - uses: actions/checkout@v4
46
+ with:
47
+ lfs: true
48
+ - uses: JacobLinCool/huggingface-sync@v1
49
+ with:
50
+ github: ${{ secrets.GITHUB_TOKEN }}
51
+ user: ${{ vars.HF_SPACE_OWNER }}
52
+ space: ${{ vars.HF_SPACE_NAME }}
53
+ token: ${{ secrets.HF_TOKEN }}
54
+ configuration: "hf-space-config.yml"
.github/workflows/on-pull-request-to-main.yml ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: On Pull Request To Main
2
+ on:
3
+ pull_request:
4
+ types: [opened, synchronize, reopened]
5
+ branches: ["main"]
6
+ jobs:
7
+ test-lint-ping:
8
+ uses: ./.github/workflows/reusable-test-lint-ping.yml
9
+ ai-pr-summarizer:
10
+ needs: [test-lint-ping]
11
+ runs-on: ubuntu-latest
12
+ name: AI PR Summarizer
13
+ permissions:
14
+ pull-requests: write
15
+ contents: read
16
+ steps:
17
+ - name: Checkout Repository
18
+ uses: actions/checkout@v4
19
+ - name: Summarize PR with AI
20
+ uses: behrouz-rad/ai-pr-summarizer@v1
21
+ with:
22
+ llm-model: 'qwen2.5-coder:7b-instruct-q8_0'
23
+ prompt-file: ./.github/workflows/ai-pr-summarizer/prompt.txt
24
+ models-file: ./.github/workflows/ai-pr-summarizer/model.txt
25
+ version-file: ./.github/workflows/ai-pr-summarizer/ollama-version.txt
26
+ context-window: 16384
27
+ upload-changes: false
28
+ fail-on-error: false
29
+ token: ${{ secrets.GITHUB_TOKEN }}
.github/workflows/on-push-to-main.yml ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ name: On Push To Main
2
+ on:
3
+ push:
4
+ branches: ["main"]
5
+ jobs:
6
+ test-lint-ping:
7
+ uses: ./.github/workflows/reusable-test-lint-ping.yml
.github/workflows/reusable-test-lint-ping.yml ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ on:
2
+ workflow_call:
3
+ jobs:
4
+ check-code-quality:
5
+ name: Check Code Quality
6
+ runs-on: ubuntu-latest
7
+ steps:
8
+ - uses: actions/checkout@v4
9
+ - uses: actions/setup-node@v4
10
+ with:
11
+ node-version: 20
12
+ cache: "npm"
13
+ - run: npm ci --ignore-scripts
14
+ - run: npm test
15
+ - run: npm run lint
16
+ check-docker-container:
17
+ needs: [check-code-quality]
18
+ name: Check Docker Container
19
+ runs-on: ubuntu-latest
20
+ steps:
21
+ - uses: actions/checkout@v4
22
+ - run: docker compose -f docker-compose.production.yml up -d
23
+ - name: Check if main page is available
24
+ run: until curl -s -o /dev/null -w "%{http_code}" localhost:7860 | grep 200; do sleep 1; done
25
+ - run: docker compose -f docker-compose.production.yml down
.github/workflows/update-searxng-docker-image.yml ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: Update SearXNG Docker Image
2
+
3
+ on:
4
+ schedule:
5
+ - cron: "0 14 * * *"
6
+ workflow_dispatch:
7
+
8
+ permissions:
9
+ contents: write
10
+
11
+ jobs:
12
+ update-searxng-image:
13
+ runs-on: ubuntu-latest
14
+ steps:
15
+ - name: Checkout code
16
+ uses: actions/checkout@v4
17
+ with:
18
+ token: ${{ secrets.GITHUB_TOKEN }}
19
+
20
+ - name: Get latest SearXNG image tag
21
+ id: get_latest_tag
22
+ run: |
23
+ LATEST_TAG=$(curl -s "https://hub.docker.com/v2/repositories/searxng/searxng/tags/?page_size=3&ordering=last_updated" | jq -r '.results[] | select(.name != "latest-build-cache" and .name != "latest") | .name' | head -n 1)
24
+ echo "LATEST_TAG=${LATEST_TAG}" >> $GITHUB_OUTPUT
25
+
26
+ - name: Update Dockerfile
27
+ run: |
28
+ sed -i 's|FROM searxng/searxng:.*|FROM searxng/searxng:${{ steps.get_latest_tag.outputs.LATEST_TAG }}|' Dockerfile
29
+
30
+ - name: Check for changes
31
+ id: git_status
32
+ run: |
33
+ git diff --exit-code || echo "changes=true" >> $GITHUB_OUTPUT
34
+
35
+ - name: Commit and push if changed
36
+ if: steps.git_status.outputs.changes == 'true'
37
+ run: |
38
+ git config --local user.email "github-actions[bot]@users.noreply.github.com"
39
+ git config --local user.name "github-actions[bot]"
40
+ git add Dockerfile
41
+ git commit -m "Update SearXNG Docker image to tag ${{ steps.get_latest_tag.outputs.LATEST_TAG }}"
42
+ git push
43
+ env:
44
+ GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
.gitignore ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ node_modules
2
+ .DS_Store
3
+ /client/dist
4
+ /server/models
5
+ .vscode
6
+ /vite-build-stats.html
7
+ .env
.npmrc ADDED
@@ -0,0 +1 @@
 
 
1
+ legacy-peer-deps = true
Dockerfile ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Use the SearXNG image as the base
2
+ FROM searxng/searxng:2024.11.6-cd384a8a6
3
+
4
+ # Set the default port to 7860 if not provided
5
+ ENV PORT=7860
6
+
7
+ # Expose the port specified by the PORT environment variable
8
+ EXPOSE $PORT
9
+
10
+ # Install necessary packages using Alpine's package manager
11
+ RUN apk add --update \
12
+ nodejs \
13
+ npm \
14
+ git \
15
+ build-base \
16
+ cmake \
17
+ ccache
18
+
19
+ # Set the SearXNG settings folder path
20
+ ARG SEARXNG_SETTINGS_FOLDER=/etc/searxng
21
+
22
+ # Modify SearXNG configuration:
23
+ # 1. Change output format from HTML to JSON
24
+ # 2. Remove user switching in the entrypoint script
25
+ # 3. Create and set permissions for the settings folder
26
+ RUN sed -i 's/- html/- json/' /usr/local/searxng/searx/settings.yml \
27
+ && sed -i 's/su-exec searxng:searxng //' /usr/local/searxng/dockerfiles/docker-entrypoint.sh \
28
+ && mkdir -p ${SEARXNG_SETTINGS_FOLDER} \
29
+ && chmod 777 ${SEARXNG_SETTINGS_FOLDER}
30
+
31
+ # Set up user and directory structure
32
+ ARG USERNAME=user
33
+ ARG HOME_DIR=/home/${USERNAME}
34
+ ARG APP_DIR=${HOME_DIR}/app
35
+
36
+ # Create a non-root user and set up the application directory
37
+ RUN adduser -D -u 1000 ${USERNAME} \
38
+ && mkdir -p ${APP_DIR} \
39
+ && chown -R ${USERNAME}:${USERNAME} ${HOME_DIR}
40
+
41
+ # Switch to the non-root user
42
+ USER ${USERNAME}
43
+
44
+ # Set the working directory to the application directory
45
+ WORKDIR ${APP_DIR}
46
+
47
+ # Define environment variables that can be passed to the container during build.
48
+ # This approach allows for dynamic configuration without relying on a `.env` file,
49
+ # which might not be suitable for all deployment scenarios.
50
+ ARG ACCESS_KEYS
51
+ ARG ACCESS_KEY_TIMEOUT_HOURS
52
+ ARG WEBLLM_DEFAULT_F16_MODEL_ID
53
+ ARG WEBLLM_DEFAULT_F32_MODEL_ID
54
+ ARG WLLAMA_DEFAULT_MODEL_ID
55
+ ARG INTERNAL_OPENAI_COMPATIBLE_API_BASE_URL
56
+ ARG INTERNAL_OPENAI_COMPATIBLE_API_KEY
57
+ ARG INTERNAL_OPENAI_COMPATIBLE_API_MODEL
58
+ ARG INTERNAL_OPENAI_COMPATIBLE_API_NAME
59
+ ARG DEFAULT_INFERENCE_TYPE
60
+
61
+ # Copy package.json, package-lock.json, and .npmrc files
62
+ COPY --chown=${USERNAME}:${USERNAME} ./package.json ./package.json
63
+ COPY --chown=${USERNAME}:${USERNAME} ./package-lock.json ./package-lock.json
64
+ COPY --chown=${USERNAME}:${USERNAME} ./.npmrc ./.npmrc
65
+
66
+ # Install Node.js dependencies
67
+ RUN npm ci
68
+
69
+ # Copy the rest of the application files
70
+ COPY --chown=${USERNAME}:${USERNAME} . .
71
+
72
+ # Configure Git to treat the app directory as safe
73
+ RUN git config --global --add safe.directory ${APP_DIR}
74
+
75
+ # Build the application
76
+ RUN npm run build
77
+
78
+ # Set the entrypoint to use a shell
79
+ ENTRYPOINT [ "/bin/sh", "-c" ]
80
+
81
+ # Run SearXNG in the background and start the Node.js application using PM2
82
+ CMD [ "(/usr/local/searxng/dockerfiles/docker-entrypoint.sh -f > /dev/null 2>&1) & (npx pm2 start ecosystem.config.cjs && npx pm2 logs production-server)" ]
README.md ADDED
@@ -0,0 +1,131 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ title: MiniSearch
3
+ emoji: πŸ‘ŒπŸ”
4
+ colorFrom: yellow
5
+ colorTo: yellow
6
+ sdk: docker
7
+ short_description: Minimalist web-searching app with browser-based AI assistant
8
+ pinned: true
9
+ custom_headers:
10
+ cross-origin-embedder-policy: require-corp
11
+ cross-origin-opener-policy: same-origin
12
+ cross-origin-resource-policy: cross-origin
13
+ ---
14
+
15
+ # MiniSearch
16
+
17
+ A minimalist web-searching app with an AI assistant that runs directly from your browser.
18
+
19
+ Live demo: https://felladrin-minisearch.hf.space
20
+
21
+ ## Screenshot
22
+
23
+ ![MiniSearch Screenshot](https://github.com/user-attachments/assets/f8d72a8e-a725-42e9-9358-e6ebade2acb2)
24
+
25
+ ## Features
26
+
27
+ - **Privacy-focused**: [No tracking, no ads, no data collection](https://docs.searxng.org/own-instance.html#how-does-searxng-protect-privacy)
28
+ - **Easy to use**: Minimalist yet intuitive interface for all users
29
+ - **Cross-platform**: Models run inside the browser, both on desktop and mobile
30
+ - **Integrated**: Search from the browser address bar by setting it as the default search engine
31
+ - **Efficient**: Models are loaded and cached only when needed
32
+ - **Customizable**: Tweakable settings for search results and text generation
33
+ - **Open-source**: [The code is available for inspection and contribution at GitHub](https://github.com/felladrin/MiniSearch)
34
+
35
+ ## Prerequisites
36
+
37
+ - [Docker](https://docs.docker.com/get-docker/)
38
+
39
+ ## Getting started
40
+
41
+ Here are the easiest ways to get started with MiniSearch. Pick the one that suits you best.
42
+
43
+ **Option 1** - Use [MiniSearch's Docker Image](https://github.com/felladrin/MiniSearch/pkgs/container/minisearch) by running in your terminal:
44
+
45
+ ```bash
46
+ docker run -p 7860:7860 ghcr.io/felladrin/minisearch:main
47
+ ```
48
+
49
+ **Option 2** - Add MiniSearch's Docker Image to your existing Docker Compose file:
50
+
51
+ ```yaml
52
+ services:
53
+ minisearch:
54
+ image: ghcr.io/felladrin/minisearch:main
55
+ ports:
56
+ - "7860:7860"
57
+ ```
58
+
59
+ **Option 3** - Build from source by [downloading the repository files](https://github.com/felladrin/MiniSearch/archive/refs/heads/main.zip) and running:
60
+
61
+ ```bash
62
+ docker compose -f docker-compose.production.yml up --build
63
+ ```
64
+
65
+ Once the container is running, open http://localhost:7860 in your browser and start searching!
66
+
67
+ ## Frequently asked questions
68
+
69
+ <details>
70
+ <summary>How do I search via the browser's address bar?</summary>
71
+ <p>
72
+ You can set MiniSearch as your browser's address-bar search engine using the pattern <code>http://localhost:7860/?q=%s</code>, in which your search term replaces <code>%s</code>.
73
+ </p>
74
+ </details>
75
+
76
+ <details>
77
+ <summary>Can I use custom models via OpenAI-Compatible API?</summary>
78
+ <p>
79
+ Yes! For this, open the Menu and change the "AI Processing Location" to <code>Remote server (API)</code>. Then configure the Base URL, and optionally set an API Key and a Model to use.
80
+ </p>
81
+ </details>
82
+
83
+ <details>
84
+ <summary>How do I restrict the access to my MiniSearch instance via password?</summary>
85
+ <p>
86
+ Create a <code>.env</code> file and set a value for <code>ACCESS_KEYS</code>. Then reset the MiniSearch docker container.
87
+ </p>
88
+ <p>
89
+ For example, if you to set the password to <code>PepperoniPizza</code>, then this is what you should add to your <code>.env</code>:<br/>
90
+ <code>ACCESS_KEYS="PepperoniPizza"</code>
91
+ </p>
92
+ <p>
93
+ You can find more examples in the <code>.env.example</code> file.
94
+ </p>
95
+ </details>
96
+
97
+ <details>
98
+ <summary>I want to serve MiniSearch to other users, allowing them to use my own OpenAI-Compatible API key, but without revealing it to them. Is it possible?</summary>
99
+ <p>Yes! In MiniSearch, we call this text-generation feature "Internal OpenAI-Compatible API". To use this it:</p>
100
+ <ol>
101
+ <li>Set up your OpenAI-Compatible API endpoint by configuring the following environment variables in your <code>.env</code> file:
102
+ <ul>
103
+ <li><code>INTERNAL_OPENAI_COMPATIBLE_API_BASE_URL</code>: The base URL for your API</li>
104
+ <li><code>INTERNAL_OPENAI_COMPATIBLE_API_KEY</code>: Your API access key</li>
105
+ <li><code>INTERNAL_OPENAI_COMPATIBLE_API_MODEL</code>: The model to use</li>
106
+ <li><code>INTERNAL_OPENAI_COMPATIBLE_API_NAME</code>: The name to display in the UI</li>
107
+ </ul>
108
+ </li>
109
+ <li>Restart MiniSearch server.</li>
110
+ <li>In the MiniSearch menu, select the new option (named as per your <code>INTERNAL_OPENAI_COMPATIBLE_API_NAME</code> setting) from the "AI Processing Location" dropdown.</li>
111
+ </ol>
112
+ </details>
113
+
114
+ <details>
115
+ <summary>How can I contribute to the development of this tool?</summary>
116
+ <p>Fork this repository and clone it. Then, start the development server by running the following command:</p>
117
+ <p><code>docker compose up</code></p>
118
+ <p>Make your changes, push them to your fork, and open a pull request! All contributions are welcome!</p>
119
+ </details>
120
+
121
+ <details>
122
+ <summary>Why is MiniSearch built upon SearXNG's Docker Image and using a single image instead of composing it from multiple services?</summary>
123
+ <p>There are a few reasons for this:</p>
124
+ <ul>
125
+ <li>MiniSearch utilizes SearXNG as its meta-search engine.</li>
126
+ <li>Manual installation of SearXNG is not trivial, so we use the docker image they provide, which has everything set up.</li>
127
+ <li>SearXNG only provides a Docker Image based on Alpine Linux.</li>
128
+ <li>The user of the image needs to be customized in a specific way to run on HuggingFace Spaces, where MiniSearch's demo runs.</li>
129
+ <li>HuggingFace only accepts a single docker image. It doesn't run docker compose or multiple images, unfortunately.</li>
130
+ </ul>
131
+ </details>
biome.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "$schema": "https://biomejs.dev/schemas/1.9.4/schema.json",
3
+ "vcs": {
4
+ "enabled": false,
5
+ "clientKind": "git",
6
+ "useIgnoreFile": false
7
+ },
8
+ "files": {
9
+ "ignoreUnknown": false,
10
+ "ignore": []
11
+ },
12
+ "formatter": {
13
+ "enabled": true,
14
+ "indentStyle": "space"
15
+ },
16
+ "organizeImports": {
17
+ "enabled": true
18
+ },
19
+ "linter": {
20
+ "enabled": true,
21
+ "rules": {
22
+ "recommended": true
23
+ }
24
+ },
25
+ "javascript": {
26
+ "formatter": {
27
+ "quoteStyle": "double"
28
+ }
29
+ }
30
+ }
client/components/AiResponse/AiModelDownloadAllowanceContent.tsx ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { Alert, Button, Group, Text } from "@mantine/core";
2
+ import { IconCheck, IconInfoCircle, IconX } from "@tabler/icons-react";
3
+ import { usePubSub } from "create-pubsub/react";
4
+ import { useState } from "react";
5
+ import { addLogEntry } from "../../modules/logEntries";
6
+ import { settingsPubSub } from "../../modules/pubSub";
7
+
8
+ export default function AiModelDownloadAllowanceContent() {
9
+ const [settings, setSettings] = usePubSub(settingsPubSub);
10
+ const [hasDeniedDownload, setDeniedDownload] = useState(false);
11
+
12
+ const handleAccept = () => {
13
+ setSettings({
14
+ ...settings,
15
+ allowAiModelDownload: true,
16
+ });
17
+ addLogEntry("User allowed the AI model download");
18
+ };
19
+
20
+ const handleDecline = () => {
21
+ setDeniedDownload(true);
22
+ addLogEntry("User denied the AI model download");
23
+ };
24
+
25
+ return hasDeniedDownload ? null : (
26
+ <Alert
27
+ variant="light"
28
+ color="blue"
29
+ title="Allow AI model download?"
30
+ icon={<IconInfoCircle />}
31
+ >
32
+ <Text size="sm" mb="md">
33
+ To obtain AI responses, a language model needs to be downloaded to your
34
+ browser. Enabling this option lets the app store it and load it
35
+ instantly on subsequent uses.
36
+ </Text>
37
+ <Text size="sm" mb="md">
38
+ Please note that the download size ranges from 100 MB to 4 GB, depending
39
+ on the model you select in the Menu, so it's best to avoid using mobile
40
+ data for this.
41
+ </Text>
42
+ <Group justify="flex-end" mt="md">
43
+ <Button
44
+ variant="subtle"
45
+ color="gray"
46
+ leftSection={<IconX size="1rem" />}
47
+ onClick={handleDecline}
48
+ size="xs"
49
+ >
50
+ Not now
51
+ </Button>
52
+ <Button
53
+ leftSection={<IconCheck size="1rem" />}
54
+ onClick={handleAccept}
55
+ size="xs"
56
+ >
57
+ Allow download
58
+ </Button>
59
+ </Group>
60
+ </Alert>
61
+ );
62
+ }
client/components/AiResponse/AiResponseContent.tsx ADDED
@@ -0,0 +1,164 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import {
2
+ ActionIcon,
3
+ Alert,
4
+ Badge,
5
+ Box,
6
+ Card,
7
+ Group,
8
+ ScrollArea,
9
+ Text,
10
+ Tooltip,
11
+ } from "@mantine/core";
12
+ import {
13
+ IconArrowsMaximize,
14
+ IconArrowsMinimize,
15
+ IconHandStop,
16
+ IconInfoCircle,
17
+ IconRefresh,
18
+ } from "@tabler/icons-react";
19
+ import type { PublishFunction } from "create-pubsub";
20
+ import { usePubSub } from "create-pubsub/react";
21
+ import { type ReactNode, Suspense, lazy, useMemo } from "react";
22
+ import { match } from "ts-pattern";
23
+ import { settingsPubSub } from "../../modules/pubSub";
24
+ import { searchAndRespond } from "../../modules/textGeneration";
25
+
26
+ const FormattedMarkdown = lazy(() => import("./FormattedMarkdown"));
27
+ const CopyIconButton = lazy(() => import("./CopyIconButton"));
28
+
29
+ export default function AiResponseContent({
30
+ textGenerationState,
31
+ response,
32
+ setTextGenerationState,
33
+ }: {
34
+ textGenerationState: string;
35
+ response: string;
36
+ setTextGenerationState: PublishFunction<
37
+ | "failed"
38
+ | "awaitingSearchResults"
39
+ | "preparingToGenerate"
40
+ | "idle"
41
+ | "loadingModel"
42
+ | "generating"
43
+ | "interrupted"
44
+ | "completed"
45
+ >;
46
+ }) {
47
+ const [settings, setSettings] = usePubSub(settingsPubSub);
48
+
49
+ const ConditionalScrollArea = useMemo(
50
+ () =>
51
+ ({ children }: { children: ReactNode }) => {
52
+ return settings.enableAiResponseScrolling ? (
53
+ <ScrollArea.Autosize mah={300} type="auto" offsetScrollbars>
54
+ {children}
55
+ </ScrollArea.Autosize>
56
+ ) : (
57
+ <Box>{children}</Box>
58
+ );
59
+ },
60
+ [settings.enableAiResponseScrolling],
61
+ );
62
+
63
+ return (
64
+ <Card withBorder shadow="sm" radius="md">
65
+ <Card.Section withBorder inheritPadding py="xs">
66
+ <Group justify="space-between">
67
+ <Group gap="xs" align="center">
68
+ <Text fw={500}>
69
+ {match(textGenerationState)
70
+ .with("generating", () => "Generating AI Response...")
71
+ .otherwise(() => "AI Response")}
72
+ </Text>
73
+ {match(textGenerationState)
74
+ .with("interrupted", () => (
75
+ <Badge variant="light" color="yellow" size="xs">
76
+ Interrupted
77
+ </Badge>
78
+ ))
79
+ .otherwise(() => null)}
80
+ </Group>
81
+ <Group gap="xs" align="center">
82
+ {match(textGenerationState)
83
+ .with("generating", () => (
84
+ <Tooltip label="Interrupt generation">
85
+ <ActionIcon
86
+ onClick={() => setTextGenerationState("interrupted")}
87
+ variant="subtle"
88
+ color="gray"
89
+ >
90
+ <IconHandStop size={16} />
91
+ </ActionIcon>
92
+ </Tooltip>
93
+ ))
94
+ .otherwise(() => (
95
+ <Tooltip label="Regenerate response">
96
+ <ActionIcon
97
+ onClick={() => searchAndRespond()}
98
+ variant="subtle"
99
+ color="gray"
100
+ >
101
+ <IconRefresh size={16} />
102
+ </ActionIcon>
103
+ </Tooltip>
104
+ ))}
105
+ {settings.enableAiResponseScrolling ? (
106
+ <Tooltip label="Show full response without scroll bar">
107
+ <ActionIcon
108
+ onClick={() => {
109
+ setSettings({
110
+ ...settings,
111
+ enableAiResponseScrolling: false,
112
+ });
113
+ }}
114
+ variant="subtle"
115
+ color="gray"
116
+ >
117
+ <IconArrowsMaximize size={16} />
118
+ </ActionIcon>
119
+ </Tooltip>
120
+ ) : (
121
+ <Tooltip label="Enable scroll bar">
122
+ <ActionIcon
123
+ onClick={() => {
124
+ setSettings({
125
+ ...settings,
126
+ enableAiResponseScrolling: true,
127
+ });
128
+ }}
129
+ variant="subtle"
130
+ color="gray"
131
+ >
132
+ <IconArrowsMinimize size={16} />
133
+ </ActionIcon>
134
+ </Tooltip>
135
+ )}
136
+ <Suspense>
137
+ <CopyIconButton value={response} tooltipLabel="Copy response" />
138
+ </Suspense>
139
+ </Group>
140
+ </Group>
141
+ </Card.Section>
142
+ <Card.Section withBorder>
143
+ <ConditionalScrollArea>
144
+ <Suspense>
145
+ <FormattedMarkdown>{response}</FormattedMarkdown>
146
+ </Suspense>
147
+ </ConditionalScrollArea>
148
+ {match(textGenerationState)
149
+ .with("failed", () => (
150
+ <Alert
151
+ variant="light"
152
+ color="yellow"
153
+ title="Failed to generate response"
154
+ icon={<IconInfoCircle />}
155
+ >
156
+ Could not generate response. It's possible that your browser or
157
+ your system is out of memory.
158
+ </Alert>
159
+ ))
160
+ .otherwise(() => null)}
161
+ </Card.Section>
162
+ </Card>
163
+ );
164
+ }
client/components/AiResponse/AiResponseSection.tsx ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { usePubSub } from "create-pubsub/react";
2
+ import { Suspense, lazy, useMemo } from "react";
3
+ import { Pattern, match } from "ts-pattern";
4
+ import {
5
+ modelLoadingProgressPubSub,
6
+ modelSizeInMegabytesPubSub,
7
+ queryPubSub,
8
+ responsePubSub,
9
+ settingsPubSub,
10
+ textGenerationStatePubSub,
11
+ } from "../../modules/pubSub";
12
+
13
+ const AiResponseContent = lazy(() => import("./AiResponseContent"));
14
+ const PreparingContent = lazy(() => import("./PreparingContent"));
15
+ const LoadingModelContent = lazy(() => import("./LoadingModelContent"));
16
+ const ChatInterface = lazy(() => import("./ChatInterface"));
17
+ const AiModelDownloadAllowanceContent = lazy(
18
+ () => import("./AiModelDownloadAllowanceContent"),
19
+ );
20
+
21
+ export default function AiResponseSection() {
22
+ const [query] = usePubSub(queryPubSub);
23
+ const [response] = usePubSub(responsePubSub);
24
+ const [textGenerationState, setTextGenerationState] = usePubSub(
25
+ textGenerationStatePubSub,
26
+ );
27
+ const [modelLoadingProgress] = usePubSub(modelLoadingProgressPubSub);
28
+ const [settings] = usePubSub(settingsPubSub);
29
+ const [modelSizeInMegabytes] = usePubSub(modelSizeInMegabytesPubSub);
30
+
31
+ return useMemo(
32
+ () =>
33
+ match([settings.enableAiResponse, textGenerationState])
34
+ .with([true, Pattern.not("idle").select()], (textGenerationState) =>
35
+ match(textGenerationState)
36
+ .with(
37
+ Pattern.union("generating", "interrupted", "completed", "failed"),
38
+ (textGenerationState) => (
39
+ <>
40
+ <Suspense>
41
+ <AiResponseContent
42
+ textGenerationState={textGenerationState}
43
+ response={response}
44
+ setTextGenerationState={setTextGenerationState}
45
+ />
46
+ </Suspense>
47
+ {textGenerationState === "completed" && (
48
+ <Suspense>
49
+ <ChatInterface
50
+ initialQuery={query}
51
+ initialResponse={response}
52
+ />
53
+ </Suspense>
54
+ )}
55
+ </>
56
+ ),
57
+ )
58
+ .with("awaitingModelDownloadAllowance", () => (
59
+ <Suspense>
60
+ <AiModelDownloadAllowanceContent />
61
+ </Suspense>
62
+ ))
63
+ .with("loadingModel", () => (
64
+ <Suspense>
65
+ <LoadingModelContent
66
+ modelLoadingProgress={modelLoadingProgress}
67
+ modelSizeInMegabytes={modelSizeInMegabytes}
68
+ />
69
+ </Suspense>
70
+ ))
71
+ .with(
72
+ Pattern.union("awaitingSearchResults", "preparingToGenerate"),
73
+ (textGenerationState) => (
74
+ <Suspense>
75
+ <PreparingContent textGenerationState={textGenerationState} />
76
+ </Suspense>
77
+ ),
78
+ )
79
+ .exhaustive(),
80
+ )
81
+ .otherwise(() => null),
82
+ [
83
+ settings,
84
+ textGenerationState,
85
+ setTextGenerationState,
86
+ modelLoadingProgress,
87
+ response,
88
+ query,
89
+ modelSizeInMegabytes,
90
+ ],
91
+ );
92
+ }
client/components/AiResponse/ChatInterface.tsx ADDED
@@ -0,0 +1,195 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import {
2
+ Button,
3
+ Card,
4
+ Group,
5
+ Paper,
6
+ Stack,
7
+ Text,
8
+ Textarea,
9
+ } from "@mantine/core";
10
+ import { IconSend } from "@tabler/icons-react";
11
+ import { usePubSub } from "create-pubsub/react";
12
+ import type { ChatMessage } from "gpt-tokenizer/GptEncoding";
13
+ import {
14
+ type KeyboardEvent,
15
+ Suspense,
16
+ lazy,
17
+ useEffect,
18
+ useRef,
19
+ useState,
20
+ } from "react";
21
+ import { match } from "ts-pattern";
22
+ import { addLogEntry } from "../../modules/logEntries";
23
+ import { settingsPubSub } from "../../modules/pubSub";
24
+ import { generateChatResponse } from "../../modules/textGeneration";
25
+
26
+ const FormattedMarkdown = lazy(() => import("./FormattedMarkdown"));
27
+ const CopyIconButton = lazy(() => import("./CopyIconButton"));
28
+
29
+ export default function ChatInterface({
30
+ initialQuery,
31
+ initialResponse,
32
+ }: {
33
+ initialQuery: string;
34
+ initialResponse: string;
35
+ }) {
36
+ const [messages, setMessages] = useState<ChatMessage[]>([]);
37
+ const [input, setInput] = useState("");
38
+ const [isGenerating, setIsGenerating] = useState(false);
39
+ const [streamedResponse, setStreamedResponse] = useState("");
40
+ const latestResponseRef = useRef("");
41
+ const [settings] = usePubSub(settingsPubSub);
42
+
43
+ useEffect(() => {
44
+ setMessages([
45
+ { role: "user", content: initialQuery },
46
+ { role: "assistant", content: initialResponse },
47
+ ]);
48
+ }, [initialQuery, initialResponse]);
49
+
50
+ const handleSend = async () => {
51
+ if (input.trim() === "" || isGenerating) return;
52
+
53
+ const newMessages: ChatMessage[] = [
54
+ ...messages,
55
+ { role: "user", content: input },
56
+ ];
57
+ setMessages(newMessages);
58
+ setInput("");
59
+ setIsGenerating(true);
60
+ setStreamedResponse("");
61
+ latestResponseRef.current = "";
62
+
63
+ try {
64
+ addLogEntry("User sent a follow-up question");
65
+ await generateChatResponse(newMessages, (partialResponse) => {
66
+ setStreamedResponse(partialResponse);
67
+ latestResponseRef.current = partialResponse;
68
+ });
69
+ setMessages((prevMessages) => [
70
+ ...prevMessages,
71
+ { role: "assistant", content: latestResponseRef.current },
72
+ ]);
73
+ addLogEntry("AI responded to follow-up question");
74
+ } catch (error) {
75
+ addLogEntry(`Error generating chat response: ${error}`);
76
+ setMessages((prevMessages) => [
77
+ ...prevMessages,
78
+ {
79
+ role: "assistant",
80
+ content: "Sorry, I encountered an error while generating a response.",
81
+ },
82
+ ]);
83
+ } finally {
84
+ setIsGenerating(false);
85
+ setStreamedResponse("");
86
+ }
87
+ };
88
+
89
+ const handleKeyDown = (event: KeyboardEvent<HTMLTextAreaElement>) => {
90
+ match([event, settings.enterToSubmit])
91
+ .with([{ code: "Enter", shiftKey: false }, true], () => {
92
+ event.preventDefault();
93
+ handleSend();
94
+ })
95
+ .with([{ code: "Enter", shiftKey: true }, false], () => {
96
+ event.preventDefault();
97
+ handleSend();
98
+ })
99
+ .otherwise(() => undefined);
100
+ };
101
+
102
+ const getChatContent = () => {
103
+ return messages
104
+ .slice(2)
105
+ .map(
106
+ (msg, index) =>
107
+ `${index + 1}. ${msg.role?.toUpperCase()}\n\n${msg.content}`,
108
+ )
109
+ .join("\n\n");
110
+ };
111
+
112
+ return (
113
+ <Card withBorder shadow="sm" radius="md">
114
+ <Card.Section withBorder inheritPadding py="xs">
115
+ <Group justify="space-between">
116
+ <Text fw={500}>Follow-up questions</Text>
117
+ {messages.length > 2 && (
118
+ <Suspense>
119
+ <CopyIconButton
120
+ value={getChatContent()}
121
+ tooltipLabel="Copy conversation"
122
+ />
123
+ </Suspense>
124
+ )}
125
+ </Group>
126
+ </Card.Section>
127
+ <Stack gap="md" pt="md">
128
+ {messages.slice(2).length > 0 && (
129
+ <Stack gap="md">
130
+ {messages.slice(2).map((message, index) => (
131
+ <Paper
132
+ key={`${message.role}-${index}`}
133
+ shadow="xs"
134
+ radius="xl"
135
+ p="sm"
136
+ maw="90%"
137
+ style={{
138
+ alignSelf:
139
+ message.role === "user" ? "flex-end" : "flex-start",
140
+ }}
141
+ >
142
+ <Suspense>
143
+ <FormattedMarkdown>{message.content}</FormattedMarkdown>
144
+ </Suspense>
145
+ </Paper>
146
+ ))}
147
+ {isGenerating && streamedResponse.length > 0 && (
148
+ <Paper
149
+ shadow="xs"
150
+ radius="xl"
151
+ p="sm"
152
+ maw="90%"
153
+ style={{ alignSelf: "flex-start" }}
154
+ >
155
+ <Suspense>
156
+ <FormattedMarkdown>{streamedResponse}</FormattedMarkdown>
157
+ </Suspense>
158
+ </Paper>
159
+ )}
160
+ </Stack>
161
+ )}
162
+ <Group align="flex-end" style={{ position: "relative" }}>
163
+ <Textarea
164
+ placeholder="Anything else you would like to know?"
165
+ value={input}
166
+ onChange={(event) => setInput(event.currentTarget.value)}
167
+ onKeyDown={handleKeyDown}
168
+ autosize
169
+ minRows={1}
170
+ maxRows={4}
171
+ style={{ flexGrow: 1, paddingRight: "50px" }}
172
+ disabled={isGenerating}
173
+ />
174
+ <Button
175
+ size="sm"
176
+ variant="default"
177
+ onClick={handleSend}
178
+ loading={isGenerating}
179
+ style={{
180
+ height: "100%",
181
+ position: "absolute",
182
+ right: 0,
183
+ top: 0,
184
+ bottom: 0,
185
+ borderTopLeftRadius: 0,
186
+ borderBottomLeftRadius: 0,
187
+ }}
188
+ >
189
+ <IconSend size={16} />
190
+ </Button>
191
+ </Group>
192
+ </Stack>
193
+ </Card>
194
+ );
195
+ }
client/components/AiResponse/CopyIconButton.tsx ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { ActionIcon, CopyButton, Tooltip } from "@mantine/core";
2
+ import { IconCheck, IconCopy } from "@tabler/icons-react";
3
+
4
+ interface CopyIconButtonProps {
5
+ value: string;
6
+ tooltipLabel?: string;
7
+ }
8
+
9
+ export default function CopyIconButton({
10
+ value,
11
+ tooltipLabel = "Copy",
12
+ }: CopyIconButtonProps) {
13
+ return (
14
+ <CopyButton value={value} timeout={2000}>
15
+ {({ copied, copy }) => (
16
+ <Tooltip
17
+ label={copied ? "Copied" : tooltipLabel}
18
+ withArrow
19
+ position="right"
20
+ >
21
+ <ActionIcon
22
+ color={copied ? "teal" : "gray"}
23
+ variant="subtle"
24
+ onClick={copy}
25
+ >
26
+ {copied ? <IconCheck size={16} /> : <IconCopy size={16} />}
27
+ </ActionIcon>
28
+ </Tooltip>
29
+ )}
30
+ </CopyButton>
31
+ );
32
+ }
client/components/AiResponse/FormattedMarkdown.tsx ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { TypographyStylesProvider } from "@mantine/core";
2
+ import Markdown from "react-markdown";
3
+ import { Prism as SyntaxHighlighter } from "react-syntax-highlighter";
4
+ import syntaxHighlighterStyle from "react-syntax-highlighter/dist/esm/styles/prism/one-dark";
5
+
6
+ const FormattedMarkdown = ({ children }: { children: string }) => {
7
+ return (
8
+ <TypographyStylesProvider p="md">
9
+ <Markdown
10
+ components={{
11
+ code(props) {
12
+ const { children, className, node, ref, ...rest } = props;
13
+ void node;
14
+ const languageMatch = /language-(\w+)/.exec(className || "");
15
+ return languageMatch ? (
16
+ <SyntaxHighlighter
17
+ {...rest}
18
+ ref={ref as never}
19
+ language={languageMatch[1]}
20
+ style={syntaxHighlighterStyle}
21
+ >
22
+ {children?.toString().replace(/\n$/, "") ?? ""}
23
+ </SyntaxHighlighter>
24
+ ) : (
25
+ <code {...rest} className={className}>
26
+ {children}
27
+ </code>
28
+ );
29
+ },
30
+ }}
31
+ >
32
+ {children}
33
+ </Markdown>
34
+ </TypographyStylesProvider>
35
+ );
36
+ };
37
+
38
+ export default FormattedMarkdown;
client/components/AiResponse/LoadingModelContent.tsx ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { Card, Group, Progress, Stack, Text } from "@mantine/core";
2
+
3
+ export default function LoadingModelContent({
4
+ modelLoadingProgress,
5
+ modelSizeInMegabytes,
6
+ }: {
7
+ modelLoadingProgress: number;
8
+ modelSizeInMegabytes: number;
9
+ }) {
10
+ const isLoadingStarting = modelLoadingProgress === 0;
11
+ const isLoadingComplete = modelLoadingProgress === 100;
12
+ const percent =
13
+ isLoadingComplete || isLoadingStarting ? 100 : modelLoadingProgress;
14
+ const strokeColor = percent === 100 ? "#52c41a" : "#3385ff";
15
+ const downloadedSize = (modelSizeInMegabytes * modelLoadingProgress) / 100;
16
+ const sizeText = `${downloadedSize.toFixed(0)} MB / ${modelSizeInMegabytes.toFixed(0)} MB`;
17
+
18
+ return (
19
+ <Card withBorder shadow="sm" radius="md">
20
+ <Card.Section withBorder inheritPadding py="xs">
21
+ <Text fw={500}>Loading AI...</Text>
22
+ </Card.Section>
23
+ <Card.Section withBorder inheritPadding py="md">
24
+ <Stack gap="xs">
25
+ <Progress color={strokeColor} value={percent} animated />
26
+ {!isLoadingStarting && (
27
+ <Group justify="space-between">
28
+ <Text size="sm" c="dimmed">
29
+ {sizeText}
30
+ </Text>
31
+ <Text size="sm" c="dimmed">
32
+ {percent.toFixed(1)}%
33
+ </Text>
34
+ </Group>
35
+ )}
36
+ </Stack>
37
+ </Card.Section>
38
+ </Card>
39
+ );
40
+ }
client/components/AiResponse/PreparingContent.tsx ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { Card, Skeleton, Stack, Text } from "@mantine/core";
2
+ import { match } from "ts-pattern";
3
+
4
+ export default function PreparingContent({
5
+ textGenerationState,
6
+ }: {
7
+ textGenerationState: string;
8
+ }) {
9
+ return (
10
+ <Card withBorder shadow="sm" radius="md">
11
+ <Card.Section withBorder inheritPadding py="xs">
12
+ <Text fw={500}>
13
+ {match(textGenerationState)
14
+ .with("awaitingSearchResults", () => "Awaiting search results...")
15
+ .with("preparingToGenerate", () => "Preparing AI response...")
16
+ .otherwise(() => null)}
17
+ </Text>
18
+ </Card.Section>
19
+ <Card.Section withBorder inheritPadding py="md">
20
+ <Stack>
21
+ <Skeleton height={8} radius="xl" />
22
+ <Skeleton height={8} width="70%" radius="xl" />
23
+ <Skeleton height={8} radius="xl" />
24
+ <Skeleton height={8} width="43%" radius="xl" />
25
+ </Stack>
26
+ </Card.Section>
27
+ </Card>
28
+ );
29
+ }
client/components/AiResponse/WebLlmModelSelect.tsx ADDED
@@ -0,0 +1,81 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { type ComboboxItem, Select } from "@mantine/core";
2
+ import { prebuiltAppConfig } from "@mlc-ai/web-llm";
3
+ import { useCallback, useEffect, useState } from "react";
4
+ import { isF16Supported } from "../../modules/webGpu";
5
+
6
+ export default function WebLlmModelSelect({
7
+ value,
8
+ onChange,
9
+ }: {
10
+ value: string;
11
+ onChange: (value: string) => void;
12
+ }) {
13
+ const [webGpuModels] = useState<ComboboxItem[]>(() => {
14
+ const models = prebuiltAppConfig.model_list
15
+ .filter((model) => {
16
+ const isSmall = isSmallModel(model);
17
+ const suffix = getModelSuffix(isF16Supported, isSmall);
18
+ return model.model_id.endsWith(suffix);
19
+ })
20
+ .sort((a, b) => (a.vram_required_MB ?? 0) - (b.vram_required_MB ?? 0))
21
+ .map((model) => {
22
+ const modelSizeInMegabytes =
23
+ Math.round(model.vram_required_MB ?? 0) || "N/A";
24
+ const isSmall = isSmallModel(model);
25
+ const suffix = getModelSuffix(isF16Supported, isSmall);
26
+ const modelName = model.model_id.replace(suffix, "");
27
+
28
+ return {
29
+ label: `${modelSizeInMegabytes} MB β€’ ${modelName}`,
30
+ value: model.model_id,
31
+ };
32
+ });
33
+
34
+ return models;
35
+ });
36
+
37
+ useEffect(() => {
38
+ const isCurrentModelValid = webGpuModels.some(
39
+ (model) => model.value === value,
40
+ );
41
+
42
+ if (!isCurrentModelValid && webGpuModels.length > 0) {
43
+ onChange(webGpuModels[0].value);
44
+ }
45
+ }, [onChange, webGpuModels, value]);
46
+
47
+ const handleChange = useCallback(
48
+ (value: string | null) => {
49
+ if (value) onChange(value);
50
+ },
51
+ [onChange],
52
+ );
53
+
54
+ return (
55
+ <Select
56
+ value={value}
57
+ onChange={handleChange}
58
+ label="AI Model"
59
+ description="Select the model to use for AI responses."
60
+ data={webGpuModels}
61
+ allowDeselect={false}
62
+ searchable
63
+ />
64
+ );
65
+ }
66
+
67
+ type ModelConfig = (typeof prebuiltAppConfig.model_list)[number];
68
+
69
+ const smallModels = ["SmolLM2-135M", "SmolLM2-360M"] as const;
70
+
71
+ function isSmallModel(model: ModelConfig) {
72
+ return smallModels.some((smallModel) =>
73
+ model.model_id.startsWith(smallModel),
74
+ );
75
+ }
76
+
77
+ function getModelSuffix(isF16: boolean, isSmall: boolean) {
78
+ if (isSmall) return isF16 ? "-q0f16-MLC" : "-q0f32-MLC";
79
+
80
+ return isF16 ? "-q4f16_1-MLC" : "-q4f32_1-MLC";
81
+ }
client/components/AiResponse/WllamaModelSelect.tsx ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { type ComboboxItem, Select } from "@mantine/core";
2
+ import { useEffect, useState } from "react";
3
+ import { wllamaModels } from "../../modules/wllama";
4
+
5
+ export default function WllamaModelSelect({
6
+ value,
7
+ onChange,
8
+ }: {
9
+ value: string;
10
+ onChange: (value: string) => void;
11
+ }) {
12
+ const [wllamaModelOptions] = useState<ComboboxItem[]>(
13
+ Object.entries(wllamaModels)
14
+ .sort(([, a], [, b]) => a.fileSizeInMegabytes - b.fileSizeInMegabytes)
15
+ .map(([value, { label, fileSizeInMegabytes }]) => ({
16
+ label: `${fileSizeInMegabytes} MB β€’ ${label}`,
17
+ value,
18
+ })),
19
+ );
20
+
21
+ useEffect(() => {
22
+ const isCurrentModelValid = wllamaModelOptions.some(
23
+ (model) => model.value === value,
24
+ );
25
+
26
+ if (!isCurrentModelValid && wllamaModelOptions.length > 0) {
27
+ onChange(wllamaModelOptions[0].value);
28
+ }
29
+ }, [onChange, wllamaModelOptions, value]);
30
+
31
+ return (
32
+ <Select
33
+ value={value}
34
+ onChange={(value) => value && onChange(value)}
35
+ label="AI Model"
36
+ description="Select the model to use for AI responses."
37
+ data={wllamaModelOptions}
38
+ allowDeselect={false}
39
+ searchable
40
+ />
41
+ );
42
+ }
client/components/App/App.tsx ADDED
@@ -0,0 +1,97 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { MantineProvider } from "@mantine/core";
2
+ import { Route, Switch } from "wouter";
3
+ import "@mantine/core/styles.css";
4
+ import { Notifications } from "@mantine/notifications";
5
+ import { usePubSub } from "create-pubsub/react";
6
+ import { lazy, useEffect, useState } from "react";
7
+ import { addLogEntry } from "../../modules/logEntries";
8
+ import { settingsPubSub } from "../../modules/pubSub";
9
+ import { defaultSettings } from "../../modules/settings";
10
+ import "@mantine/notifications/styles.css";
11
+ import { match } from "ts-pattern";
12
+ import { verifyStoredAccessKey } from "../../modules/accessKey";
13
+
14
+ const MainPage = lazy(() => import("../Pages/Main/MainPage"));
15
+ const AccessPage = lazy(() => import("../Pages/AccessPage"));
16
+
17
+ export function App() {
18
+ useInitializeSettings();
19
+ const { hasValidatedAccessKey, isCheckingStoredKey, setValidatedAccessKey } =
20
+ useAccessKeyValidation();
21
+
22
+ return match(isCheckingStoredKey)
23
+ .with(false, () => (
24
+ <MantineProvider defaultColorScheme="dark">
25
+ <Notifications />
26
+ <Switch>
27
+ <Route path="/">
28
+ {match([VITE_ACCESS_KEYS_ENABLED, hasValidatedAccessKey])
29
+ .with([true, false], () => (
30
+ <AccessPage
31
+ onAccessKeyValid={() => setValidatedAccessKey(true)}
32
+ />
33
+ ))
34
+ .otherwise(() => (
35
+ <MainPage />
36
+ ))}
37
+ </Route>
38
+ </Switch>
39
+ </MantineProvider>
40
+ ))
41
+ .otherwise(() => null);
42
+ }
43
+
44
+ /**
45
+ * A custom React hook that initializes the application settings.
46
+ *
47
+ * @returns The initialized settings object.
48
+ *
49
+ * @remarks
50
+ * This hook uses the `usePubSub` hook to access and update the settings state.
51
+ * It initializes the settings by merging the default settings with any existing settings.
52
+ * The initialization is performed once when the component mounts.
53
+ */
54
+ function useInitializeSettings() {
55
+ const [settings, setSettings] = usePubSub(settingsPubSub);
56
+ const [settingsInitialized, setSettingsInitialized] = useState(false);
57
+
58
+ useEffect(() => {
59
+ if (settingsInitialized) return;
60
+
61
+ setSettings({ ...defaultSettings, ...settings });
62
+
63
+ setSettingsInitialized(true);
64
+
65
+ addLogEntry("Settings initialized");
66
+ }, [settings, setSettings, settingsInitialized]);
67
+
68
+ return settings;
69
+ }
70
+
71
+ /**
72
+ * A custom React hook that validates the stored access key on mount.
73
+ *
74
+ * @returns An object containing the validation state and loading state
75
+ */
76
+ function useAccessKeyValidation() {
77
+ const [hasValidatedAccessKey, setValidatedAccessKey] = useState(false);
78
+ const [isCheckingStoredKey, setCheckingStoredKey] = useState(true);
79
+
80
+ useEffect(() => {
81
+ async function checkStoredAccessKey() {
82
+ if (VITE_ACCESS_KEYS_ENABLED) {
83
+ const isValid = await verifyStoredAccessKey();
84
+ if (isValid) setValidatedAccessKey(true);
85
+ }
86
+ setCheckingStoredKey(false);
87
+ }
88
+
89
+ checkStoredAccessKey();
90
+ }, []);
91
+
92
+ return {
93
+ hasValidatedAccessKey,
94
+ isCheckingStoredKey,
95
+ setValidatedAccessKey,
96
+ };
97
+ }
client/components/Logs/LogsModal.tsx ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import {
2
+ Alert,
3
+ Button,
4
+ Center,
5
+ Group,
6
+ Modal,
7
+ Pagination,
8
+ Table,
9
+ } from "@mantine/core";
10
+ import { IconInfoCircle } from "@tabler/icons-react";
11
+ import { usePubSub } from "create-pubsub/react";
12
+ import { useCallback, useMemo, useState } from "react";
13
+ import { logEntriesPubSub } from "../../modules/logEntries";
14
+
15
+ export default function LogsModal({
16
+ opened,
17
+ onClose,
18
+ }: {
19
+ opened: boolean;
20
+ onClose: () => void;
21
+ }) {
22
+ const [logEntries] = usePubSub(logEntriesPubSub);
23
+
24
+ const [page, setPage] = useState(1);
25
+
26
+ const logEntriesPerPage = 5;
27
+
28
+ const logEntriesFromCurrentPage = useMemo(
29
+ () =>
30
+ logEntries.slice(
31
+ (page - 1) * logEntriesPerPage,
32
+ page * logEntriesPerPage,
33
+ ),
34
+ [logEntries, page],
35
+ );
36
+
37
+ const downloadLogsAsJson = useCallback(() => {
38
+ const jsonString = JSON.stringify(logEntries, null, 2);
39
+ const blob = new Blob([jsonString], { type: "application/json" });
40
+ const url = URL.createObjectURL(blob);
41
+ const link = document.createElement("a");
42
+ link.href = url;
43
+ link.download = "logs.json";
44
+ document.body.appendChild(link);
45
+ link.click();
46
+ document.body.removeChild(link);
47
+ URL.revokeObjectURL(url);
48
+ }, [logEntries]);
49
+
50
+ return (
51
+ <Modal opened={opened} onClose={onClose} size="xl" title="Logs">
52
+ <Alert variant="light" color="blue" icon={<IconInfoCircle />} mb="md">
53
+ <Group justify="space-between" align="center">
54
+ <span>
55
+ This information is stored solely in your browser for personal use.
56
+ It isn't sent automatically and is retained for debugging purposes
57
+ should you need to{" "}
58
+ <a
59
+ href="https://github.com/felladrin/MiniSearch/issues/new?labels=bug&template=bug_report.yml"
60
+ target="_blank"
61
+ rel="noopener noreferrer"
62
+ >
63
+ report a bug
64
+ </a>
65
+ .
66
+ </span>
67
+ <Button onClick={downloadLogsAsJson} size="xs" data-autofocus>
68
+ Download Logs
69
+ </Button>
70
+ </Group>
71
+ </Alert>
72
+ <Table striped highlightOnHover withTableBorder>
73
+ <Table.Thead>
74
+ <Table.Tr>
75
+ <Table.Th>Time</Table.Th>
76
+ <Table.Th>Message</Table.Th>
77
+ </Table.Tr>
78
+ </Table.Thead>
79
+ <Table.Tbody>
80
+ {logEntriesFromCurrentPage.map((entry, index) => (
81
+ <Table.Tr key={`${entry.timestamp}-${index}`}>
82
+ <Table.Td>
83
+ {new Date(entry.timestamp).toLocaleTimeString()}
84
+ </Table.Td>
85
+ <Table.Td>{entry.message}</Table.Td>
86
+ </Table.Tr>
87
+ ))}
88
+ </Table.Tbody>
89
+ </Table>
90
+ <Center>
91
+ <Pagination
92
+ total={Math.ceil(logEntries.length / logEntriesPerPage)}
93
+ value={page}
94
+ onChange={setPage}
95
+ size="sm"
96
+ mt="md"
97
+ />
98
+ </Center>
99
+ </Modal>
100
+ );
101
+ }
client/components/Logs/ShowLogsButton.tsx ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { Button, Center, Loader, Stack, Text } from "@mantine/core";
2
+ import { Suspense, lazy, useState } from "react";
3
+ import { addLogEntry } from "../../modules/logEntries";
4
+
5
+ const LogsModal = lazy(() => import("./LogsModal"));
6
+
7
+ export default function ShowLogsButton() {
8
+ const [isLogsModalOpen, setLogsModalOpen] = useState(false);
9
+
10
+ const handleShowLogsButtonClick = () => {
11
+ addLogEntry("User opened the logs modal");
12
+ setLogsModalOpen(true);
13
+ };
14
+
15
+ const handleCloseLogsButtonClick = () => {
16
+ addLogEntry("User closed the logs modal");
17
+ setLogsModalOpen(false);
18
+ };
19
+
20
+ return (
21
+ <Stack gap="xs">
22
+ <Suspense
23
+ fallback={
24
+ <Center>
25
+ <Loader color="gray" type="bars" />
26
+ </Center>
27
+ }
28
+ >
29
+ <Button size="sm" onClick={handleShowLogsButtonClick} variant="default">
30
+ Show logs
31
+ </Button>
32
+ <Text size="xs" c="dimmed">
33
+ View session logs for debugging.
34
+ </Text>
35
+ <LogsModal
36
+ opened={isLogsModalOpen}
37
+ onClose={handleCloseLogsButtonClick}
38
+ />
39
+ </Suspense>
40
+ </Stack>
41
+ );
42
+ }
client/components/Pages/AccessPage.tsx ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { Button, Container, Stack, TextInput, Title } from "@mantine/core";
2
+ import { type FormEvent, useState } from "react";
3
+ import { validateAccessKey } from "../../modules/accessKey";
4
+ import { addLogEntry } from "../../modules/logEntries";
5
+
6
+ export default function AccessPage({
7
+ onAccessKeyValid,
8
+ }: {
9
+ onAccessKeyValid: () => void;
10
+ }) {
11
+ const [accessKey, setAccessKey] = useState("");
12
+ const [error, setError] = useState("");
13
+
14
+ const handleSubmit = async (formEvent: FormEvent<HTMLFormElement>) => {
15
+ formEvent.preventDefault();
16
+ setError("");
17
+ try {
18
+ const isValid = await validateAccessKey(accessKey);
19
+ if (isValid) {
20
+ addLogEntry("Valid access key entered");
21
+ onAccessKeyValid();
22
+ } else {
23
+ setError("Invalid access key");
24
+ addLogEntry("Invalid access key attempt");
25
+ }
26
+ } catch (error) {
27
+ setError("Error validating access key");
28
+ addLogEntry(`Error validating access key: ${error}`);
29
+ }
30
+ };
31
+
32
+ return (
33
+ <Container size="xs">
34
+ <Stack p="lg" mih="100vh" justify="center">
35
+ <Title order={2} ta="center">
36
+ Access Restricted
37
+ </Title>
38
+ <form onSubmit={handleSubmit}>
39
+ <Stack gap="xs">
40
+ <TextInput
41
+ value={accessKey}
42
+ onChange={({ target }) => setAccessKey(target.value)}
43
+ placeholder="Enter your access key to continue"
44
+ required
45
+ autoFocus
46
+ error={error}
47
+ styles={{
48
+ input: {
49
+ textAlign: "center",
50
+ },
51
+ }}
52
+ />
53
+ <Button size="xs" type="submit">
54
+ Submit
55
+ </Button>
56
+ </Stack>
57
+ </form>
58
+ </Stack>
59
+ </Container>
60
+ );
61
+ }
client/components/Pages/Main/MainPage.tsx ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { Center, Container, Loader, Stack } from "@mantine/core";
2
+ import { usePubSub } from "create-pubsub/react";
3
+ import { Suspense } from "react";
4
+ import { lazy } from "react";
5
+ import { Pattern, match } from "ts-pattern";
6
+ import {
7
+ queryPubSub,
8
+ searchStatePubSub,
9
+ textGenerationStatePubSub,
10
+ } from "../../../modules/pubSub";
11
+
12
+ const AiResponseSection = lazy(
13
+ () => import("../../AiResponse/AiResponseSection"),
14
+ );
15
+ const SearchResultsSection = lazy(
16
+ () => import("../../Search/Results/SearchResultsSection"),
17
+ );
18
+ const MenuButton = lazy(() => import("./Menu/MenuButton"));
19
+ const SearchForm = lazy(() => import("../../Search/Form/SearchForm"));
20
+
21
+ export default function MainPage() {
22
+ const [query, updateQuery] = usePubSub(queryPubSub);
23
+ const [searchState] = usePubSub(searchStatePubSub);
24
+ const [textGenerationState] = usePubSub(textGenerationStatePubSub);
25
+
26
+ return (
27
+ <Container>
28
+ <Stack
29
+ py="md"
30
+ mih="100vh"
31
+ justify={match(query)
32
+ .with(Pattern.string.length(0), () => "center")
33
+ .otherwise(() => undefined)}
34
+ >
35
+ <Suspense
36
+ fallback={
37
+ <Center>
38
+ <Loader type="bars" />
39
+ </Center>
40
+ }
41
+ >
42
+ <SearchForm
43
+ query={query}
44
+ updateQuery={updateQuery}
45
+ additionalButtons={<MenuButton />}
46
+ />
47
+ </Suspense>
48
+ {match(textGenerationState)
49
+ .with(Pattern.not("idle"), () => (
50
+ <Suspense>
51
+ <AiResponseSection />
52
+ </Suspense>
53
+ ))
54
+ .otherwise(() => null)}
55
+ {match(searchState)
56
+ .with(Pattern.not("idle"), () => (
57
+ <Suspense>
58
+ <SearchResultsSection />
59
+ </Suspense>
60
+ ))
61
+ .otherwise(() => null)}
62
+ </Stack>
63
+ </Container>
64
+ );
65
+ }
client/components/Pages/Main/Menu/AISettingsForm.tsx ADDED
@@ -0,0 +1,368 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import {
2
+ Group,
3
+ NumberInput,
4
+ Select,
5
+ Skeleton,
6
+ Slider,
7
+ Stack,
8
+ Switch,
9
+ Text,
10
+ TextInput,
11
+ Textarea,
12
+ } from "@mantine/core";
13
+ import { useForm } from "@mantine/form";
14
+ import { IconInfoCircle } from "@tabler/icons-react";
15
+ import { usePubSub } from "create-pubsub/react";
16
+ import { Suspense, lazy, useEffect, useState } from "react";
17
+ import { Pattern, match } from "ts-pattern";
18
+ import { addLogEntry } from "../../../../modules/logEntries";
19
+ import { getOpenAiClient } from "../../../../modules/openai";
20
+ import { settingsPubSub } from "../../../../modules/pubSub";
21
+ import { defaultSettings, inferenceTypes } from "../../../../modules/settings";
22
+ import { isWebGPUAvailable } from "../../../../modules/webGpu";
23
+
24
+ const WebLlmModelSelect = lazy(
25
+ () => import("../../../AiResponse/WebLlmModelSelect"),
26
+ );
27
+ const WllamaModelSelect = lazy(
28
+ () => import("../../../AiResponse/WllamaModelSelect"),
29
+ );
30
+
31
+ export default function AISettingsForm() {
32
+ const [settings, setSettings] = usePubSub(settingsPubSub);
33
+ const [openAiModels, setOpenAiModels] = useState<
34
+ {
35
+ label: string;
36
+ value: string;
37
+ }[]
38
+ >([]);
39
+ const [openAiApiModelError, setOpenAiApiModelError] = useState<
40
+ string | undefined
41
+ >(undefined);
42
+
43
+ const form = useForm({
44
+ initialValues: settings,
45
+ onValuesChange: setSettings,
46
+ });
47
+
48
+ useEffect(() => {
49
+ async function fetchOpenAiModels() {
50
+ try {
51
+ const openai = getOpenAiClient({
52
+ baseURL: settings.openAiApiBaseUrl,
53
+ apiKey: settings.openAiApiKey,
54
+ });
55
+ const response = await openai.models.list();
56
+ const models = response.data.map((model) => ({
57
+ label: model.id,
58
+ value: model.id,
59
+ }));
60
+ setOpenAiModels(models);
61
+ setOpenAiApiModelError(undefined);
62
+ } catch (error) {
63
+ const errorMessage =
64
+ error instanceof Error ? error.message : String(error);
65
+ addLogEntry(`Error fetching OpenAI models: ${errorMessage}`);
66
+ setOpenAiModels([]);
67
+ setOpenAiApiModelError(errorMessage);
68
+ }
69
+ }
70
+
71
+ if (settings.inferenceType === "openai" && settings.openAiApiBaseUrl) {
72
+ fetchOpenAiModels();
73
+ }
74
+ }, [
75
+ settings.inferenceType,
76
+ settings.openAiApiBaseUrl,
77
+ settings.openAiApiKey,
78
+ ]);
79
+
80
+ useEffect(() => {
81
+ if (openAiApiModelError === form.errors.openAiApiModel) return;
82
+
83
+ form.setFieldError("openAiApiModel", openAiApiModelError);
84
+ }, [openAiApiModelError, form.setFieldError, form.errors.openAiApiModel]);
85
+
86
+ useEffect(() => {
87
+ if (openAiModels.length > 0) {
88
+ const hasNoModelSelected = !form.values.openAiApiModel;
89
+ const isModelInvalid = !openAiModels.find(
90
+ (model) => model.value === form.values.openAiApiModel,
91
+ );
92
+
93
+ if (hasNoModelSelected || isModelInvalid) {
94
+ form.setFieldValue("openAiApiModel", openAiModels[0].value);
95
+ }
96
+ } else if (form.values.openAiApiModel) {
97
+ form.setFieldValue("openAiApiModel", "");
98
+ }
99
+ }, [openAiModels, form.setFieldValue, form.values.openAiApiModel]);
100
+
101
+ const isUsingCustomInstructions =
102
+ form.values.systemPrompt !== defaultSettings.systemPrompt;
103
+
104
+ const handleRestoreDefaultInstructions = () => {
105
+ form.setFieldValue("systemPrompt", defaultSettings.systemPrompt);
106
+ };
107
+
108
+ const suggestedCpuThreads =
109
+ (navigator.hardwareConcurrency &&
110
+ Math.max(
111
+ defaultSettings.cpuThreads,
112
+ navigator.hardwareConcurrency - 2,
113
+ )) ??
114
+ defaultSettings.cpuThreads;
115
+
116
+ return (
117
+ <Stack gap="md">
118
+ <Switch
119
+ label="AI Response"
120
+ {...form.getInputProps("enableAiResponse", {
121
+ type: "checkbox",
122
+ })}
123
+ labelPosition="left"
124
+ description="Enable or disable AI-generated responses to your queries. When disabled, you'll only see web search results."
125
+ />
126
+
127
+ {form.values.enableAiResponse && (
128
+ <>
129
+ <Stack gap="xs" mb="md">
130
+ <Text size="sm">Search results to consider</Text>
131
+ <Text size="xs" c="dimmed">
132
+ Determines the number of search results to consider when
133
+ generating AI responses. A higher value may enhance accuracy, but
134
+ it will also increase response time.
135
+ </Text>
136
+ <Slider
137
+ {...form.getInputProps("searchResultsToConsider")}
138
+ min={0}
139
+ max={6}
140
+ marks={Array.from({ length: 7 }, (_, index) => ({
141
+ value: index,
142
+ label: index.toString(),
143
+ }))}
144
+ />
145
+ </Stack>
146
+
147
+ <Select
148
+ {...form.getInputProps("inferenceType")}
149
+ label="AI Processing Location"
150
+ data={inferenceTypes}
151
+ allowDeselect={false}
152
+ />
153
+
154
+ {form.values.inferenceType === "openai" && (
155
+ <>
156
+ <TextInput
157
+ {...form.getInputProps("openAiApiBaseUrl")}
158
+ label="API Base URL"
159
+ placeholder="http://localhost:11434/v1"
160
+ required
161
+ />
162
+ <Group gap="xs">
163
+ <IconInfoCircle size={16} />
164
+ <Text size="xs" c="dimmed" flex={1}>
165
+ You may need to add{" "}
166
+ <em>
167
+ {`${self.location.protocol}//${self.location.hostname}`}
168
+ </em>{" "}
169
+ to the list of allowed network origins in your API server
170
+ settings.
171
+ </Text>
172
+ </Group>
173
+ <TextInput
174
+ {...form.getInputProps("openAiApiKey")}
175
+ label="API Key"
176
+ type="password"
177
+ description="Optional, as local API servers usually do not require it."
178
+ />
179
+ <Select
180
+ {...form.getInputProps("openAiApiModel")}
181
+ label="API Model"
182
+ data={openAiModels}
183
+ description="Optional, as some API servers don't provide a model list."
184
+ allowDeselect={false}
185
+ disabled={openAiModels.length === 0}
186
+ searchable
187
+ />
188
+ </>
189
+ )}
190
+
191
+ {form.values.inferenceType === "browser" && (
192
+ <>
193
+ {isWebGPUAvailable && (
194
+ <Switch
195
+ label="WebGPU"
196
+ {...form.getInputProps("enableWebGpu", {
197
+ type: "checkbox",
198
+ })}
199
+ labelPosition="left"
200
+ description="Enable or disable WebGPU usage. When disabled, the app will use the CPU instead."
201
+ />
202
+ )}
203
+
204
+ {match([isWebGPUAvailable, form.values.enableWebGpu])
205
+ .with([true, true], () => (
206
+ <Suspense fallback={<Skeleton height={50} />}>
207
+ <WebLlmModelSelect
208
+ value={form.values.webLlmModelId}
209
+ onChange={(value) =>
210
+ form.setFieldValue("webLlmModelId", value)
211
+ }
212
+ />
213
+ </Suspense>
214
+ ))
215
+ .with([false, Pattern.any], [Pattern.any, false], () => (
216
+ <>
217
+ <Suspense fallback={<Skeleton height={50} />}>
218
+ <WllamaModelSelect
219
+ value={form.values.wllamaModelId}
220
+ onChange={(value) =>
221
+ form.setFieldValue("wllamaModelId", value)
222
+ }
223
+ />
224
+ </Suspense>
225
+ <NumberInput
226
+ label="CPU threads to use"
227
+ description={
228
+ <>
229
+ <span>
230
+ Number of threads to use for the AI model. Lower
231
+ values will use less CPU but may take longer to
232
+ respond. A value that is too high may cause the app
233
+ to hang.
234
+ </span>
235
+ {suggestedCpuThreads > defaultSettings.cpuThreads && (
236
+ <span>
237
+ {" "}
238
+ The default value is{" "}
239
+ <Text
240
+ component="span"
241
+ size="xs"
242
+ c="blue"
243
+ style={{ cursor: "pointer" }}
244
+ onClick={() =>
245
+ form.setFieldValue(
246
+ "cpuThreads",
247
+ defaultSettings.cpuThreads,
248
+ )
249
+ }
250
+ >
251
+ {defaultSettings.cpuThreads}
252
+ </Text>
253
+ , but based on the number of logical processors in
254
+ your CPU, the suggested value is{" "}
255
+ <Text
256
+ component="span"
257
+ size="xs"
258
+ c="blue"
259
+ style={{ cursor: "pointer" }}
260
+ onClick={() =>
261
+ form.setFieldValue(
262
+ "cpuThreads",
263
+ suggestedCpuThreads,
264
+ )
265
+ }
266
+ >
267
+ {suggestedCpuThreads}
268
+ </Text>
269
+ .
270
+ </span>
271
+ )}
272
+ </>
273
+ }
274
+ min={1}
275
+ {...form.getInputProps("cpuThreads")}
276
+ />
277
+ </>
278
+ ))
279
+ .otherwise(() => null)}
280
+ </>
281
+ )}
282
+
283
+ <Textarea
284
+ label="Instructions for AI"
285
+ descriptionProps={{ component: "div" }}
286
+ description={
287
+ <>
288
+ <span>
289
+ Customize instructions for the AI to tailor its responses.
290
+ </span>
291
+ <br />
292
+ <span>For example:</span>
293
+ <ul>
294
+ <li>
295
+ Specify preferences
296
+ <ul>
297
+ <li>
298
+ <em>"use simple language"</em>
299
+ </li>
300
+ <li>
301
+ <em>"provide step-by-step explanations"</em>
302
+ </li>
303
+ </ul>
304
+ </li>
305
+ <li>
306
+ Set a response style
307
+ <ul>
308
+ <li>
309
+ <em>"answer in a friendly tone"</em>
310
+ </li>
311
+ <li>
312
+ <em>"write your response in Spanish"</em>
313
+ </li>
314
+ </ul>
315
+ </li>
316
+ <li>
317
+ Provide context about the audience
318
+ <ul>
319
+ <li>
320
+ <em>"you're talking to a high school student"</em>
321
+ </li>
322
+ <li>
323
+ <em>
324
+ "consider that your audience is composed of
325
+ professionals in the field of graphic design"
326
+ </em>
327
+ </li>
328
+ </ul>
329
+ </li>
330
+ </ul>
331
+ <span>
332
+ The special tag <em>{"{{searchResults}}"}</em> will be
333
+ replaced with the search results, while{" "}
334
+ <em>{"{{dateTime}}"}</em> will be replaced with the current
335
+ date and time.
336
+ </span>
337
+ {isUsingCustomInstructions && (
338
+ <>
339
+ <br />
340
+ <br />
341
+ <span>
342
+ Currently, you're using custom instructions. If you ever
343
+ need to restore the default instructions, you can do so by
344
+ clicking
345
+ </span>{" "}
346
+ <Text
347
+ component="span"
348
+ size="xs"
349
+ c="blue"
350
+ style={{ cursor: "pointer" }}
351
+ onClick={handleRestoreDefaultInstructions}
352
+ >
353
+ here
354
+ </Text>
355
+ <span>.</span>
356
+ </>
357
+ )}
358
+ </>
359
+ }
360
+ autosize
361
+ maxRows={10}
362
+ {...form.getInputProps("systemPrompt")}
363
+ />
364
+ </>
365
+ )}
366
+ </Stack>
367
+ );
368
+ }
client/components/Pages/Main/Menu/ActionsForm.tsx ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { Stack } from "@mantine/core";
2
+ import { Suspense, lazy } from "react";
3
+
4
+ const ClearDataButton = lazy(() => import("./ClearDataButton"));
5
+ const ShowLogsButton = lazy(() => import("../../../Logs/ShowLogsButton"));
6
+
7
+ export default function ActionsForm() {
8
+ return (
9
+ <Stack gap="lg">
10
+ <Suspense>
11
+ <ClearDataButton />
12
+ </Suspense>
13
+ <Suspense>
14
+ <ShowLogsButton />
15
+ </Suspense>
16
+ </Stack>
17
+ );
18
+ }
client/components/Pages/Main/Menu/ClearDataButton.tsx ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { Button, Stack, Text } from "@mantine/core";
2
+ import { useState } from "react";
3
+ import { useLocation } from "wouter";
4
+ import { addLogEntry } from "../../../../modules/logEntries";
5
+
6
+ export default function ClearDataButton() {
7
+ const [isClearingData, setIsClearingData] = useState(false);
8
+ const [hasClearedData, setHasClearedData] = useState(false);
9
+ const [, navigate] = useLocation();
10
+
11
+ const handleClearDataButtonClick = async () => {
12
+ const sureToDelete = self.confirm(
13
+ "Are you sure you want to reset the settings and delete all files in cache?",
14
+ );
15
+
16
+ if (!sureToDelete) return;
17
+
18
+ addLogEntry("User initiated data clearing");
19
+
20
+ setIsClearingData(true);
21
+
22
+ self.localStorage.clear();
23
+
24
+ for (const cacheName of await self.caches.keys()) {
25
+ await self.caches.delete(cacheName);
26
+ }
27
+
28
+ for (const databaseInfo of await self.indexedDB.databases()) {
29
+ if (databaseInfo.name) self.indexedDB.deleteDatabase(databaseInfo.name);
30
+ }
31
+
32
+ setIsClearingData(false);
33
+
34
+ setHasClearedData(true);
35
+
36
+ addLogEntry("All data cleared successfully");
37
+
38
+ navigate("/", { replace: true });
39
+
40
+ self.location.reload();
41
+ };
42
+
43
+ return (
44
+ <Stack gap="xs">
45
+ <Button
46
+ onClick={handleClearDataButtonClick}
47
+ variant="default"
48
+ loading={isClearingData}
49
+ loaderProps={{ type: "bars" }}
50
+ disabled={hasClearedData}
51
+ >
52
+ {hasClearedData ? "Data cleared" : "Clear all data"}
53
+ </Button>
54
+ <Text size="xs" c="dimmed">
55
+ Reset settings and delete all files in cache to free up space.
56
+ </Text>
57
+ </Stack>
58
+ );
59
+ }
client/components/Pages/Main/Menu/InterfaceSettingsForm.tsx ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import {
2
+ Stack,
3
+ Switch,
4
+ useComputedColorScheme,
5
+ useMantineColorScheme,
6
+ } from "@mantine/core";
7
+ import { useForm } from "@mantine/form";
8
+ import { usePubSub } from "create-pubsub/react";
9
+ import { settingsPubSub } from "../../../../modules/pubSub";
10
+
11
+ export default function InterfaceSettingsForm() {
12
+ const [settings, setSettings] = usePubSub(settingsPubSub);
13
+ const form = useForm({
14
+ initialValues: settings,
15
+ onValuesChange: setSettings,
16
+ });
17
+ const { setColorScheme } = useMantineColorScheme();
18
+ const computedColorScheme = useComputedColorScheme("light");
19
+
20
+ const toggleColorScheme = () => {
21
+ setColorScheme(computedColorScheme === "dark" ? "light" : "dark");
22
+ };
23
+
24
+ return (
25
+ <Stack gap="md">
26
+ <Switch
27
+ label="Dark Mode"
28
+ checked={computedColorScheme === "dark"}
29
+ onChange={toggleColorScheme}
30
+ labelPosition="left"
31
+ description="Enable or disable the dark color scheme."
32
+ styles={{ labelWrapper: { width: "100%" } }}
33
+ />
34
+
35
+ <Switch
36
+ {...form.getInputProps("enterToSubmit", {
37
+ type: "checkbox",
38
+ })}
39
+ label="Enter to Submit"
40
+ labelPosition="left"
41
+ description="Enable or disable using Enter key to submit the search query. When disabled, you'll need to click the Search button or use Shift+Enter to submit."
42
+ />
43
+ </Stack>
44
+ );
45
+ }
client/components/Pages/Main/Menu/MenuButton.tsx ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { Button } from "@mantine/core";
2
+ import { Suspense, lazy, useCallback, useEffect, useState } from "react";
3
+ import { addLogEntry } from "../../../../modules/logEntries";
4
+
5
+ const MenuDrawer = lazy(() => import("./MenuDrawer"));
6
+
7
+ export default function MenuButton() {
8
+ const [isDrawerOpen, setDrawerOpen] = useState(false);
9
+ const [isDrawerLoaded, setDrawerLoaded] = useState(false);
10
+
11
+ const openDrawer = useCallback(() => {
12
+ setDrawerOpen(true);
13
+ addLogEntry("User opened the menu");
14
+ }, []);
15
+
16
+ const closeDrawer = useCallback(() => {
17
+ setDrawerOpen(false);
18
+ addLogEntry("User closed the menu");
19
+ }, []);
20
+
21
+ const handleDrawerLoad = useCallback(() => {
22
+ if (!isDrawerLoaded) {
23
+ addLogEntry("Menu drawer loaded");
24
+ setDrawerLoaded(true);
25
+ }
26
+ }, [isDrawerLoaded]);
27
+
28
+ return (
29
+ <>
30
+ <Button
31
+ size="xs"
32
+ onClick={openDrawer}
33
+ variant="default"
34
+ loading={isDrawerOpen && !isDrawerLoaded}
35
+ >
36
+ Menu
37
+ </Button>
38
+ {(isDrawerOpen || isDrawerLoaded) && (
39
+ <Suspense fallback={<SuspenseListener onUnload={handleDrawerLoad} />}>
40
+ <MenuDrawer onClose={closeDrawer} opened={isDrawerOpen} />
41
+ </Suspense>
42
+ )}
43
+ </>
44
+ );
45
+ }
46
+
47
+ function SuspenseListener({ onUnload }: { onUnload: () => void }) {
48
+ useEffect(() => {
49
+ return () => onUnload();
50
+ }, [onUnload]);
51
+
52
+ return null;
53
+ }
client/components/Pages/Main/Menu/MenuDrawer.tsx ADDED
@@ -0,0 +1,111 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import {
2
+ Accordion,
3
+ ActionIcon,
4
+ Center,
5
+ Drawer,
6
+ type DrawerProps,
7
+ FocusTrap,
8
+ Group,
9
+ HoverCard,
10
+ Stack,
11
+ } from "@mantine/core";
12
+ import { IconBrandGithub } from "@tabler/icons-react";
13
+ import prettyMilliseconds from "pretty-ms";
14
+ import { Suspense, lazy } from "react";
15
+ import { repository } from "../../../../../package.json";
16
+ import { addLogEntry } from "../../../../modules/logEntries";
17
+ import { getSemanticVersion } from "../../../../modules/stringFormatters";
18
+
19
+ const AISettingsForm = lazy(() => import("./AISettingsForm"));
20
+ const SearchSettingsForm = lazy(() => import("./SearchSettingsForm"));
21
+ const InterfaceSettingsForm = lazy(() => import("./InterfaceSettingsForm"));
22
+ const ActionsForm = lazy(() => import("./ActionsForm"));
23
+
24
+ export default function MenuDrawer(drawerProps: DrawerProps) {
25
+ const repoName = repository.url.split("/").pop();
26
+
27
+ return (
28
+ <Drawer
29
+ {...drawerProps}
30
+ position="right"
31
+ size="md"
32
+ title={
33
+ <Group gap="xs">
34
+ <ActionIcon
35
+ variant="subtle"
36
+ component="a"
37
+ color="var(--mantine-color-text)"
38
+ href={repository.url}
39
+ target="_blank"
40
+ onClick={() => addLogEntry("User clicked the GitHub link")}
41
+ >
42
+ <IconBrandGithub size={16} />
43
+ </ActionIcon>
44
+ <HoverCard shadow="md" withArrow>
45
+ <HoverCard.Target>
46
+ <Center>{repoName}</Center>
47
+ </HoverCard.Target>
48
+ <HoverCard.Dropdown>
49
+ <Stack gap="xs">
50
+ <Center>{repoName}</Center>
51
+ <Center>
52
+ {`v${getSemanticVersion(VITE_BUILD_DATE_TIME)}+${VITE_COMMIT_SHORT_HASH}`}
53
+ </Center>
54
+ <Center>
55
+ Released{" "}
56
+ {prettyMilliseconds(
57
+ new Date().getTime() -
58
+ new Date(VITE_BUILD_DATE_TIME).getTime(),
59
+ {
60
+ compact: true,
61
+ verbose: true,
62
+ },
63
+ )}{" "}
64
+ ago
65
+ </Center>
66
+ </Stack>
67
+ </HoverCard.Dropdown>
68
+ </HoverCard>
69
+ </Group>
70
+ }
71
+ >
72
+ <FocusTrap.InitialFocus />
73
+ <Drawer.Body>
74
+ <Accordion variant="separated" multiple>
75
+ <Accordion.Item value="aiSettings">
76
+ <Accordion.Control>AI Settings</Accordion.Control>
77
+ <Accordion.Panel>
78
+ <Suspense>
79
+ <AISettingsForm />
80
+ </Suspense>
81
+ </Accordion.Panel>
82
+ </Accordion.Item>
83
+ <Accordion.Item value="searchSettings">
84
+ <Accordion.Control>Search Settings</Accordion.Control>
85
+ <Accordion.Panel>
86
+ <Suspense>
87
+ <SearchSettingsForm />
88
+ </Suspense>
89
+ </Accordion.Panel>
90
+ </Accordion.Item>
91
+ <Accordion.Item value="interfaceSettings">
92
+ <Accordion.Control>Interface Settings</Accordion.Control>
93
+ <Accordion.Panel>
94
+ <Suspense>
95
+ <InterfaceSettingsForm />
96
+ </Suspense>
97
+ </Accordion.Panel>
98
+ </Accordion.Item>
99
+ <Accordion.Item value="actions">
100
+ <Accordion.Control>Actions</Accordion.Control>
101
+ <Accordion.Panel>
102
+ <Suspense>
103
+ <ActionsForm />
104
+ </Suspense>
105
+ </Accordion.Panel>
106
+ </Accordion.Item>
107
+ </Accordion>
108
+ </Drawer.Body>
109
+ </Drawer>
110
+ );
111
+ }
client/components/Pages/Main/Menu/SearchSettingsForm.tsx ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { Slider, Stack, Switch, Text } from "@mantine/core";
2
+ import { useForm } from "@mantine/form";
3
+ import { usePubSub } from "create-pubsub/react";
4
+ import { settingsPubSub } from "../../../../modules/pubSub";
5
+
6
+ export default function SearchSettingsForm() {
7
+ const [settings, setSettings] = usePubSub(settingsPubSub);
8
+ const form = useForm({
9
+ initialValues: settings,
10
+ onValuesChange: setSettings,
11
+ });
12
+
13
+ return (
14
+ <Stack gap="md">
15
+ <Stack gap="xs" mb="md">
16
+ <Text size="sm">Search Results Limit</Text>
17
+ <Text size="xs" c="dimmed">
18
+ Maximum number of search results to fetch. A higher value provides
19
+ more results but may increase search time.
20
+ </Text>
21
+ <Slider
22
+ {...form.getInputProps("searchResultsLimit")}
23
+ min={5}
24
+ max={30}
25
+ step={5}
26
+ marks={[5, 10, 15, 20, 25, 30].map((value) => ({
27
+ value,
28
+ label: value.toString(),
29
+ }))}
30
+ />
31
+ </Stack>
32
+
33
+ <Switch
34
+ {...form.getInputProps("enableImageSearch", {
35
+ type: "checkbox",
36
+ })}
37
+ label="Image Search"
38
+ labelPosition="left"
39
+ description="Enable or disable image search results. When enabled, relevant images will be displayed alongside web search results."
40
+ />
41
+ </Stack>
42
+ );
43
+ }
client/components/Search/Form/SearchForm.tsx ADDED
@@ -0,0 +1,140 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { Button, Group, Stack, Textarea } from "@mantine/core";
2
+ import { usePubSub } from "create-pubsub/react";
3
+ import {
4
+ type ChangeEvent,
5
+ type KeyboardEvent,
6
+ type ReactNode,
7
+ useCallback,
8
+ useEffect,
9
+ useRef,
10
+ useState,
11
+ } from "react";
12
+ import { Pattern, match } from "ts-pattern";
13
+ import { useLocation } from "wouter";
14
+ import { addLogEntry } from "../../../modules/logEntries";
15
+ import { postMessageToParentWindow } from "../../../modules/parentWindow";
16
+ import { settingsPubSub } from "../../../modules/pubSub";
17
+ import { getRandomQuerySuggestion } from "../../../modules/querySuggestions";
18
+ import { sleepUntilIdle } from "../../../modules/sleep";
19
+ import { searchAndRespond } from "../../../modules/textGeneration";
20
+
21
+ export default function SearchForm({
22
+ query,
23
+ updateQuery,
24
+ additionalButtons,
25
+ }: {
26
+ query: string;
27
+ updateQuery: (query: string) => void;
28
+ additionalButtons?: ReactNode;
29
+ }) {
30
+ const textAreaRef = useRef<HTMLTextAreaElement>(null);
31
+ const [textAreaValue, setTextAreaValue] = useState(query);
32
+ const defaultSuggestedQuery = "Anything you need!";
33
+ const [suggestedQuery, setSuggestedQuery] = useState(defaultSuggestedQuery);
34
+ const [, navigate] = useLocation();
35
+ const [settings] = usePubSub(settingsPubSub);
36
+
37
+ const handleMount = useCallback(async () => {
38
+ await sleepUntilIdle();
39
+ searchAndRespond();
40
+ }, []);
41
+
42
+ const handleInitialSuggestion = useCallback(async () => {
43
+ const suggestion = await getRandomQuerySuggestion();
44
+ setSuggestedQuery(suggestion);
45
+ }, []);
46
+
47
+ useEffect(() => {
48
+ handleMount();
49
+ handleInitialSuggestion();
50
+ }, [handleMount, handleInitialSuggestion]);
51
+
52
+ const handleInputChange = async (event: ChangeEvent<HTMLTextAreaElement>) => {
53
+ const text = event.target.value;
54
+
55
+ setTextAreaValue(text);
56
+
57
+ if (text.length === 0) {
58
+ setSuggestedQuery(await getRandomQuerySuggestion());
59
+ }
60
+ };
61
+
62
+ const handleClearButtonClick = async () => {
63
+ setSuggestedQuery(await getRandomQuerySuggestion());
64
+ setTextAreaValue("");
65
+ textAreaRef.current?.focus();
66
+ addLogEntry("User cleaned the search query field");
67
+ };
68
+
69
+ const startSearching = useCallback(() => {
70
+ const queryToEncode = match(textAreaValue.trim())
71
+ .with(Pattern.string.minLength(1), () => textAreaValue)
72
+ .otherwise(() => suggestedQuery);
73
+
74
+ setTextAreaValue(queryToEncode);
75
+
76
+ const queryString = `q=${encodeURIComponent(queryToEncode)}`;
77
+
78
+ postMessageToParentWindow({ queryString, hash: "" });
79
+
80
+ navigate(`/?${queryString}`, { replace: true });
81
+
82
+ updateQuery(queryToEncode);
83
+
84
+ searchAndRespond();
85
+
86
+ addLogEntry(
87
+ `User submitted a search with ${queryToEncode.length} characters length`,
88
+ );
89
+ }, [textAreaValue, suggestedQuery, updateQuery, navigate]);
90
+
91
+ const handleSubmit = (event: { preventDefault: () => void }) => {
92
+ event.preventDefault();
93
+ startSearching();
94
+ };
95
+
96
+ const handleKeyDown = (event: KeyboardEvent<HTMLTextAreaElement>) => {
97
+ match([event, settings.enterToSubmit])
98
+ .with(
99
+ [{ code: "Enter", shiftKey: false }, true],
100
+ [{ code: "Enter", shiftKey: true }, false],
101
+ () => handleSubmit(event),
102
+ )
103
+ .otherwise(() => undefined);
104
+ };
105
+
106
+ return (
107
+ <form onSubmit={handleSubmit} style={{ width: "100%" }}>
108
+ <Stack gap="xs">
109
+ <Textarea
110
+ value={textAreaValue}
111
+ placeholder={suggestedQuery}
112
+ ref={textAreaRef}
113
+ onKeyDown={handleKeyDown}
114
+ onChange={handleInputChange}
115
+ autosize
116
+ minRows={1}
117
+ maxRows={8}
118
+ autoFocus
119
+ />
120
+ <Group gap="xs">
121
+ {match(textAreaValue)
122
+ .with(Pattern.string.minLength(1), () => (
123
+ <Button
124
+ size="xs"
125
+ onClick={handleClearButtonClick}
126
+ variant="default"
127
+ >
128
+ Clear
129
+ </Button>
130
+ ))
131
+ .otherwise(() => null)}
132
+ <Button size="xs" type="submit" variant="default" flex={1}>
133
+ Search
134
+ </Button>
135
+ {additionalButtons}
136
+ </Group>
137
+ </Stack>
138
+ </form>
139
+ );
140
+ }
client/components/Search/Results/Graphical/ImageResultsList.tsx ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { Carousel } from "@mantine/carousel";
2
+ import { Button, Group, Stack, Text, Transition, rem } from "@mantine/core";
3
+ import { useEffect, useState } from "react";
4
+ import type { SearchResults } from "../../../../modules/search";
5
+ import "@mantine/carousel/styles.css";
6
+ import Lightbox from "yet-another-react-lightbox";
7
+ import Captions from "yet-another-react-lightbox/plugins/captions";
8
+ import "yet-another-react-lightbox/styles.css";
9
+ import "yet-another-react-lightbox/plugins/captions.css";
10
+ import { addLogEntry } from "../../../../modules/logEntries";
11
+ import { getHostname } from "../../../../modules/stringFormatters";
12
+
13
+ export default function ImageResultsList({
14
+ imageResults,
15
+ }: {
16
+ imageResults: SearchResults["imageResults"];
17
+ }) {
18
+ const [isLightboxOpen, setLightboxOpen] = useState(false);
19
+ const [lightboxIndex, setLightboxIndex] = useState(0);
20
+ const [canStartTransition, setCanStartTransition] = useState(false);
21
+
22
+ useEffect(() => {
23
+ setCanStartTransition(true);
24
+ }, []);
25
+
26
+ const handleImageClick = (index: number) => {
27
+ setLightboxIndex(index);
28
+ setLightboxOpen(true);
29
+ };
30
+
31
+ const imageStyle = {
32
+ objectFit: "cover",
33
+ height: rem(180),
34
+ width: rem(240),
35
+ borderRadius: rem(4),
36
+ border: `${rem(2)} solid var(--mantine-color-default-border)`,
37
+ cursor: "zoom-in",
38
+ } as const;
39
+
40
+ return (
41
+ <>
42
+ <Carousel slideSize="0" slideGap="xs" align="start" dragFree loop>
43
+ {imageResults.map(([title, sourceUrl, thumbnailUrl], index) => (
44
+ <Transition
45
+ key={`${title}-${sourceUrl}-${thumbnailUrl}`}
46
+ mounted={canStartTransition}
47
+ transition="fade"
48
+ timingFunction="ease"
49
+ enterDelay={index * 250}
50
+ duration={1500}
51
+ >
52
+ {(styles) => (
53
+ <Carousel.Slide style={styles}>
54
+ <img
55
+ alt={title}
56
+ src={thumbnailUrl}
57
+ onClick={() => handleImageClick(index)}
58
+ onKeyDown={(e) => {
59
+ if (e.key === "Enter") {
60
+ handleImageClick(index);
61
+ }
62
+ }}
63
+ style={imageStyle}
64
+ />
65
+ </Carousel.Slide>
66
+ )}
67
+ </Transition>
68
+ ))}
69
+ </Carousel>
70
+ <Lightbox
71
+ open={isLightboxOpen}
72
+ close={() => setLightboxOpen(false)}
73
+ plugins={[Captions]}
74
+ index={lightboxIndex}
75
+ slides={imageResults.map(([title, url, thumbnailUrl, sourceUrl]) => ({
76
+ src: thumbnailUrl,
77
+ description: (
78
+ <Stack align="center" gap="md">
79
+ {title && (
80
+ <Text component="cite" ta="center">
81
+ {title}
82
+ </Text>
83
+ )}
84
+ <Group align="center" justify="center" gap="xs">
85
+ <Button
86
+ variant="subtle"
87
+ component="a"
88
+ size="xs"
89
+ href={sourceUrl}
90
+ target="_blank"
91
+ title="Click to see the image in full size"
92
+ rel="noopener noreferrer"
93
+ onClick={() => {
94
+ addLogEntry("User visited an image result in full size");
95
+ }}
96
+ >
97
+ View in full resolution
98
+ </Button>
99
+ <Button
100
+ variant="subtle"
101
+ component="a"
102
+ href={url}
103
+ target="_blank"
104
+ size="xs"
105
+ title="Click to visit the page where the image was found"
106
+ rel="noopener noreferrer"
107
+ onClick={() => {
108
+ addLogEntry("User visited an image result source");
109
+ }}
110
+ >
111
+ Visit {getHostname(url)}
112
+ </Button>
113
+ </Group>
114
+ </Stack>
115
+ ),
116
+ }))}
117
+ />
118
+ </>
119
+ );
120
+ }
client/components/Search/Results/SearchResultsSection.tsx ADDED
@@ -0,0 +1,132 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import {
2
+ Alert,
3
+ AspectRatio,
4
+ Divider,
5
+ Group,
6
+ Skeleton,
7
+ Space,
8
+ Stack,
9
+ em,
10
+ } from "@mantine/core";
11
+ import { useMediaQuery } from "@mantine/hooks";
12
+ import { IconInfoCircle } from "@tabler/icons-react";
13
+ import { usePubSub } from "create-pubsub/react";
14
+ import { Suspense, lazy, useMemo } from "react";
15
+ import { Pattern, match } from "ts-pattern";
16
+ import {
17
+ searchResultsPubSub,
18
+ searchStatePubSub,
19
+ settingsPubSub,
20
+ } from "../../../modules/pubSub";
21
+ import type { SearchResults } from "../../../modules/search";
22
+ import type { Settings } from "../../../modules/settings";
23
+
24
+ const ImageResultsList = lazy(() => import("./Graphical/ImageResultsList"));
25
+ const SearchResultsList = lazy(() => import("./Textual/SearchResultsList"));
26
+
27
+ export default function SearchResultsSection() {
28
+ const [searchResults] = usePubSub(searchResultsPubSub);
29
+ const [searchState] = usePubSub(searchStatePubSub);
30
+ const [settings] = usePubSub(settingsPubSub);
31
+
32
+ return useMemo(
33
+ () =>
34
+ match(searchState)
35
+ .with("running", () => <RunningSearchContent />)
36
+ .with("failed", () => <FailedSearchContent />)
37
+ .with("completed", () => (
38
+ <CompletedSearchContent
39
+ searchResults={searchResults}
40
+ settings={settings}
41
+ />
42
+ ))
43
+ .otherwise(() => null),
44
+ [searchState, searchResults, settings],
45
+ );
46
+ }
47
+
48
+ function RunningSearchContent() {
49
+ const hasSmallScreen = useMediaQuery(`(max-width: ${em(530)})`);
50
+ const numberOfSquareSkeletons = hasSmallScreen ? 4 : 6;
51
+ const skeletonIds = Array.from(
52
+ { length: numberOfSquareSkeletons },
53
+ (_, i) => `skeleton-${i}`,
54
+ );
55
+
56
+ return (
57
+ <>
58
+ <Divider
59
+ mb="sm"
60
+ variant="dashed"
61
+ labelPosition="center"
62
+ label="Searching the web..."
63
+ />
64
+ <Stack>
65
+ <Group>
66
+ {skeletonIds.map((id) => (
67
+ <AspectRatio key={id} ratio={1} flex={1}>
68
+ <Skeleton />
69
+ </AspectRatio>
70
+ ))}
71
+ </Group>
72
+ <Stack>
73
+ <Skeleton height={8} radius="xl" />
74
+ <Skeleton height={8} width="87%" radius="xl" />
75
+ <Skeleton height={8} radius="xl" />
76
+ <Skeleton height={8} width="70%" radius="xl" />
77
+ <Skeleton height={8} radius="xl" />
78
+ <Skeleton height={8} width="52%" radius="xl" />
79
+ <Skeleton height={8} radius="xl" />
80
+ <Skeleton height={8} width="63%" radius="xl" />
81
+ </Stack>
82
+ </Stack>
83
+ </>
84
+ );
85
+ }
86
+
87
+ function FailedSearchContent() {
88
+ return (
89
+ <>
90
+ <Divider
91
+ mb="sm"
92
+ variant="dashed"
93
+ labelPosition="center"
94
+ label="Search Results"
95
+ />
96
+ <Alert
97
+ variant="light"
98
+ color="yellow"
99
+ title="No results found"
100
+ icon={<IconInfoCircle />}
101
+ >
102
+ It looks like your current search did not return any results. Try
103
+ refining your search by adding more keywords or rephrasing your query.
104
+ </Alert>
105
+ </>
106
+ );
107
+ }
108
+
109
+ function CompletedSearchContent({
110
+ searchResults,
111
+ settings,
112
+ }: {
113
+ searchResults: SearchResults;
114
+ settings: Settings;
115
+ }) {
116
+ return (
117
+ <>
118
+ <Divider variant="dashed" labelPosition="center" label="Search Results" />
119
+ {match([settings.enableImageSearch, searchResults.imageResults.length])
120
+ .with([true, Pattern.number.positive()], () => (
121
+ <Suspense>
122
+ <ImageResultsList imageResults={searchResults.imageResults} />
123
+ <Space h={8} />
124
+ </Suspense>
125
+ ))
126
+ .otherwise(() => null)}
127
+ <Suspense>
128
+ <SearchResultsList searchResults={searchResults.textResults} />
129
+ </Suspense>
130
+ </>
131
+ );
132
+ }
client/components/Search/Results/Textual/SearchResultsList.tsx ADDED
@@ -0,0 +1,87 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import {
2
+ Flex,
3
+ Stack,
4
+ Text,
5
+ Tooltip,
6
+ Transition,
7
+ UnstyledButton,
8
+ em,
9
+ } from "@mantine/core";
10
+ import { useMediaQuery } from "@mantine/hooks";
11
+ import { useEffect, useState } from "react";
12
+ import { addLogEntry } from "../../../../modules/logEntries";
13
+ import type { SearchResults } from "../../../../modules/search";
14
+ import { getHostname } from "../../../../modules/stringFormatters";
15
+
16
+ export default function SearchResultsList({
17
+ searchResults,
18
+ }: {
19
+ searchResults: SearchResults["textResults"];
20
+ }) {
21
+ const shouldDisplayDomainBelowTitle = useMediaQuery(
22
+ `(max-width: ${em(720)})`,
23
+ );
24
+ const [canStartTransition, setCanStartTransition] = useState(false);
25
+
26
+ useEffect(() => {
27
+ setCanStartTransition(true);
28
+ }, []);
29
+
30
+ return (
31
+ <Stack gap={40}>
32
+ {searchResults.map(([title, snippet, url], index) => (
33
+ <Transition
34
+ key={url}
35
+ mounted={canStartTransition}
36
+ transition="fade"
37
+ timingFunction="ease"
38
+ enterDelay={index * 200}
39
+ duration={750}
40
+ >
41
+ {(styles) => (
42
+ <Stack gap={16} style={styles}>
43
+ <Flex
44
+ gap={shouldDisplayDomainBelowTitle ? 0 : 16}
45
+ justify="space-between"
46
+ align="flex-start"
47
+ direction={shouldDisplayDomainBelowTitle ? "column" : "row"}
48
+ >
49
+ <UnstyledButton
50
+ variant="transparent"
51
+ component="a"
52
+ href={url}
53
+ target="_blank"
54
+ onClick={() => {
55
+ addLogEntry("User clicked a text result");
56
+ }}
57
+ >
58
+ <Text fw="bold" c="var(--mantine-color-blue-light-color)">
59
+ {title}
60
+ </Text>
61
+ </UnstyledButton>
62
+ <Tooltip label={url}>
63
+ <UnstyledButton
64
+ variant="transparent"
65
+ component="a"
66
+ href={url}
67
+ target="_blank"
68
+ fs="italic"
69
+ ta="end"
70
+ onClick={() => {
71
+ addLogEntry("User clicked a text result");
72
+ }}
73
+ >
74
+ {getHostname(url)}
75
+ </UnstyledButton>
76
+ </Tooltip>
77
+ </Flex>
78
+ <Text size="sm" c="dimmed" style={{ wordWrap: "break-word" }}>
79
+ {snippet}
80
+ </Text>
81
+ </Stack>
82
+ )}
83
+ </Transition>
84
+ ))}
85
+ </Stack>
86
+ );
87
+ }
client/index.html ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ <!doctype html>
2
+ <html lang="en">
3
+ <head>
4
+ <meta charset="utf-8" />
5
+ <meta
6
+ name="viewport"
7
+ content="width=device-width, initial-scale=1.0, user-scalable=no"
8
+ />
9
+ <meta
10
+ name="description"
11
+ content="Minimalist web-searching app with an AI assistant that is always available and runs directly from your browser."
12
+ />
13
+ <meta itemprop="name" content="MiniSearch" />
14
+ <meta
15
+ itemprop="description"
16
+ content="Minimalist web-searching app with an AI assistant that is always available and runs directly from your browser."
17
+ />
18
+ <meta property="og:type" content="website" />
19
+ <meta property="og:title" content="MiniSearch" />
20
+ <meta
21
+ property="og:description"
22
+ content="Minimalist web-searching app with an AI assistant that is always available and runs directly from your browser."
23
+ />
24
+ <meta name="twitter:card" content="summary" />
25
+ <meta name="twitter:title" content="MiniSearch" />
26
+ <meta
27
+ name="twitter:description"
28
+ content="Minimalist web-searching app with an AI assistant that is always available and runs directly from your browser."
29
+ />
30
+ <title>MiniSearch</title>
31
+ <link rel="icon" href="/favicon.png" />
32
+ </head>
33
+ <body>
34
+ <script type="module" src="./index.tsx"></script>
35
+ </body>
36
+ </html>
client/index.tsx ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import { createRoot } from "react-dom/client";
2
+ import { App } from "./components/App/App";
3
+ import { addLogEntry } from "./modules/logEntries";
4
+
5
+ createRoot(document.body.appendChild(document.createElement("div"))).render(
6
+ <App />,
7
+ );
8
+
9
+ addLogEntry("App initialized");
client/modules/accessKey.ts ADDED
@@ -0,0 +1,95 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { notifications } from "@mantine/notifications";
2
+ import { argon2id } from "hash-wasm";
3
+ import { addLogEntry } from "./logEntries";
4
+
5
+ const ACCESS_KEY_STORAGE_KEY = "accessKeyHash";
6
+
7
+ interface StoredAccessKey {
8
+ hash: string;
9
+ timestamp: number;
10
+ }
11
+
12
+ async function hashAccessKey(accessKey: string): Promise<string> {
13
+ const salt = new Uint8Array(16);
14
+ crypto.getRandomValues(salt);
15
+
16
+ return argon2id({
17
+ password: accessKey,
18
+ salt,
19
+ parallelism: 1,
20
+ iterations: 16,
21
+ memorySize: 512,
22
+ hashLength: 8,
23
+ outputType: "encoded",
24
+ });
25
+ }
26
+
27
+ export async function validateAccessKey(accessKey: string): Promise<boolean> {
28
+ try {
29
+ const hash = await hashAccessKey(accessKey);
30
+ const response = await fetch("/api/validate-access-key", {
31
+ method: "POST",
32
+ headers: { "Content-Type": "application/json" },
33
+ body: JSON.stringify({ accessKeyHash: hash }),
34
+ });
35
+ const data = await response.json();
36
+
37
+ if (data.valid) {
38
+ const storedData: StoredAccessKey = {
39
+ hash,
40
+ timestamp: Date.now(),
41
+ };
42
+ localStorage.setItem(ACCESS_KEY_STORAGE_KEY, JSON.stringify(storedData));
43
+ addLogEntry("Access key hash stored");
44
+ }
45
+
46
+ return data.valid;
47
+ } catch (error) {
48
+ addLogEntry(`Error validating access key: ${error}`);
49
+ notifications.show({
50
+ title: "Error validating access key",
51
+ message: "Please contact the administrator",
52
+ color: "red",
53
+ position: "top-right",
54
+ });
55
+ return false;
56
+ }
57
+ }
58
+
59
+ export async function verifyStoredAccessKey(): Promise<boolean> {
60
+ if (VITE_ACCESS_KEY_TIMEOUT_HOURS === 0) return false;
61
+
62
+ const storedData = localStorage.getItem(ACCESS_KEY_STORAGE_KEY);
63
+ if (!storedData) return false;
64
+
65
+ try {
66
+ const { hash, timestamp }: StoredAccessKey = JSON.parse(storedData);
67
+
68
+ const expirationTime = VITE_ACCESS_KEY_TIMEOUT_HOURS * 60 * 60 * 1000;
69
+ if (Date.now() - timestamp > expirationTime) {
70
+ localStorage.removeItem(ACCESS_KEY_STORAGE_KEY);
71
+ addLogEntry("Stored access key expired");
72
+ return false;
73
+ }
74
+
75
+ const response = await fetch("/api/validate-access-key", {
76
+ method: "POST",
77
+ headers: { "Content-Type": "application/json" },
78
+ body: JSON.stringify({ accessKeyHash: hash }),
79
+ });
80
+
81
+ const data = await response.json();
82
+ if (!data.valid) {
83
+ localStorage.removeItem(ACCESS_KEY_STORAGE_KEY);
84
+ addLogEntry("Stored access key is no longer valid");
85
+ return false;
86
+ }
87
+
88
+ addLogEntry("Using stored access key");
89
+ return true;
90
+ } catch (error) {
91
+ addLogEntry(`Error verifying stored access key: ${error}`);
92
+ localStorage.removeItem(ACCESS_KEY_STORAGE_KEY);
93
+ return false;
94
+ }
95
+ }
client/modules/logEntries.ts ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { createPubSub } from "create-pubsub";
2
+
3
+ type LogEntry = {
4
+ timestamp: string;
5
+ message: string;
6
+ };
7
+
8
+ export const logEntriesPubSub = createPubSub<LogEntry[]>([]);
9
+
10
+ const [updateLogEntries, , getLogEntries] = logEntriesPubSub;
11
+
12
+ export function addLogEntry(message: string) {
13
+ updateLogEntries([
14
+ ...getLogEntries(),
15
+ {
16
+ timestamp: new Date().toISOString(),
17
+ message,
18
+ },
19
+ ]);
20
+ }
client/modules/openai.ts ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import OpenAI, { type ClientOptions } from "openai";
2
+
3
+ export function getOpenAiClient({
4
+ baseURL,
5
+ apiKey,
6
+ }: {
7
+ baseURL: ClientOptions["baseURL"];
8
+ apiKey: ClientOptions["apiKey"];
9
+ }) {
10
+ return new OpenAI({
11
+ baseURL,
12
+ apiKey,
13
+ dangerouslyAllowBrowser: true,
14
+ defaultHeaders: { "X-Stainless-Retry-Count": null },
15
+ });
16
+ }
client/modules/parentWindow.ts ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ export function postMessageToParentWindow(message: unknown) {
2
+ const parentWindow = self.parent;
3
+ const targetOrigin = parentWindow?.[0]?.location?.ancestorOrigins?.[0];
4
+ if (targetOrigin) parentWindow.postMessage(message, targetOrigin);
5
+ }
client/modules/pubSub.ts ADDED
@@ -0,0 +1,106 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { createPubSub } from "create-pubsub";
2
+ import throttle from "throttleit";
3
+ import { addLogEntry } from "./logEntries";
4
+ import { defaultSettings } from "./settings";
5
+
6
+ function createLocalStoragePubSub<T>(localStorageKey: string, defaultValue: T) {
7
+ const localStorageValue = localStorage.getItem(localStorageKey);
8
+ const localStoragePubSub = createPubSub(
9
+ localStorageValue ? (JSON.parse(localStorageValue) as T) : defaultValue,
10
+ );
11
+
12
+ const [, onValueChange] = localStoragePubSub;
13
+
14
+ onValueChange((value) =>
15
+ localStorage.setItem(localStorageKey, JSON.stringify(value)),
16
+ );
17
+
18
+ return localStoragePubSub;
19
+ }
20
+
21
+ const querySuggestionsPubSub = createLocalStoragePubSub<string[]>(
22
+ "querySuggestions",
23
+ [],
24
+ );
25
+
26
+ const lastSearchTokenHashPubSub = createLocalStoragePubSub(
27
+ "lastSearchTokenHash",
28
+ "",
29
+ );
30
+
31
+ export const [updateLastSearchTokenHash, , getLastSearchTokenHash] =
32
+ lastSearchTokenHashPubSub;
33
+
34
+ export const [updateQuerySuggestions, , getQuerySuggestions] =
35
+ querySuggestionsPubSub;
36
+
37
+ export const queryPubSub = createPubSub(
38
+ new URLSearchParams(self.location.search).get("q") ?? "",
39
+ );
40
+
41
+ export const [, , getQuery] = queryPubSub;
42
+
43
+ export const responsePubSub = createPubSub("");
44
+
45
+ export const updateResponse = throttle(responsePubSub[0], 1000 / 12);
46
+
47
+ export const searchResultsPubSub = createPubSub<
48
+ import("./search").SearchResults
49
+ >({
50
+ textResults: [],
51
+ imageResults: [],
52
+ });
53
+
54
+ export const [updateSearchResults, , getSearchResults] = searchResultsPubSub;
55
+
56
+ export const [updateSearchPromise, , getSearchPromise] = createPubSub<
57
+ Promise<import("./search").SearchResults>
58
+ >(Promise.resolve({ textResults: [], imageResults: [] }));
59
+
60
+ export const textGenerationStatePubSub = createPubSub<
61
+ | "idle"
62
+ | "awaitingModelDownloadAllowance"
63
+ | "loadingModel"
64
+ | "awaitingSearchResults"
65
+ | "preparingToGenerate"
66
+ | "generating"
67
+ | "interrupted"
68
+ | "failed"
69
+ | "completed"
70
+ >("idle");
71
+
72
+ export const [
73
+ updateTextGenerationState,
74
+ listenToTextGenerationStateChanges,
75
+ getTextGenerationState,
76
+ ] = textGenerationStatePubSub;
77
+
78
+ listenToTextGenerationStateChanges((textGenerationState) => {
79
+ addLogEntry(`Text generation state changed to '${textGenerationState}'`);
80
+ });
81
+
82
+ export const searchStatePubSub = createPubSub<
83
+ "idle" | "running" | "failed" | "completed"
84
+ >("idle");
85
+
86
+ export const [updateSearchState, listenToSearchStateChanges] =
87
+ searchStatePubSub;
88
+
89
+ listenToSearchStateChanges((searchState) => {
90
+ addLogEntry(`Search state changed to '${searchState}'`);
91
+ });
92
+
93
+ export const modelLoadingProgressPubSub = createPubSub(0);
94
+
95
+ export const [updateModelLoadingProgress] = modelLoadingProgressPubSub;
96
+
97
+ export const settingsPubSub = createLocalStoragePubSub(
98
+ "settings",
99
+ defaultSettings,
100
+ );
101
+
102
+ export const [, listenToSettingsChanges, getSettings] = settingsPubSub;
103
+
104
+ export const modelSizeInMegabytesPubSub = createPubSub(0);
105
+
106
+ export const [updateModelSizeInMegabytes] = modelSizeInMegabytesPubSub;
client/modules/querySuggestions.ts ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { addLogEntry } from "./logEntries";
2
+ import { getQuerySuggestions, updateQuerySuggestions } from "./pubSub";
3
+
4
+ export async function getRandomQuerySuggestion() {
5
+ if (getQuerySuggestions().length === 0) await refillQuerySuggestions(25);
6
+
7
+ const querySuggestions = getQuerySuggestions();
8
+
9
+ const randomQuerySuggestion = querySuggestions.pop() as string;
10
+
11
+ updateQuerySuggestions(querySuggestions);
12
+
13
+ return randomQuerySuggestion;
14
+ }
15
+
16
+ async function refillQuerySuggestions(limit?: number) {
17
+ const querySuggestionsFileUrl = new URL(
18
+ "/query-suggestions.json",
19
+ self.location.origin,
20
+ );
21
+
22
+ const fetchResponse = await fetch(querySuggestionsFileUrl.toString());
23
+
24
+ const querySuggestionsList: string[] = await fetchResponse.json();
25
+
26
+ updateQuerySuggestions(
27
+ querySuggestionsList.sort(() => Math.random() - 0.5).slice(0, limit),
28
+ );
29
+
30
+ addLogEntry(`Query suggestions refilled with ${limit} suggestions`);
31
+ }