initial

app.py

@@ -0,0 +1,93 @@
+# source .venv/bin/activate
+# streamlit run <>.py
+
+import streamlit as st
+import app_utils as util
+from pathlib import Path
+from PIL import Image
+from io import BytesIO
+import requests
+import time
+
+st.set_page_config(page_title="SAnDL", page_icon="🩴", layout="wide", initial_sidebar_state="collapsed")
+
+# Example URL
+image_url = "https://media.discordapp.net/attachments/1013678935231442944/1175575324210249740/sandll2.png?ex=656bbad6&is=655945d6&hm=d8f4a93a7c3ae90978f820093e2e7c17c18b6125546255afaaeca8ed6716bd4d&=&width=2268&height=486"
+
+# Open image from URL
+response = requests.get(image_url)
+img = Image.open(BytesIO(response.content))
+
+# close taskbar, page margin, ...
+st.markdown("""
+<style>
+    .st-emotion-cache-1yeub2j {
+      display: none;
+    }
+    .st-emotion-cache-79elbk {
+      margin-top: -45px;
+    }
+    .slogan-font {
+      font-size:18px !important;
+      margin-bottom: 40px;
+      margin-top: 0px;
+      text-align: center;
+    }
+    .bolded {
+      font-weight: 1000;   
+    }
+    [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;
+    }
+    section[data-testid="stSidebar"][aria-expanded="true"]{
+      display: none;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+logincol1, logincol2, logincol3 = st.columns([1, 8, 1])
+
+with logincol2:
+    
+    # logo
+    st.image(img)
+    # caption
+    st.markdown('<p class="slogan-font">' + "Developing <span class = \"bolded\">anomaly detection</span> systems for filtration of <span class = \"bolded\">large language model</span> prompts." + '</p>', unsafe_allow_html=True)
+
+    #login form
+    profilecontainer = st.container()
+    login_form = profilecontainer.form('Login')
+    username = login_form.text_input('Username').lower()
+    st.session_state['username'] = username
+    password = login_form.text_input('Password', type='password')
+    if login_form.form_submit_button('Login'):
+        st.session_state.pop('FormSubmitter:Login-Login', None)
+
+        # deletes all keys from previous logins
+        for key in st.session_state.keys():
+            if (key != 'username'):
+                st.session_state.pop(key)
+
+        res = requests.post("https://sandl-backend-ny3lmzb4dq-uc.a.run.app/login", json={"email": username, "password": password})
+        error = res.status_code
+
+        # login validation
+        if error == 200:
+          st.session_state['userdata'] = res.json()["jwt"]
+          st.success('Login Successful!', icon="✅")
+          util.switch_page("query")
+        # username exists in db, but password is incorrect
+        elif error == 400:
+          st.error('Incorrect Password.', icon="🚨")
+        # makes new account
+        elif error == 500: 
+          st.info('Account Not Found: New account created.', icon="ℹ️")
+          res = requests.post("https://sandl-backend-ny3lmzb4dq-uc.a.run.app/sign_up", json={"email": username, "password": password})
+          res = requests.post("https://sandl-backend-ny3lmzb4dq-uc.a.run.app/login", json={"email": username, "password": password})
+          st.session_state['userdata'] = res.json()["jwt"]
+          requests.post("https://sandl-backend-ny3lmzb4dq-uc.a.run.app/add_api_key", json={"jwt": st.session_state['userdata']})
+          time.sleep(.5)
+          util.switch_page("query")

app_utils.py

@@ -0,0 +1,24 @@
+# switch page util
+def switch_page(page_name: str):
+    from streamlit.runtime.scriptrunner import RerunData, RerunException
+    from streamlit.source_util import get_pages
+
+    def standardize_name(name: str) -> str:
+        return name.lower().replace("_", " ")
+
+    page_name = standardize_name(page_name)
+
+    pages = get_pages("app.py")
+
+    for page_hash, config in pages.items():
+        if standardize_name(config["page_name"]) == page_name:
+            raise RerunException(
+                RerunData(
+                    page_script_hash=page_hash,
+                    page_name=page_name,
+                )
+            )
+
+    page_names = [standardize_name(config["page_name"]) for config in pages.values()]
+
+    raise ValueError(f"Could not find page {page_name}. Must be one of {page_names}")

pages/1_Profile.py

@@ -0,0 +1,72 @@
+import requests
+import streamlit as st;
+import app_utils as util
+
+st.set_page_config(page_title="sandl", page_icon="🩴", layout="wide", initial_sidebar_state="collapsed")
+
+# goes to login page if not logged in
+try:
+  set = st.session_state['username']
+except:
+  util.switch_page('app')
+#        position: relative;
+
+st.markdown("""
+  <style>
+    .st-emotion-cache-79elbk {
+      margin-top: -65px;
+    } 
+    .st-emotion-cache-1wmy9hl {
+      margin-top: -32px;
+    }
+            
+    .circle-image {
+        width: 200px;
+        height: 200px;
+        border-radius: 25px;
+        overflow: hidden;
+        box-shadow: 0 0 5px rgba(0, 0, 0, 0.3);
+    }
+    
+    .circle-image img {
+        width: 100%;
+        height: 100%;
+        object-fit: cover;
+    }
+    [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)
+
+profilecol1, profilecol2, profilecol3 = st.columns([1, 8, 1])
+
+with profilecol1:
+  # goes to prompt page
+  if st.button('◀  Back'):
+    util.switch_page('query')
+
+with profilecol2:
+  # pfp
+  st.write('')
+  st.markdown("<div class=\"circle-image\"><img src=\"https://img.pikbest.com/png-images/qiantu/blue-hand-drawn-rounded-socks-cartoon-icon_2687993.png!sw800\"></div>", unsafe_allow_html=True)
+
+  st.header(st.session_state['username'], divider='gray', anchor=False)
+
+  st.write("")
+  if st.button('Show API Key'):
+    # gets the api key
+    res = requests.post("https://sandl-backend-ny3lmzb4dq-uc.a.run.app/get_api_keys", json={"jwt": st.session_state['userdata']})
+    # # test code
+    # print(res.status_code)
+    apikey = res.json()["keys"][0]
+    
+    st.text(apikey)
+  
+  st.write('')
+  if st.button('Logout'):
+    util.switch_page('app')
+  

pages/2_Query.py

@@ -0,0 +1,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 + '%')
+