0.40 system prompts

app.py

@@ -18,6 +18,13 @@ HF_TOKEN = os.environ.get("HF_TOKEN", None)
 login(token=HF_TOKEN)
 
 
+system_prompts = {
+    "English": "You are a helpful chatbot that adheres to the prompted request.",
+    "German": "Du bist ein hilfreicher Chatbot, der die gestellte Anfrage befolgt.",
+    "French": "Vous êtes un chatbot utile qui respecte la demande soumise.",
+    "Spanish": "Eres un chatbot útil que sigue la solicitud proporcionada."
+}
+
 model_info = [{"id": "NousResearch/Meta-Llama-3.1-8B-Instruct",
                     "name": "Meta Llama 3.1 8B Instruct"},
                     {"id": "mistralai/Mistral-7B-Instruct-v0.3",
@@ -202,7 +209,18 @@ with gr.Blocks() as demo:
         with gr.Column():
             gr.HTML("<center><h1>🤖le Royale</h1></center>")
             gr.Markdown(arena_notes)
-            system_prompt = gr.Textbox(lines=1, label="System Prompt", value="You are a helpful chatbot that adheres to the prompted request.", show_copy_button=True)
+
+            language_dropdown = gr.Dropdown(
+                choices=["English", "German", "French", "Spanish"], 
+                label="Select Language for System Prompt",
+                value="English"
+            )
+            system_prompt = gr.Textbox(
+                lines=1, 
+                label="System Prompt", 
+                value=system_prompts["English"], 
+                show_copy_button=True
+            )
             with gr.Row(variant="panel"):
                 with gr.Column(scale=1):
                     submit_btn = gr.Button(value="Generate", variant="primary")
@@ -221,6 +239,13 @@ with gr.Blocks() as demo:
                 top_p = gr.Slider(minimum=0.0, maximum=1.0, value=1.0, label="Top-p", step=0.01)
                 repetition_penalty = gr.Slider(minimum=0.1, maximum=2.0, value=1.1, label="Repetition Penalty", step=0.1)
 
+
+        language_dropdown.change(
+            lambda lang: system_prompts[lang], 
+            inputs=[language_dropdown], 
+            outputs=[system_prompt]
+        )
+
         better_bot.select(reveal_bot, inputs=[better_bot, chatbot_a, chatbot_b], outputs=[chatbot_a, chatbot_b]) #fckp outputs=[chatbot_a, chatbot_b]
         input_text.submit(generate_both, inputs=[system_prompt, input_text, chatbot_a, chatbot_b, max_new_tokens, temperature, top_p, repetition_penalty], outputs=[chatbot_a, chatbot_b])
         submit_btn.click(generate_both, inputs=[system_prompt, input_text, chatbot_a, chatbot_b, max_new_tokens, temperature, top_p, repetition_penalty], outputs=[chatbot_a, chatbot_b])