Update app.py

app.py

@@ -15,7 +15,7 @@ def format_prompt(message, history):
   return prompt
 
 def generate(
-    prompt, history, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
+    prompt, history, system_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
 ):
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -31,8 +31,7 @@ def generate(
         seed=42,
     )
 
-    formatted_prompt = format_prompt(prompt, history)
-
+    formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
 
@@ -43,6 +42,11 @@ def generate(
 
 
 additional_inputs=[
+    gr.Textbox(
+        label="System Prompt",
+        max_lines=1,
+        interactive=True,
+    ),
     gr.Slider(
         label="Temperature",
         value=0.9,