📦 lichon / streamlit-demo

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68import streamlit as st
import subprocess
import os

from streamlit_wakeup import keepAlive
from peertunnel import peerTunnel
from cloudflared import cloudflared
from streamlit_ttyd import get_ttyd
from streamlit.components.v1 import iframe


def ttyd(
    cmd: str = "echo terminal-speaking... && sleep 99999",
    auth: str = None,
    writable: bool = True,
    port: int = 1234,
    exit_on_disconnect: bool = False,
    base_url: str = None,
):
    flags = f"--port {port} "
    if base_url:
        flags += f"-b {base_url} "
    if exit_on_disconnect:
        flags += "--once "
    if auth:
        flags += f"--credential {auth} "
    if writable:
        flags += "-W"

    # check if user provided path to ttyd
    ttyd = get_ttyd()
    ttydproc = subprocess.Popen(
        f"{ttyd} {flags} {cmd}",
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
        shell=True,
    )
    return ttydproc, port


# start the peer tunnel
peerTunnel(secrets=st.secrets)

# start the ttyd server
ttyd(cmd="bash", port=1234, auth=st.secrets["ttyd_auth"], base_url='/r/localhost:1234')

tty_url = 'http://localhost:1234/r/localhost:1234'
peer_url = 'http://localhost:2234'

# streamlit server
if os.getenv("HOSTNAME") == "streamlit":
    # start playwright
    keepAlive(secrets=st.secrets)
    # set dns for cloudflared
    peer_url = cloudflared(2234, update_dns=True, secrets=st.secrets).tunnel
    tty_url = peer_url + '/r/localhost:1234'


st.set_page_config(page_title="Streamlit Terminal", layout="wide")
# embed ttyd
iframe(tty_url, height=700)
# info on ttyd port
st.text(f"peer server is running at : {peer_url}")
st.text(
    f"peer pid {peerTunnel.proc.pid if peerTunnel.is_alive() else 'dead'} "
    f"playwright pid {keepAlive.proc.pid if keepAlive.is_alive() else 'dead'}"
)