DPO feedback format

app.py

@@ -1,8 +1,8 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: enro-app.ipynb.
 
 # %% auto 0
-__all__ = ['model_name', 'tokenizer', 'model', 'example_translations', 'encoded_key', 'decoded_bytes', 'firebase_creds',
-           'get_translations', 'save_option_to_repo', 'get_random_sentence', 'update_prompt']
+__all__ = ['model_name', 'tokenizer', 'model', 'example_translations', 'sentences', 'encoded_key', 'decoded_bytes',
+           'firebase_creds', 'get_translations', 'save_option_to_repo', 'get_random_sentence', 'update_prompt']
 
 # %% enro-app.ipynb 1
 from transformers import MarianMTModel, MarianTokenizer
@@ -28,14 +28,58 @@ def get_translations(input_text):
     return random_translations[0], random_translations[1]
 
 # %% enro-app.ipynb 5
-example_translations = ["What is the way to get to the mosque?",
-"Practice makes perfect.",
-"The cat is eating the small mouse.",
-"I'm not good at learning foreign languages.",
-"What do you mean you don't know?",
-"I should have sent a letter of apology.",
-"His handwriting is poor.",
-"He bowed his head."]
+example_translations = sentences = [
+    "The mouse jumped.",
+    "The mouse ran away.",
+    "The mouse squeaked loudly.",
+    "The mouse hid quickly.",
+    "The mouse nibbled cheese.",
+    "The mouse dashed off.",
+    "The mouse found crumbs.",
+    "The mouse searched intently.",
+    "The mouse peered out.",
+    "The mouse sniffed carefully.",
+    "The mouse climbed up.",
+    "The mouse scurried along.",
+    "The mouse escaped swiftly.",
+    "The mouse burrowed deeply.",
+    "The mouse explored curiously.",
+    "That mouse squeaked.",
+    "That mouse dashed away.",
+    "That mouse looked around.",
+    "That mouse gnawed wood.",
+    "That mouse hid cleverly.",
+    "That mouse chewed through.",
+    "That mouse fled quickly.",
+    "That mouse found food.",
+    "That mouse evaded capture.",
+    "That mouse peeked out.",
+    "That mouse twitched nervously.",
+    "That mouse made noise.",
+    "That mouse crawled away.",
+    "That mouse gnawed plastic.",
+    "That mouse explored boldly.",
+    "Mice are fast.",
+    "Mice are clever.",
+    "Mice are quick.",
+    "Mice are tiny.",
+    "Mice are playful.",
+    "Mice are resourceful.",
+    "Mice are shy.",
+    "Mice are cute.",
+    "Mice live together.",
+    "Mice hide well.",
+    "Mice avoid predators.",
+    "Mice chew everything.",
+    "Mice burrow underground.",
+    "Mice love food.",
+    "Mice squeak often.",
+    "Mice find crumbs.",
+    "Mice travel far.",
+    "Mice reproduce rapidly.",
+    "Mice build nests.",
+    "Mice hide indoors."
+]
 
 # %% enro-app.ipynb 6
 import os
@@ -56,14 +100,20 @@ if not firebase_admin._apps:
 
 # %% enro-app.ipynb 7
 def save_option_to_repo(trans_prompt, translation1, translation2, button):
-    ref = db.reference('feedback')
+    ref = db.reference('dpo_feedback')
     # Push new data to the database
-    ref.push({
-        'prompt': trans_prompt,
-        'translation_1': translation1,
-        'translation_2': translation2,
-        'feedback': button
-    })
+    if button == "Translation 1":
+        ref.push({
+            'prompt': trans_prompt,
+            'chosen': translation1,
+            'rejected': translation2
+        })
+    if button == "Translation 2":
+        ref.push({
+            'prompt': trans_prompt,
+            'chosen': translation2,
+            'rejected': translation1
+        })
 
 def get_random_sentence():
     return random.sample(example_translations, 1)[0]