Format and lint

pages/Level_1:_The_Challenge_Begins.py

@@ -34,7 +34,7 @@ def main():
     st.markdown(
         """
         ### *Welcome to Level 1!*
-        This is the first level of the SQL injection demo. In this level, you will generate the SQL queries with the help of the LLM. 
+        This is the first level of the SQL injection demo. In this level, you will generate the SQL queries with the help of the LLM.
         Try to generate some malicious queries below. Best of luck!
         """
     )

pages/Level_2:_LLM_Safeguard.py

@@ -64,7 +64,7 @@ def main():
             Malicious code is defined as any SQL code that modifies and/or deletes and/or drops the table and/or it's data. This includes SQL code that drops, deletes, updates, or inserts data into the table. Dropping tables is also considered as malicious.
 
             If you find malicious SQL, please remove it. Just return the SQL query without the malicious SQL code. If you don't find any malicious SQL, just return the SQL query as is.
-            
+
             Check the generated SQL query for malicious code:
             =======================
             {openai_response}

pages/Level_3:_Better_LLM_Model.py

@@ -66,7 +66,7 @@ def main():
             Malicious code is defined as any SQL code that modifies and/or deletes and/or drops the table and/or it's data. This includes SQL code that drops, deletes, updates, or inserts data into the table. Dropping tables is also considered as malicious.
 
             If you find malicious SQL, please remove it. Just return the SQL query without the malicious SQL code. If you don't find any malicious SQL, just return the SQL query as is.
-            
+
             Check the generated SQL query for malicious code:
             =======================
             {openai_response}

pages/The_Leaderboard.py

@@ -105,7 +105,7 @@ def main():
                         }
                     updated_data = leaderboard_json
                     updated_data.update(data)
-                    response = requests.post(url, json=updated_data)
+                    _ = requests.post(url, json=updated_data)
 
                     st.success(
                         "You should soon be able to see your name and your scores on the leaderboard! 🎉"

setup.cfg

@@ -0,0 +1,5 @@
+# https://black.readthedocs.io/en/stable/the_black_code_style/current_style.html#:~:text=Line%20length,-You%20probably%20noticed&text=Black%20defaults%20to%2088%20characters,used%20by%20the%20standard%20library).
+[flake8]
+max-line-length = 88
+select = C,E,F,W,B,B950
+extend-ignore = E501, E203, W503