Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
81 |
-
|
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 |
+
|