Start on final touches

Introduction.py

@@ -45,9 +45,9 @@ def main():
         #### The levels
         Try to inject malicious SQL code to alter the SQL table, each level is more difficult than the previous one!
 
-        - **Level 0**: You generate the SQL queries with the help of the LLM.
-        - **Level 1**: The SQL queries are first checked by an LLM Safeguard, which detects and removes malicious SQL queries.
-        - **Level 2**: The only difference is that we are using a better LLM model, GPT-4, for the safeguard. Otherwise they are the same.
+        - **Level 1**: You generate the SQL queries with the help of the LLM.
+        - **Level 2**: The SQL queries are first checked by an LLM Safeguard, which detects and removes malicious SQL queries.
+        - **Level 3**: The only difference is that we are using a better LLM model, GPT-4, for the safeguard. Otherwise they are the same.
 
         Are you happy with your results? Submit the keys on the leaderboard to see how you compare to others!
         """

pages/Level_1:_The_Challenge_Begins.py

@@ -64,7 +64,7 @@ def main():
                     st.error(e)
             if success:
                 st.success(
-                    f"Congratulations! You have successfully altered the database and passed Level 1! Here's your key: `{os.environ.get('LEVEL_0_KEY')}`"
+                    f"Congratulations! You have successfully altered the database and passed Level 1! Here's your key: `{os.environ.get('LEVEL_1_KEY')}`"
                 )
 
 

pages/Level_2:_LLM_Safeguard.py

@@ -109,7 +109,7 @@ def main():
                     st.error(e)
             if success:
                 st.success(
-                    f"Congratulations! You have successfully altered the database and passed Level 2! Here's your key: `{os.environ.get('LEVEL_1_KEY')}`"
+                    f"Congratulations! You have successfully altered the database and passed Level 2! Here's your key: `{os.environ.get('LEVEL_2_KEY')}`"
                 )
             else:
                 st.success("Done!")

pages/Level_3:_Better_LLM_Model.py

@@ -113,7 +113,7 @@ def main():
                     st.error(e)
             if success:
                 st.success(
-                    f"Wow! Well done, you passed Level 3! Here's your key: `{os.getenv('LEVEL_2_KEY')}`"
+                    f"Wow! Well done, you passed Level 3! Here's your key: `{os.getenv('LEVEL_3_KEY')}`"
                 )
             else:
                 st.success("Done!")

pages/The_Leaderboard.py

@@ -37,18 +37,24 @@ def main():
         leaderboard_data = (
             pd.DataFrame(leaderboard_json)
             .T[["level 0", "level 1", "level 2"]]
-            .applymap(lambda x: "✅" if x else "❌")
-        )
-        leaderboard_data = leaderboard_data.rename(
-            columns={"level 0": "Level 0", "level 1": "Level 1", "level 2": "Level 2"}
-        )
-        leaderboard_data["Score"] = leaderboard_data.apply(
-            lambda x: x.value_counts().get("✅", 0) * 100, axis=1
+            .rename(
+                columns={
+                    "level 0": "Level 1",
+                    "level 1": "Level 2",
+                    "level 2": "Level 3",
+                },
+            )
+            .map(lambda x: "✅" if x else "❌")
+            .assign(
+                Score=lambda df: df.apply(
+                    lambda x: x.value_counts().get("✅", 0) * 100, axis=1
+                )
+            )
+            .sort_values(by="Score", ascending=False)
+            .reset_index()
+            .rename(columns={"index": "Name"})
         )
-        leaderboard_data = leaderboard_data.sort_values(by="Score", ascending=False)
-        leaderboard_data = leaderboard_data.reset_index()
-        leaderboard_data = leaderboard_data.rename(columns={"index": "Name"})
-        leaderboard_data.index += 1
+        # leaderboard_data.index += 1
         st.dataframe(leaderboard_data)
     else:
         st.error("An error occurred while fetching the leaderboard.")
@@ -77,30 +83,36 @@ def main():
                         data = {
                             display_name: {
                                 "email": email,
-                                "level 0": key == os.environ.get("LEVEL_0_KEY"),
                                 "level 1": key == os.environ.get("LEVEL_1_KEY"),
                                 "level 2": key == os.environ.get("LEVEL_2_KEY"),
+                                "level 3": key == os.environ.get("LEVEL_3_KEY"),
                             }
                         }
                     else:
                         data = {
                             display_name: {
                                 "email": email,
-                                "level 0": key == os.environ.get("LEVEL_0_KEY")
-                                or leaderboard_data[
-                                    leaderboard_data["Name"] == display_name
-                                ]["Level 0"].values[0]
-                                == "✅",
-                                "level 1": key == os.environ.get("LEVEL_1_KEY")
-                                or leaderboard_data[
-                                    leaderboard_data["Name"] == display_name
-                                ]["Level 1"].values[0]
-                                == "✅",
-                                "level 2": key == os.environ.get("LEVEL_2_KEY")
-                                or leaderboard_data[
-                                    leaderboard_data["Name"] == display_name
-                                ]["Level 2"].values[0]
-                                == "✅",
+                                "level 1": (
+                                    key == os.environ.get("LEVEL_1_KEY")
+                                    or leaderboard_data[
+                                        leaderboard_data["Name"] == display_name
+                                    ]["Level 1"].values[0]
+                                    == "✅"
+                                ),
+                                "level 2": (
+                                    key == os.environ.get("LEVEL_2_KEY")
+                                    or leaderboard_data[
+                                        leaderboard_data["Name"] == display_name
+                                    ]["Level 2"].values[0]
+                                    == "✅"
+                                ),
+                                "level 3": (
+                                    key == os.environ.get("LEVEL_3_KEY")
+                                    or leaderboard_data[
+                                        leaderboard_data["Name"] == display_name
+                                    ]["Level 3"].values[0]
+                                    == "✅"
+                                ),
                             }
                         }
                     updated_data = leaderboard_json