Davide Fiocco commited on
Commit
0188bb2
·
1 Parent(s): bca9ece

Try out a different approach for statefulness

Browse files
Files changed (1) hide show
  1. app.py +6 -12
app.py CHANGED
@@ -30,23 +30,17 @@ def load_engine() -> Pipeline:
30
 
31
  return nlp_qa
32
 
33
- syslog = SysLogHandler(address=(st.secrets["logging_address"], int(st.secrets["logging_port"])))
34
-
35
- @st.cache(hash_funcs={SysLogHandler: id})
36
- def load_logger(syslog: SysLogHandler) -> Logger:
37
- logger = logging.getLogger()
38
- logger.addHandler(syslog)
39
- logger.setLevel(logging.INFO)
40
-
41
- return logger
42
-
43
 
44
  with st.spinner(
45
  text="Sto preparando il necessario per rispondere alle tue domande personali..."
46
  ):
47
 
48
  engine = load_engine()
49
- logger = load_logger(syslog)
50
 
51
  st.title("Le risposte alle tue domande personali")
52
 
@@ -55,7 +49,7 @@ input = st.text_input("Scrivi una domanda in italiano e comparirà la risposta!"
55
  if input:
56
  try:
57
  context = get_context()
58
- logger.info(input)
59
  answer = get_answer(input, context, engine)
60
  st.subheader(answer)
61
  except:
 
30
 
31
  return nlp_qa
32
 
33
+ if 'syslog' not in st.session_state:
34
+ st.session_state['syslog'] = SysLogHandler(address=("st.secrets["logging_address"]", int(st.secrets["logging_port"])))
35
+ st.session_state['logger'] = logging.getLogger()
36
+ st.session_state['logger'].setLevel(logging.INFO)
37
+ st.session_state['logger'].addHandler(st.session_state['syslog'])
 
 
 
 
 
38
 
39
  with st.spinner(
40
  text="Sto preparando il necessario per rispondere alle tue domande personali..."
41
  ):
42
 
43
  engine = load_engine()
 
44
 
45
  st.title("Le risposte alle tue domande personali")
46
 
 
49
  if input:
50
  try:
51
  context = get_context()
52
+ st.session_state['logger'].info(input)
53
  answer = get_answer(input, context, engine)
54
  st.subheader(answer)
55
  except: