Update app.py

app.py

@@ -7,11 +7,11 @@ client = InferenceClient(
 	api_key= os.getenv("superSecretKey")
 )
 
-def evaluate_password_strength(password):
+def evaluate_code_security(code):
     messages = [
         {
             "role": "user",
-            "content": f"Rate the strength of the password: {password} as either 'Weak', 'Medium', or 'Strong'"
+            "content": f"You are BoxyReviewer. You work for OmniBlocks, a Python IDE. Your job is to review code before it is uploaded to the cloud. Not for bugs, that's the compiler's thing. Your job is to look at the code and make sure none of it is malicious before it gets to OmniBlocks' server. Your response must be formatted in a JSON format. The key 'result' should be either 'Safe' or 'Unsafe'. The key 'reason' should explain your reasoning. Here's the code {code}"
         }
     ]
 
@@ -25,10 +25,10 @@ def evaluate_password_strength(password):
     return response
 
 demo = gr.Interface(
-    evaluate_password_strength,
-    gr.Textbox(label="Enter your password"),
-    gr.Textbox(label="Password Strength Evaluation"),
-    title="Password Strength Evaluator",
+    evaluate_code_security,
+    gr.Textbox(label="Enter your code"),
+    gr.Textbox(label="Code Security Evaluation"),
+    title="Code Security Evaluator",
     description="Get the AI's evaluation of your password strength.",
 )