(feat) add prompt

app.py

@@ -219,6 +219,25 @@ class PromptArenaApp:
                 ],
             )
 
+            with gr.Row():
+                prompt_id_box = gr.Textbox(label="ID du prompt", interactive=True)
+                prompt_text_box = gr.Textbox(label="Texte du prompt", interactive=True)
+                save_btn = gr.Button("Ajouter le prompt", variant="secondary")
+
+            def save_prompt(prompt_id, prompt_text):
+                try:
+                    self.arena.add_prompt(prompt_id, prompt_text)
+                    return gr.update(value="Prompt ajouté avec succès !")
+                except Exception as e:
+                    return gr.update(value=f"Erreur: {str(e)}")
+
+            save_result = gr.Textbox(label="Résultat de l'ajout", interactive=False)
+            save_btn.click(
+                save_prompt,
+                inputs=[prompt_id_box, prompt_text_box],
+                outputs=save_result,
+            )
+
             gr.Row([progress_info, rankings_table])
 
         return app

arena.py

@@ -4,6 +4,7 @@ import json
 import os
 import datetime
 from typing import Dict, List, Tuple, Union
+import pandas as pd
 
 
 class PromptArena:
@@ -76,6 +77,9 @@ class PromptArena:
             prompt_text: Texte du prompt
         """
         self.prompts[prompt_id] = prompt_text
+        prompts = pd.read_csv("prompts.csv", header=None)
+        prompts.loc[len(prompts)] = [prompt_text]
+        prompts.to_csv("prompts.csv", index=False, header=False)
         if prompt_id not in self.ratings:
             self.ratings[prompt_id] = trueskill.Rating()
         self._save_ratings()