root
/
tabby
mirror of https://gitee.com/zhang_1334717033/tabby.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: hide streamlit menu in production (#50)

add-more-languages
Meng Zhang 2023-04-06 17:21:26 +08:00 committed by GitHub
parent ef483564fe
commit aacc7e575e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
12 changed files with 38 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
development/docker-compose.dev.yml
View File

@ -14,5 +14,6 @@ services:
VECTOR_WATCH_CONFIG: true VECTOR_WATCH_CONFIG: true
STREAMLIT_RUN_ON_SAVE: true STREAMLIT_RUN_ON_SAVE: true
CADDY_WATCH_CONFIG: true CADDY_WATCH_CONFIG: true
STREAMLIT_HIDE_MENU: false
volumes: volumes:
- ../tabby:/home/app/tabby - ../tabby:/home/app/tabby

13
tabby/admin/Home.py
View File

@ -1,5 +1,6 @@
import streamlit as st import streamlit as st
from utils.service_info import ServiceInfo from utils.service_info import ServiceInfo
from utils.streamlit import set_page_config
SERVICES = [ SERVICES = [
ServiceInfo(label="triton", health_url="http://localhost:8002/metrics"), ServiceInfo(label="triton", health_url="http://localhost:8002/metrics"),
@ -13,7 +14,7 @@ def make_badge_markdown(x: ServiceInfo):
return f"![{x.label}]({x.badge_url})" return f"![{x.label}]({x.badge_url})"
st.set_page_config(page_title="Tabby Admin - Home", layout="wide") set_page_config(page_title="Home")
badges = " ".join(map(make_badge_markdown, SERVICES)) badges = " ".join(map(make_badge_markdown, SERVICES))
st.markdown( st.markdown(
@ -24,13 +25,11 @@ st.markdown(
**Congrats, your server is live!** **Congrats, your server is live!**
you can now query the server using `/v1/completions` endpoint: To get started with Tabby, you can either install the extensions below or use the [Editor](./Editor).
```bash ### Extensions
curl -X POST http://localhost:5000/v1/completions -H 'Content-Type: application/json' --data '{
"prompt": "def binarySearch(arr, left, right, x):\\n mid = (left +" * [VSCode](https://marketplace.visualstudio.com/items?itemName=TabbyML.vscode-tabby)
}'
```
""".replace( """.replace(
"{badges}", badges "{badges}", badges

4
tabby/admin/components/monaco/__init__.py
View File

@ -30,8 +30,8 @@ else:
# `declare_component` and call it done. The wrapper allows us to customize # `declare_component` and call it done. The wrapper allows us to customize
# our component's API: we can pre-process its input args, post-process its # our component's API: we can pre-process its input args, post-process its
# output value, and add a docstring for users. # output value, and add a docstring for users.
def st_monaco(key, tabby_server_url=None, code=None): def st_monaco(key, tabby_server_url=None, code=None, height=400):
_editor_func(tabby_server_url=tabby_server_url, code=code, key=key) _editor_func(tabby_server_url=tabby_server_url, code=code, height=height, key=key)
# Add some test code to play with the component while it's in development. # Add some test code to play with the component while it's in development.

4
tabby/admin/components/monaco/frontend/build/asset-manifest.json
View File

@ -1,9 +1,9 @@
{ {
"files": { "files": {
"main.js": "./static/js/main.13cf5263.js", "main.js": "./static/js/main.c928cca6.js",
"index.html": "./index.html" "index.html": "./index.html"
}, },
"entrypoints": [ "entrypoints": [
"static/js/main.13cf5263.js" "static/js/main.c928cca6.js"
] ]
} }

2
tabby/admin/components/monaco/frontend/build/index.html
View File

@ -1 +1 @@
<!doctype html><html lang="en"><head><title>Streamlit Component</title><meta charset="UTF-8"/><meta name="viewport" content="width=device-width,initial-scale=1"/><meta name="theme-color" content="#000000"/><meta name="description" content="Streamlit Component"/><link rel="stylesheet" href="bootstrap.min.css"/><script defer="defer" src="./static/js/main.13cf5263.js"></script></head><body><noscript>You need to enable JavaScript to run this app.</noscript><div id="root"></div></body></html> <!doctype html><html lang="en"><head><title>Streamlit Component</title><meta charset="UTF-8"/><meta name="viewport" content="width=device-width,initial-scale=1"/><meta name="theme-color" content="#000000"/><meta name="description" content="Streamlit Component"/><link rel="stylesheet" href="bootstrap.min.css"/><script defer="defer" src="./static/js/main.c928cca6.js"></script></head><body><noscript>You need to enable JavaScript to run this app.</noscript><div id="root"></div></body></html>

4
tabby/admin/components/monaco/frontend/build/static/js/main.13cf5263.js → tabby/admin/components/monaco/frontend/build/static/js/main.c928cca6.js
View File

File diff suppressed because one or more lines are too long

0
tabby/admin/components/monaco/frontend/build/static/js/main.13cf5263.js.LICENSE.txt → tabby/admin/components/monaco/frontend/build/static/js/main.c928cca6.js.LICENSE.txt
View File

2
tabby/admin/components/monaco/frontend/src/Monaco.jsx
View File

@ -38,7 +38,7 @@ export default function MonacoEditor() {
}, [renderData.args.code, editorRef]) }, [renderData.args.code, editorRef])
return ( return (
<div style={{ height: 400 }}> <div style={{ height: renderData.args.height }}>
<Editor <Editor
theme="vs-dark" theme="vs-dark"
defaultLanguage="python" defaultLanguage="python"

8
tabby/admin/pages/Editor.py
View File

@ -1,10 +1,8 @@
import streamlit as st import streamlit as st
from components import monaco from components import monaco
from utils.streamlit import set_page_config
st.set_page_config(page_title="Tabby Admin - Editor", layout="wide") set_page_config(page_title="Editor")
st.markdown("## Editor")
st.markdown("---")
SNIPPETS = { SNIPPETS = {
"Clear": "# Write some code ...", "Clear": "# Write some code ...",
@ -37,4 +35,4 @@ def code_presets():
return code return code
monaco.st_monaco(key="default", code=code_presets()) monaco.st_monaco(key="default", code=code_presets(), height=600)

3
tabby/admin/pages/Metrics.py
View File

@ -2,8 +2,9 @@ import os
import duckdb import duckdb
import streamlit as st import streamlit as st
from utils.streamlit import set_page_config
st.set_page_config(page_title="Tabby Admin - Metrics", layout="wide") set_page_config(page_title="Metrics")
def query_data(): def query_data():

3
tabby/admin/pages/Projects.py
View File

@ -6,8 +6,9 @@ import streamlit as st
import toml import toml
from datasets import load_from_disk from datasets import load_from_disk
from git import Repo from git import Repo
from utils.streamlit import set_page_config
st.set_page_config(page_title="Tabby Admin - Projects", layout="wide") set_page_config(page_title="Projects")
dataset_dir = os.environ.get("DATASET_DIR", None) dataset_dir = os.environ.get("DATASET_DIR", None)
git_repositories_dir = os.environ.get("GIT_REPOSITORIES_DIR", None) git_repositories_dir = os.environ.get("GIT_REPOSITORIES_DIR", None)

16
tabby/admin/utils/streamlit.py Normal file
View File

@ -0,0 +1,16 @@
import os
import streamlit as st
hide_streamlit_menu = os.environ.get("STREAMLIT_HIDE_MENU", "True") == "True"
def set_page_config(page_title, **kwargs):
st.set_page_config(
page_title=f"Tabby Admin - {page_title}", layout="wide", **kwargs
)
if hide_streamlit_menu:
hide_streamlit_style = (
"<style>#MainMenu {visibility: hidden;}footer {visibility: hidden;}</style>"
)
st.markdown(hide_streamlit_style, unsafe_allow_html=True)