109 lines
2.9 KiB
Python
109 lines
2.9 KiB
Python
from typing import Dict, Any
|
|
import os
|
|
|
|
|
|
class AppSettings:
|
|
# UI Configuration
|
|
UPLOAD_STYLE = {
|
|
"width": "100%",
|
|
"height": "60px",
|
|
"lineHeight": "60px",
|
|
"borderWidth": "1px",
|
|
"borderStyle": "dashed",
|
|
"borderRadius": "5px",
|
|
"textAlign": "center",
|
|
"margin-bottom": "20px",
|
|
}
|
|
|
|
PROMPTS_UPLOAD_STYLE = {**UPLOAD_STYLE, "borderColor": "#28a745"}
|
|
|
|
PLOT_CONFIG = {"responsive": True, "displayModeBar": True}
|
|
|
|
PLOT_STYLE = {"height": "85vh", "width": "100%"}
|
|
|
|
PLOT_LAYOUT_CONFIG = {
|
|
"height": None,
|
|
"autosize": True,
|
|
"margin": dict(l=0, r=0, t=50, b=0),
|
|
}
|
|
|
|
# Dimensionality Reduction Settings
|
|
DEFAULT_N_COMPONENTS_3D = 3
|
|
DEFAULT_N_COMPONENTS_2D = 2
|
|
DEFAULT_RANDOM_STATE = 42
|
|
|
|
# Available Methods
|
|
REDUCTION_METHODS = [
|
|
{"label": "PCA", "value": "pca"},
|
|
{"label": "t-SNE", "value": "tsne"},
|
|
{"label": "UMAP", "value": "umap"},
|
|
]
|
|
|
|
COLOR_OPTIONS = [
|
|
{"label": "Category", "value": "category"},
|
|
{"label": "Subcategory", "value": "subcategory"},
|
|
{"label": "Tags", "value": "tags"},
|
|
]
|
|
|
|
DIMENSION_OPTIONS = [{"label": "2D", "value": "2d"}, {"label": "3D", "value": "3d"}]
|
|
|
|
# Default Values
|
|
DEFAULT_METHOD = "pca"
|
|
DEFAULT_COLOR_BY = "category"
|
|
DEFAULT_DIMENSIONS = "3d"
|
|
DEFAULT_SHOW_PROMPTS = ["show"]
|
|
|
|
# Plot Marker Settings
|
|
DOCUMENT_MARKER_SIZE_2D = 8
|
|
DOCUMENT_MARKER_SIZE_3D = 5
|
|
PROMPT_MARKER_SIZE_2D = 10
|
|
PROMPT_MARKER_SIZE_3D = 6
|
|
|
|
DOCUMENT_MARKER_SYMBOL = "circle"
|
|
PROMPT_MARKER_SYMBOL = "diamond"
|
|
|
|
DOCUMENT_OPACITY = 1.0
|
|
PROMPT_OPACITY = 0.8
|
|
|
|
# Text Processing
|
|
TEXT_PREVIEW_LENGTH = 100
|
|
|
|
# App Configuration
|
|
DEBUG = os.getenv("EMBEDDINGBUDDY_DEBUG", "True").lower() == "true"
|
|
HOST = os.getenv("EMBEDDINGBUDDY_HOST", "127.0.0.1")
|
|
PORT = int(os.getenv("EMBEDDINGBUDDY_PORT", "8050"))
|
|
|
|
# OpenSearch Configuration
|
|
OPENSEARCH_DEFAULT_SIZE = 100
|
|
OPENSEARCH_SAMPLE_SIZE = 5
|
|
OPENSEARCH_CONNECTION_TIMEOUT = 30
|
|
OPENSEARCH_VERIFY_CERTS = True
|
|
|
|
# Bootstrap Theme
|
|
EXTERNAL_STYLESHEETS = [
|
|
"https://cdn.jsdelivr.net/npm/bootstrap@5.1.3/dist/css/bootstrap.min.css"
|
|
]
|
|
|
|
@classmethod
|
|
def get_plot_marker_config(
|
|
cls, dimensions: str, is_prompt: bool = False
|
|
) -> Dict[str, Any]:
|
|
if is_prompt:
|
|
size = (
|
|
cls.PROMPT_MARKER_SIZE_3D
|
|
if dimensions == "3d"
|
|
else cls.PROMPT_MARKER_SIZE_2D
|
|
)
|
|
symbol = cls.PROMPT_MARKER_SYMBOL
|
|
opacity = cls.PROMPT_OPACITY
|
|
else:
|
|
size = (
|
|
cls.DOCUMENT_MARKER_SIZE_3D
|
|
if dimensions == "3d"
|
|
else cls.DOCUMENT_MARKER_SIZE_2D
|
|
)
|
|
symbol = cls.DOCUMENT_MARKER_SYMBOL
|
|
opacity = cls.DOCUMENT_OPACITY
|
|
|
|
return {"size": size, "symbol": symbol, "opacity": opacity}
|