mj-new commited on
Commit
423e823
1 Parent(s): c378394

Added promptset in helpers file and iterative prompt retrieval

Browse files
Files changed (2) hide show
  1. app.py +8 -5
  2. helpers.py +32 -0
app.py CHANGED
@@ -46,9 +46,8 @@ def get_unique_name():
46
 
47
  def get_prompts(domain, type, size, language_code):
48
  print(f"Retrieving prompts for domain {domain} with method: {type} for language_code {language_code} of size {size}")
49
- promptset = ["test1", "test2"]
50
- first_prompt = promptset[0]
51
- return(promptset, first_prompt)
52
 
53
  def save_recording_and_meta(project_name, recording, transcript, language_code, spk_age, spk_accent, spk_city, spk_gender, spk_nativity, promptset, prompt_number):
54
  #, name, age, gender):
@@ -117,8 +116,12 @@ def save_recording_and_meta(project_name, recording, transcript, language_code,
117
  )
118
 
119
  output = print(f"Recording {audio_fn} and meta file {meta_fn} successfully saved to repo!")
 
120
  # None resets the audio component
121
- return ["Next prompt", 1, None]
 
 
 
122
 
123
 
124
  def whisper_model_change(radio_whisper_model):
@@ -355,7 +358,7 @@ with block:
355
 
356
  with gr.Accordion(label="Promptset settings"):
357
  radio_prompts_domain = gr.Dropdown(["Bridge"], label="Select promptset domain", info="")
358
- radio_promptset_type = gr.Radio(["New promptset generation", "Existing promptset use"], label="Language", value ="Existing promptset use", info="New promptset is generated using ChatGPT")
359
  var_promptset_size = gr.Textbox(label="Specify number of prompts (min 10, max 200)")
360
  button_get_prompts = gr.Button("Save settings and get first prompt to record")
361
 
 
46
 
47
  def get_prompts(domain, type, size, language_code):
48
  print(f"Retrieving prompts for domain {domain} with method: {type} for language_code {language_code} of size {size}")
49
+
50
+ return(promptset[domain], promptset[domain][0])
 
51
 
52
  def save_recording_and_meta(project_name, recording, transcript, language_code, spk_age, spk_accent, spk_city, spk_gender, spk_nativity, promptset, prompt_number):
53
  #, name, age, gender):
 
116
  )
117
 
118
  output = print(f"Recording {audio_fn} and meta file {meta_fn} successfully saved to repo!")
119
+
120
  # None resets the audio component
121
+ prompt_number = prompt_number + 1
122
+ prompt = promptset[prompt_number]
123
+
124
+ return [prompt, prompt_number, None]
125
 
126
 
127
  def whisper_model_change(radio_whisper_model):
 
358
 
359
  with gr.Accordion(label="Promptset settings"):
360
  radio_prompts_domain = gr.Dropdown(["Bridge"], label="Select promptset domain", info="")
361
+ radio_promptset_type = gr.Radio(["New promptset generation", "Existing promptset use"], label="Language", value ="Existing promptset use", info="New promptset is generated using. Requires providing open AI key in general settings tab")
362
  var_promptset_size = gr.Textbox(label="Specify number of prompts (min 10, max 200)")
363
  button_get_prompts = gr.Button("Save settings and get first prompt to record")
364
 
helpers.py CHANGED
@@ -45,3 +45,35 @@ dict_origin = {
45
  ]
46
  }
47
  }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
45
  ]
46
  }
47
  }
48
+ dict_promptset={
49
+ "bridge":[
50
+ "Licytacja pozwala okre艣li膰, jaka gra zostanie rozegrana w danym rozdaniu.",
51
+ "Kiedy twoja para jest w defensywie, wa偶ne jest, aby rozumie膰 sygna艂y dawane przez partnera.",
52
+ "Kontra pokazuje pewne zasoby punktowe i/lub okre艣lone kolory w r臋ce gracza.",
53
+ "Rekontra zwi臋ksza wynik je艣li przeciwnik zostanie skontraktowany.",
54
+ "Manche to poziom licytacji, na kt贸rym mo偶na zdoby膰 co najmniej 100 punkt贸w za poni偶ej linii.",
55
+ "Slem to licytacja na poziomie 6, kt贸ry wymaga wzi臋cia wszystkich lew opr贸cz jednej.",
56
+ "Wielki slem to licytacja na poziomie 7, gdzie musisz wzi膮膰 wszystkie 13 lew.",
57
+ "Finesse to technika, dzi臋ki kt贸rej mo偶esz zdoby膰 lew臋, mimo 偶e przeciwnik ma wy偶sz膮 kart臋.",
58
+ "Ci膮g licytacyjny sk艂ada si臋 z kilku kolejnych deklaracji licytacyjnych.",
59
+ "W bryd偶u sportowym wa偶na jest komunikacja niewerbalna, dlatego nie wolno u偶ywa膰 偶adnych znak贸w ani gest贸w.",
60
+ "Je艣li omy艂kowo zadeklarujesz niew艂a艣ciwy kontrakt, mo偶esz poprosi膰 o poprawienie b艂臋du przed wyj艣ciem.",
61
+ "Deklarant pr贸buje zdoby膰 zadeklarowan膮 liczb臋 lew, podczas gdy obro艅cy staraj膮 si臋 to uniemo偶liwi膰.",
62
+ "Je艣li grasz w kolorze trefl, twoim zadaniem jest zdoby膰 jak najwi臋cej lew w tym kolorze.",
63
+ "As to najwy偶sza karta w ka偶dym kolorze.",
64
+ "Zawist膮 nazywamy pierwsz膮 kart臋 wychodzon膮 przez obro艅c臋.",
65
+ "Balansuj膮ca licytacja mo偶e pom贸c twojej parze wej艣膰 do gry, kiedy przeciwnicy s膮 blisko kontraktu.",
66
+ "Nie wszyscy gracze stosuj膮 ten sam system licytacyjny.",
67
+ "W bryd偶u wa偶ne jest, aby pami臋ta膰 o kolejno艣ci kart i taktyce ich wyk艂adania.",
68
+ "Zapadka to sytuacja, gdy kt贸ry艣 z graczy jest zmuszony do zagrania karty, kt贸ra przyniesie korzy艣膰 przeciwnikowi.",
69
+ "Pami臋taj, aby zawsze pod膮偶a膰 kolorem je艣li masz kart臋 w tym kolorze.",
70
+ "Zabawa w bryd偶a wymaga koncentracji, taktyki i dobrej komunikacji z partnerem.",
71
+ "Ka偶dy gracz ma 13 kart w r臋ku podczas rozgrywki.",
72
+ "Otwarcie to pierwsza licytacja w rozdaniu.",
73
+ "Niekt贸re systemy licytacyjne zawieraj膮 szczeg贸艂owe konwencje, kt贸re precyzyjnie opisuj膮 si艂臋 i kolor r臋ki.",
74
+ "W bryd偶u impasem nazywamy sytuacj臋, gdy mamy jedn膮 kart臋 ni偶sz膮 od karty przeciwnika w tym samym kolorze i mamy szans臋 wzi膮膰 lew臋, je艣li ten kolor zostanie wychodzony z odpowiedniej strony.",
75
+ "Czasami kontrakt jest niewykonalny i celem staje si臋 minimalizacja strat.",
76
+ "Gdy grasz kontrakt bez atu, trumfem staje si臋 kolor kier.",
77
+ "Niekt贸re pary u偶ywaj膮 specjalnych system贸w sygna艂贸w, aby przekazywa膰 sobie dodatkowe informacje podczas obro艅y."
78
+ ]
79
+ }