reab5555 commited on
Commit
aebcc3f
·
verified ·
1 Parent(s): d81e882

Update visualization.py

Browse files
Files changed (1) hide show
  1. visualization.py +6 -4
visualization.py CHANGED
@@ -17,8 +17,10 @@ def extract_data_and_explanation(text):
17
  key, value = line.split(':', 1)
18
  key = key.strip()
19
  value = value.strip()
20
- if key in ["Self", "Others", "Anxiety", "Avoidance"]:
21
- speakers_data[current_speaker][key] = float(value.split('|')[0].strip())
 
 
22
  else:
23
  try:
24
  speakers_data[current_speaker][key] = float(value)
@@ -43,8 +45,8 @@ def create_bar_chart(data, title):
43
 
44
  def create_radar_chart(data, title):
45
  fig = go.Figure(data=go.Scatterpolar(
46
- r=[data.get('Anxiety', 0), data.get('Avoidance', 0), data.get('Self', 0), data.get('Others', 0)],
47
- theta=['Anxiety', 'Avoidance', 'Self', 'Others'],
48
  fill='toself'
49
  ))
50
  fig.update_layout(
 
17
  key, value = line.split(':', 1)
18
  key = key.strip()
19
  value = value.strip()
20
+ if '|' in value:
21
+ for part in value.split('|'):
22
+ sub_key, sub_value = part.split(':') if ':' in part else part.split()
23
+ speakers_data[current_speaker][sub_key.strip()] = float(sub_value.strip())
24
  else:
25
  try:
26
  speakers_data[current_speaker][key] = float(value)
 
45
 
46
  def create_radar_chart(data, title):
47
  fig = go.Figure(data=go.Scatterpolar(
48
+ r=[data.get('Self', 0), data.get('Avoidance', 0), data.get('Others', 0), data.get('Anxiety', 0)],
49
+ theta=['Self', 'Avoidance', 'Others', 'Anxiety'],
50
  fill='toself'
51
  ))
52
  fig.update_layout(