Spaces:
Sleeping
Sleeping
File size: 4,127 Bytes
3e7a501 |
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 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 |
import app_utils as util
import streamlit as st
import time
from pathlib import Path
from datetime import datetime
import requests
st.set_page_config(page_title="sandl", page_icon="🩴", layout="wide", initial_sidebar_state="collapsed")
# sidebar, page, sidebar
# margin-bottom: 30px;
st.markdown("""
<style>
.prompt-font {
font-size:18px !important;
margin-bottom: 10px;
margin-top: 20px;
}
.percent-font {
font-size:30px !important;
margin-top: 50px;
}
.st-emotion-cache-79elbk {
margin-top: -65px;
}
.st-emotion-cache-1wmy9hl {
margin-top: -32px;
}
[data-testid="stSidebarNav"] {
background-image: url(https://media.discordapp.net/attachments/1013678935231442944/1175577023574454402/sandlsmall_2.png?ex=656bbc6b&is=6559476b&hm=c920463c2062c3c42e555224fb0dc0afc5188e9451b45c9eb62c64f297c1a1f1&=&width=596&height=127);
background-repeat: no-repeat;
margin-top: 20px;
background-position: 20px 20px;
}
</style>
""", unsafe_allow_html=True)
# Create a row with 3 columns
col1, col2, col3 = st.columns([1, 8, 1])
with col3:
if st.button('👤 Profile'):
util.switch_page('profile')
with col2:
# tab bar
query = r"$\textsf{\large Query}$"
recents = r"$\textsf{\large History}$"
q, searches = st.tabs([query, recents])
# query tab
with q:
st.write('')
st.markdown('<p class="prompt-font">Prompt:</p>', unsafe_allow_html=True)
promptcontainer = st.container()
user_input = promptcontainer.text_area("", height=50, on_change=None)
container2 = st.container()
if (user_input == ''):
container2.write('')
container2.write('')
else:
# TO-DO:
# call api to get value (this would be added to value put into the dict (prob at first pos. hopefully its a set num of chars))
# Verify Login **
if 'userdata' not in st.session_state.keys():
st.info('Not Logged In: Redirecting to Login page...')
time.sleep(.5)
util.switch_page('app')
else:
res = requests.post("https://sandl-backend-ny3lmzb4dq-uc.a.run.app/get_intent", json={"jwt": st.session_state['userdata'], "prompt": user_input})
value = round(res.json()["certaintyValue"] * 100)
if (value < 10):
value = '0' + str(value)
else:
value = str(value)
# visualization (loading bar)
container2.markdown("<p class='percent-font'>" + value + "%</p>", unsafe_allow_html=True)
my_bar = container2.progress(0, text='')
for percent_complete in range(int(value)):
time.sleep(0.01)
my_bar.progress(percent_complete + 1, text='')
container2.text('Your prompt has a ' + value + "% chance to be non-malicious.")
container2.text('')
container2.text('')
container2.text("SAnDL: Sophisticated Anomaly Detection for LLMs")
now = datetime.now()
current_time = now.strftime("%H:%M:%S")
if (user_input != ''):
st.session_state[value + user_input + " " + current_time] = value + user_input + " " + current_time
# Searches Tab:
with searches:
st.write('')
x = st.header('Recent Searches:', divider='grey', anchor=False)
containerx = st.container()
containerx.write('')
containerx.write('')
# sorts dict by reverse time stamp
t = dict(sorted(st.session_state.items(), key=lambda item: item[1][len(item[1]) - 8:], reverse = True))
for key in t.keys():
if (key == st.session_state[key]):
# everything besides the time stamp
tempstring = st.session_state[key][2:len(st.session_state[key]) - 8]
percent = st.session_state[key][:2]
# adds ... if tempstring too long
if len(tempstring) > 110:
tempstring = st.session_state[key][2:112] + "..."
containerx.text(tempstring + ' — ' + percent + '%')
|