Spaces:
Sleeping
Sleeping
| 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 + '%') | |