Add max input limit

app.py

@@ -5,7 +5,6 @@ import re
 import pandas as pd
 from llm_guard.input_scanners import PromptInjection
 import streamlit as st
-from streamlit_extras.stylable_container import stylable_container
 
 import config
 import utils
@@ -69,6 +68,7 @@ for idx, level in enumerate(config.LEVELS):
                         label_visibility="visible",
                         height=200,
                         placeholder="Your prompt",
+                        max_chars=config.MAX_INPUT_CHARS,
                     )
                     btn_submit_prompt = st.button(
                         "Send prompt", key=f"submit_prompt_{level}"
@@ -453,7 +453,6 @@ for idx, level in enumerate(config.LEVELS):
                     info_cont.write(config.LEVEL_DESCRIPTIONS[level]["explanation"])
                     info_cont.write("##### Real-life usage")
                     info_cont.write(config.LEVEL_DESCRIPTIONS[level]["real_life"])
-                    # info_cont.write("##### Benefits and drawbacks")
                     df = pd.DataFrame(
                         {
                             "Benefits": [config.LEVEL_DESCRIPTIONS[level]["benefits"]],
@@ -528,7 +527,6 @@ with st.expander("🏆 Record", expanded=True):
         )
 
     # show as pandas dataframe
-    # st.table(
     st.markdown(
         pd.DataFrame(
             table_data,
@@ -538,10 +536,6 @@ with st.expander("🏆 Record", expanded=True):
                 "Prompt tries",
                 "Secret guesses",
                 "Hint used",
-                # "Used hint 1",
-                # "Used hint 2",
-                # "Used hint 3",
-                # "Used info",
                 "Solved",
                 "Secret",
                 "Mitigation",
@@ -557,12 +551,5 @@ with st.expander("🏆 Record", expanded=True):
     )
 
 # TODOS:
-# - use Gemini-Pro-Flash for supervisor LLM
-# - story telling --> new field of study hard to be 100 percentage save
-# - switch to azure deployment --> currently not working under "GPT-4o"
 # - mark the user input with color in prompt
-# benefits and drawbacks, real world example
 # TODO: https://docs.streamlit.io/develop/api-reference/caching-and-state/st.cache_resource
-# Upgrade to bigger CPU
-
-

config.py

@@ -1,12 +1,13 @@
 ORQ_DEPLOYMENT_NAME = "llm-security-challenge-demo"
+MAX_INPUT_CHARS = 600
 
 # define which levels are shown and in which order
 LEVELS = [
     "no_mitigation",
     "basic_guardrails",
     "special_characters",
-    "llm_judge_input",
     "chain_of_thought",
+    "llm_judge_input",
     "basic_check_secret_in_output",
     "advanced_check_secret_in_output",
     "llm_judge_output",