Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
Documentation and comment update
Browse files
state.py
CHANGED
|
@@ -29,6 +29,20 @@ import utils
|
|
| 29 |
verbosity = 4 # 0=None 1=Low 2=Medium 3=High 4=Debug
|
| 30 |
|
| 31 |
def printV(message, level):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
if verbosity >= level:
|
| 33 |
print(message)
|
| 34 |
|
|
@@ -147,19 +161,37 @@ def build_all_csv_zip():
|
|
| 147 |
# ---------------------------------------------------------------------------
|
| 148 |
|
| 149 |
def addCategory():
|
|
|
|
|
|
|
|
|
|
|
|
|
| 150 |
new = st.session_state.categoryInput.strip()
|
| 151 |
if not new:
|
| 152 |
return
|
|
|
|
| 153 |
st.toast(f"Adding {new}")
|
|
|
|
| 154 |
st.session_state.categories.append(new)
|
| 155 |
st.session_state.categorySelect.append([])
|
| 156 |
st.session_state.pop(f"multiselect_{new}", None)
|
|
|
|
| 157 |
st.session_state.categoryInput = ""
|
| 158 |
|
| 159 |
|
| 160 |
def removeCategory(index):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 161 |
name = st.session_state.categories[index]
|
|
|
|
| 162 |
st.toast(f"Removing {name}")
|
|
|
|
| 163 |
st.session_state.pop(f"multiselect_{name}", None)
|
| 164 |
del st.session_state.categories[index]
|
| 165 |
del st.session_state.categorySelect[index]
|
|
@@ -369,6 +401,9 @@ def on_grename_change(idx, token_display_map=None):
|
|
| 369 |
# ---------------------------------------------------------------------------
|
| 370 |
|
| 371 |
def updateMultiSelect():
|
|
|
|
|
|
|
|
|
|
| 372 |
fileName = st.session_state["select_currFile"]
|
| 373 |
st.session_state.resetResult = True
|
| 374 |
result = st.session_state.results.get(fileName)
|
|
@@ -610,11 +645,21 @@ def build_table_df(displayDF):
|
|
| 610 |
# ---------------------------------------------------------------------------
|
| 611 |
|
| 612 |
def analyze(inFileName):
|
| 613 |
-
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 614 |
try:
|
| 615 |
printV(f"Start analyzing {inFileName}", 4)
|
| 616 |
st.session_state.resetResult = False
|
| 617 |
|
|
|
|
| 618 |
if not (
|
| 619 |
inFileName in st.session_state.results
|
| 620 |
and inFileName in st.session_state.summaries
|
|
|
|
| 29 |
verbosity = 4 # 0=None 1=Low 2=Medium 3=High 4=Debug
|
| 30 |
|
| 31 |
def printV(message, level):
|
| 32 |
+
'''
|
| 33 |
+
Logging function to print based on verbosity level
|
| 34 |
+
|
| 35 |
+
...
|
| 36 |
+
|
| 37 |
+
Parameters
|
| 38 |
+
----------
|
| 39 |
+
message : str
|
| 40 |
+
Message to display
|
| 41 |
+
level : int
|
| 42 |
+
Target verbosity level of message; 0=None 1=Low 2=Medium 3=High 4=Debug
|
| 43 |
+
'''
|
| 44 |
+
# Read from global value above this function
|
| 45 |
+
global verbosity
|
| 46 |
if verbosity >= level:
|
| 47 |
print(message)
|
| 48 |
|
|
|
|
| 161 |
# ---------------------------------------------------------------------------
|
| 162 |
|
| 163 |
def addCategory():
|
| 164 |
+
'''
|
| 165 |
+
Create new category
|
| 166 |
+
'''
|
| 167 |
+
# Clean up input
|
| 168 |
new = st.session_state.categoryInput.strip()
|
| 169 |
if not new:
|
| 170 |
return
|
| 171 |
+
# Notify user
|
| 172 |
st.toast(f"Adding {new}")
|
| 173 |
+
# Update all relevant session_state variables
|
| 174 |
st.session_state.categories.append(new)
|
| 175 |
st.session_state.categorySelect.append([])
|
| 176 |
st.session_state.pop(f"multiselect_{new}", None)
|
| 177 |
+
# Reset category input field
|
| 178 |
st.session_state.categoryInput = ""
|
| 179 |
|
| 180 |
|
| 181 |
def removeCategory(index):
|
| 182 |
+
'''
|
| 183 |
+
Remove category
|
| 184 |
+
|
| 185 |
+
Parameters
|
| 186 |
+
----------
|
| 187 |
+
index : int
|
| 188 |
+
Index of category to be removed
|
| 189 |
+
'''
|
| 190 |
+
# Get name of category
|
| 191 |
name = st.session_state.categories[index]
|
| 192 |
+
# Notify user
|
| 193 |
st.toast(f"Removing {name}")
|
| 194 |
+
# Update all relevant session_state variables
|
| 195 |
st.session_state.pop(f"multiselect_{name}", None)
|
| 196 |
del st.session_state.categories[index]
|
| 197 |
del st.session_state.categorySelect[index]
|
|
|
|
| 401 |
# ---------------------------------------------------------------------------
|
| 402 |
|
| 403 |
def updateMultiSelect():
|
| 404 |
+
'''
|
| 405 |
+
Updates multi-select field
|
| 406 |
+
'''
|
| 407 |
fileName = st.session_state["select_currFile"]
|
| 408 |
st.session_state.resetResult = True
|
| 409 |
result = st.session_state.results.get(fileName)
|
|
|
|
| 645 |
# ---------------------------------------------------------------------------
|
| 646 |
|
| 647 |
def analyze(inFileName):
|
| 648 |
+
"""
|
| 649 |
+
Compute and store all summary DataFrames for inFileName.
|
| 650 |
+
|
| 651 |
+
Updates all session_state variables based on stored diarization results
|
| 652 |
+
|
| 653 |
+
Parameters
|
| 654 |
+
----------
|
| 655 |
+
inFileName : str
|
| 656 |
+
Name of file to analyze
|
| 657 |
+
"""
|
| 658 |
try:
|
| 659 |
printV(f"Start analyzing {inFileName}", 4)
|
| 660 |
st.session_state.resetResult = False
|
| 661 |
|
| 662 |
+
# If file has not been processed by Sonogram, then end
|
| 663 |
if not (
|
| 664 |
inFileName in st.session_state.results
|
| 665 |
and inFileName in st.session_state.summaries
|