Update app.py

app.py

@@ -75,51 +75,7 @@ def generate(
         outputs.append(text)
         yield "".join(outputs)
 
-
-chat_interface = gr.ChatInterface(
-    fn=generate,
-    chatbot=gr.Chatbot(height="480px"),
-    additional_inputs=[
-        gr.Textbox(label="System prompt", lines=4),
-        gr.Slider(
-            label="Max new tokens",
-            minimum=1,
-            maximum=MAX_MAX_NEW_TOKENS,
-            step=1,
-            value=DEFAULT_MAX_NEW_TOKENS,
-        ),
-        gr.Slider(
-            label="Temperature",
-            minimum=0.1,
-            maximum=4.0,
-            step=0.1,
-            value=0.6,
-        ),
-        gr.Slider(
-            label="Top-p (nucleus sampling)",
-            minimum=0.05,
-            maximum=1.0,
-            step=0.05,
-            value=0.9,
-        ),
-        gr.Slider(
-            label="Top-k",
-            minimum=1,
-            maximum=1000,
-            step=1,
-            value=50,
-        ),
-        gr.Slider(
-            label="Repetition penalty",
-            minimum=1.0,
-            maximum=2.0,
-            step=0.05,
-            value=1.2,
-        ),
-    ],
-    stop_btn=None,
-    examples=[
-        ["You are a senior software engineer who is best in the world at fixing vulnerabilities.
+example1='''You are a senior software engineer who is best in the world at fixing vulnerabilities.
 Users will give you vulnerable code and you will generate a fix based on the provided INSTRUCTION. 
 INSTRUCTION:
 Detected MD5 hash algorithm which is considered insecure. MD5 is not collision resistant and is therefore not suitable as a cryptographic signature. Use SHA256 or SHA3 instead.
@@ -130,8 +86,9 @@ def md5_hash(path):
      with open(path, "rb") as f:
          content = f.read()
      return hashlib.md5(content).hexdigest()
-"],
-        ["You are a software engineer who is best in the world at summarizing code changes.
+'''
+
+example2='''You are a software engineer who is best in the world at summarizing code changes.
 Carefullly analyze the given old code and new code and generate a summary of the changes.
 
 Old Code:
@@ -193,8 +150,9 @@ void processList() {
         printf("Accessing freed list: %d\n", head->next->data);
     }
 }
-"],
-        ["You are a senior security researcher who is best in the world at detecting vulnerabilities.
+'''
+
+example3='''You are a senior security researcher who is best in the world at detecting vulnerabilities.
 Carefully analyze the given code snippet and track the data flows from various sources to sinks.
 Assume that any call to an unknown external API is not sanitized.
 Respond only with either "YES" or "NO".
@@ -216,8 +174,9 @@ def get_data():
     logger.info("Received request with API Key: %s", api_key)  
     data = {"message": "Data processed"}
     return jsonify(data)
-"],
-        ["You are a senior software engineer who is best in the world at fixing vulnerabilities.
+'''
+
+example4='''You are a senior software engineer who is best in the world at fixing vulnerabilities.
 Users will give you vulnerable code and you will generate a fix based on the provided INSTRUCTION. 
 INSTRUCTION:
 Detected subprocess function 'run' with user controlled data. A malicious actor could leverage this to perform command injection. You may consider using 'shlex.escape()'.
@@ -249,8 +208,55 @@ def run(command, desc=None, errdesc=None, custom_env=None, live: bool = default_
             error_bits.append(f"stderr: {{result.stderr}}")
         raise RuntimeError("\n".join(error_bits))
     return (result.stdout or "")
-    
-"],
+'''
+
+chat_interface = gr.ChatInterface(
+    fn=generate,
+    chatbot=gr.Chatbot(height="480px"),
+    additional_inputs=[
+        gr.Textbox(label="System prompt", lines=4),
+        gr.Slider(
+            label="Max new tokens",
+            minimum=1,
+            maximum=MAX_MAX_NEW_TOKENS,
+            step=1,
+            value=DEFAULT_MAX_NEW_TOKENS,
+        ),
+        gr.Slider(
+            label="Temperature",
+            minimum=0.1,
+            maximum=4.0,
+            step=0.1,
+            value=0.6,
+        ),
+        gr.Slider(
+            label="Top-p (nucleus sampling)",
+            minimum=0.05,
+            maximum=1.0,
+            step=0.05,
+            value=0.9,
+        ),
+        gr.Slider(
+            label="Top-k",
+            minimum=1,
+            maximum=1000,
+            step=1,
+            value=50,
+        ),
+        gr.Slider(
+            label="Repetition penalty",
+            minimum=1.0,
+            maximum=2.0,
+            step=0.05,
+            value=1.2,
+        ),
+    ],
+    stop_btn=None,
+    examples=[
+        [example1],
+        [example2],
+        [example3],
+        [example4],
         ["You are a coding assitant, who is best in the world at debugging. Create a snake game in Python."],
     ],
 )