ofermend commited on
Commit
0fc680c
1 Parent(s): 57fd6e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -5,6 +5,10 @@ import os
5
  import streamlit as st
6
  from PIL import Image
7
 
 
 
 
 
8
 
9
  def launch_bot():
10
  def generate_response(question):
@@ -24,7 +28,7 @@ def launch_bot():
24
  'title': os.environ['title'],
25
  'description': os.environ['description'],
26
  'source_data_desc': os.environ['source_data_desc'],
27
- 'streaming': os.environ.get('streaming', False),
28
  'prompt_name': os.environ.get('prompt_name', None)
29
  })
30
  st.session_state.cfg = cfg
@@ -77,8 +81,9 @@ def launch_bot():
77
  with st.spinner("Thinking..."):
78
  response = generate_response(prompt)
79
  st.write(response)
80
- message = {"role": "assistant", "content": response}
81
- st.session_state.messages.append(message)
82
 
83
  if __name__ == "__main__":
84
- launch_bot()
 
 
5
  import streamlit as st
6
  from PIL import Image
7
 
8
+ def isTrue(x) -> bool:
9
+ if isinstance(x, bool):
10
+ return s
11
+ return x.strip().lower() == 'true'
12
 
13
  def launch_bot():
14
  def generate_response(question):
 
28
  'title': os.environ['title'],
29
  'description': os.environ['description'],
30
  'source_data_desc': os.environ['source_data_desc'],
31
+ 'streaming': isTrue(os.environ.get('streaming', False)),
32
  'prompt_name': os.environ.get('prompt_name', None)
33
  })
34
  st.session_state.cfg = cfg
 
81
  with st.spinner("Thinking..."):
82
  response = generate_response(prompt)
83
  st.write(response)
84
+ message = {"role": "assistant", "content": response}
85
+ st.session_state.messages.append(message)
86
 
87
  if __name__ == "__main__":
88
+ launch_bot()
89
+