czyoung commited on
Commit
9c31ce8
·
verified ·
1 Parent(s): 9c2ec10

Documentation and comment update

Browse files
Files changed (1) hide show
  1. state.py +46 -1
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
- """Compute and store all summary DataFrames for inFileName."""
 
 
 
 
 
 
 
 
 
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