Update utils.py
Browse files
utils.py
CHANGED
@@ -9,16 +9,16 @@ import os
|
|
9 |
from huggingface_hub import CommitScheduler
|
10 |
from functools import partial
|
11 |
|
|
|
12 |
def enable_buttons_side_by_side():
|
13 |
return tuple(gr.update(visible=True, interactive=True) for i in range(6))
|
14 |
|
15 |
def disable_buttons_side_by_side():
|
16 |
return tuple(gr.update(visible=i>=4, interactive=False) for i in range(6))
|
17 |
|
18 |
-
def get_conv_log_filename():
|
19 |
-
LOG_FILENAME = 'data/log.json'
|
20 |
-
return LOG_FILENAME
|
21 |
|
|
|
|
|
22 |
enable_btn = gr.update(interactive=True, visible=True)
|
23 |
disable_btn = gr.update(interactive=False)
|
24 |
invisible_btn = gr.update(interactive=False, visible=False)
|
@@ -30,7 +30,7 @@ TOKEN = os.getenv('TOKEN')
|
|
30 |
scheduler = CommitScheduler(
|
31 |
repo_id= DS_ID,
|
32 |
repo_type="dataset",
|
33 |
-
folder_path= os.path.dirname(
|
34 |
path_in_repo="data",
|
35 |
token = TOKEN,
|
36 |
every = 10,
|
@@ -39,7 +39,7 @@ scheduler = CommitScheduler(
|
|
39 |
|
40 |
def vote_last_response(state, vote_type, request: gr.Request):
|
41 |
with scheduler.lock:
|
42 |
-
with open(
|
43 |
data = {
|
44 |
"tstamp": round(time.time(), 4),
|
45 |
"type": vote_type,
|
@@ -75,7 +75,7 @@ def get_ip(request: gr.Request):
|
|
75 |
|
76 |
|
77 |
def get_song():
|
78 |
-
audio_a = "
|
79 |
state = AudioStateIG("Random_Model")
|
80 |
return state, audio_a
|
81 |
|
@@ -88,8 +88,8 @@ def generate_songs(state0, state1):
|
|
88 |
|
89 |
def random_sample_button(prompt):
|
90 |
|
91 |
-
audio_a = "
|
92 |
-
audio_b = "
|
93 |
return audio_a, audio_b
|
94 |
|
95 |
def leftvote_last_response(
|
|
|
9 |
from huggingface_hub import CommitScheduler
|
10 |
from functools import partial
|
11 |
|
12 |
+
|
13 |
def enable_buttons_side_by_side():
|
14 |
return tuple(gr.update(visible=True, interactive=True) for i in range(6))
|
15 |
|
16 |
def disable_buttons_side_by_side():
|
17 |
return tuple(gr.update(visible=i>=4, interactive=False) for i in range(6))
|
18 |
|
|
|
|
|
|
|
19 |
|
20 |
+
os.makedirs('data', exist_ok = True)
|
21 |
+
LOG_FILENAME = os.path.join('data', f'log_{datetime.now().isoformat()}.json')
|
22 |
enable_btn = gr.update(interactive=True, visible=True)
|
23 |
disable_btn = gr.update(interactive=False)
|
24 |
invisible_btn = gr.update(interactive=False, visible=False)
|
|
|
30 |
scheduler = CommitScheduler(
|
31 |
repo_id= DS_ID,
|
32 |
repo_type="dataset",
|
33 |
+
folder_path= os.path.dirname(LOG_FILENAME),
|
34 |
path_in_repo="data",
|
35 |
token = TOKEN,
|
36 |
every = 10,
|
|
|
39 |
|
40 |
def vote_last_response(state, vote_type, request: gr.Request):
|
41 |
with scheduler.lock:
|
42 |
+
with open(LOG_FILENAME, "a") as fout:
|
43 |
data = {
|
44 |
"tstamp": round(time.time(), 4),
|
45 |
"type": vote_type,
|
|
|
75 |
|
76 |
|
77 |
def get_song():
|
78 |
+
audio_a = "marine.mp3"
|
79 |
state = AudioStateIG("Random_Model")
|
80 |
return state, audio_a
|
81 |
|
|
|
88 |
|
89 |
def random_sample_button(prompt):
|
90 |
|
91 |
+
audio_a = "marine.mp3"
|
92 |
+
audio_b = "marine.mp3"
|
93 |
return audio_a, audio_b
|
94 |
|
95 |
def leftvote_last_response(
|