Delete app_test_toolbar_demo2.py
Browse files- app_test_toolbar_demo2.py +0 -140
app_test_toolbar_demo2.py
DELETED
@@ -1,140 +0,0 @@
|
|
1 |
-
import streamlit as st
|
2 |
-
import torch
|
3 |
-
import uuid
|
4 |
-
import nest_asyncio
|
5 |
-
import asyncio
|
6 |
-
import os
|
7 |
-
#import omegaConf
|
8 |
-
|
9 |
-
torch.classes.__path__ = []
|
10 |
-
|
11 |
-
# Setup for HTTP API Calls
|
12 |
-
if 'device_id' not in st.session_state:
|
13 |
-
st.session_state.device_id = str(uuid.uuid4())
|
14 |
-
|
15 |
-
if "feedback_key" not in st.session_state:
|
16 |
-
st.session_state.feedback_key = 0
|
17 |
-
|
18 |
-
#corpus_keys = 'first.last'
|
19 |
-
cfg = {"title": "SBS MMMMMMMMAAAAAAAAAAAPPPPPPPPPPP"}
|
20 |
-
|
21 |
-
def launch_bot():
|
22 |
-
def reset():
|
23 |
-
for i in range(100):
|
24 |
-
st.write(f"This is scrollable content line {i}")
|
25 |
-
#st.session_state.messages = [{"role": "assistant", "content": "How may I help you?", "avatar": 'π€'}]
|
26 |
-
st.session_state.ex_prompt = None
|
27 |
-
st.session_state.first_turn = True
|
28 |
-
|
29 |
-
|
30 |
-
def generate_response(question):
|
31 |
-
response = vq.submit_query(question, languages[st.session_state.language])
|
32 |
-
return response
|
33 |
-
|
34 |
-
def generate_streaming_response(question):
|
35 |
-
response = vq.submit_query_streaming(question, languages[st.session_state.language])
|
36 |
-
return response
|
37 |
-
|
38 |
-
def show_example_questions():
|
39 |
-
if len(st.session_state.example_messages) > 0 and st.session_state.first_turn:
|
40 |
-
selected_example = pills("Questions to Try:", st.session_state.example_messages, index=None)
|
41 |
-
if selected_example:
|
42 |
-
st.session_state.ex_prompt = selected_example
|
43 |
-
st.session_state.first_turn = False
|
44 |
-
return True
|
45 |
-
return False
|
46 |
-
|
47 |
-
if 'cfg' not in st.session_state:
|
48 |
-
yield
|
49 |
-
#corpus_keys = str(os.environ["first, last"]).split(',')
|
50 |
-
#cfg = {title: "SBS MMMMMMMMAAAAAAAAAAAPPPPPPPPPPP",}
|
51 |
-
#cfg = "SBS MMMMAPPPPEEERR", # omegaConf.create({
|
52 |
-
# 'corpus_keys': corpus_keys,
|
53 |
-
# 'api_key': str(os.environ['api_key']),
|
54 |
-
# 'title': os.environ['title'],
|
55 |
-
# 'source_data_desc': os.environ['source_data_desc'],
|
56 |
-
# 'streaming': isTrue(os.environ.get('streaming', False)),
|
57 |
-
# 'prompt_name': os.environ.get('prompt_name', None),
|
58 |
-
# 'examples': os.environ.get('examples', None),
|
59 |
-
# 'language': 'English'
|
60 |
-
#})
|
61 |
-
#st.session_state.cfg = cfg
|
62 |
-
#st.session_state.ex_prompt = None
|
63 |
-
#st.session_state.first_turn = True
|
64 |
-
#st.session_state.language = cfg.language
|
65 |
-
#example_messages = [example.strip() for example in cfg.examples.split(",")]
|
66 |
-
#st.session_state.example_messages = [em for em in example_messages if len(em)>0][:max_examples]
|
67 |
-
|
68 |
-
#st.session_state.vq = VectaraQuery(cfg.api_key, cfg.corpus_keys, cfg.prompt_name)
|
69 |
-
|
70 |
-
#cfg = st.session_state.cfg
|
71 |
-
#vq = st.session_state.vq
|
72 |
-
st.set_page_config(page_title="SSSSSSSSBBBBBBBBBSSSSSSSSSSS", layout="wide")
|
73 |
-
|
74 |
-
# left side content
|
75 |
-
with st.sidebar:
|
76 |
-
#image = Image.open('Vectara-logo.png')
|
77 |
-
#st.image(image, width=175)
|
78 |
-
st.markdown(f"## About\n\n")
|
79 |
-
#f"This demo uses outside RAG to ask questions about {cfg.source_data_desc}\n")
|
80 |
-
|
81 |
-
#cfg.language = st.selectbox('Language:', languages.keys())
|
82 |
-
#if st.session_state.language != cfg.language:
|
83 |
-
# st.session_state.language = cfg.language
|
84 |
-
# reset()
|
85 |
-
# st.rerun()
|
86 |
-
|
87 |
-
st.markdown("\n")
|
88 |
-
bc1, _ = st.columns([1, 1])
|
89 |
-
with bc1:
|
90 |
-
if st.button('Start Over'):
|
91 |
-
reset()
|
92 |
-
st.rerun()
|
93 |
-
|
94 |
-
st.markdown("---")
|
95 |
-
st.markdown(
|
96 |
-
"## Temporary test demo only\n"
|
97 |
-
)
|
98 |
-
|
99 |
-
st.markdown(f"<center> <h2> Header Demo Test: {'SSSSSSSSSSSBBBBBBBBBBBBSSSSSSSSSS'} </h2> </center>", unsafe_allow_html=True)
|
100 |
-
|
101 |
-
if "messages" not in st.session_state.keys():
|
102 |
-
reset()
|
103 |
-
|
104 |
-
# Display chat messages
|
105 |
-
for message in st.session_state.messages:
|
106 |
-
with st.chat_message(message["role"], avatar=message["avatar"]):
|
107 |
-
st.write(message["content"])
|
108 |
-
|
109 |
-
example_container = st.empty()
|
110 |
-
with example_container:
|
111 |
-
if show_example_questions():
|
112 |
-
example_container.empty()
|
113 |
-
st.rerun()
|
114 |
-
|
115 |
-
# select prompt from example question or user provided input
|
116 |
-
if st.session_state.ex_prompt:
|
117 |
-
prompt = st.session_state.ex_prompt
|
118 |
-
else:
|
119 |
-
prompt = st.chat_input()
|
120 |
-
if prompt:
|
121 |
-
st.session_state.messages.append({"role": "user", "content": prompt, "avatar": 'π§βπ»'})
|
122 |
-
with st.chat_message("user", avatar="π§βπ»"):
|
123 |
-
st.write(prompt)
|
124 |
-
st.session_state.ex_prompt = None
|
125 |
-
|
126 |
-
if (st.session_state.messages[-1]["role"] == "assistant") & (st.session_state.messages[-1]["content"] != "How may I help you?"):
|
127 |
-
streamlit_feedback(feedback_type="thumbs", on_submit = thumbs_feedback, key = st.session_state.feedback_key,
|
128 |
-
kwargs = {"user_query": st.session_state.messages[-2]["content"],
|
129 |
-
"chat_response": st.session_state.messages[-1]["content"],
|
130 |
-
"demo_name": "SSSSSSSSSSSBBBBBBBBBBBBBBSSSSSSSSSSSS",
|
131 |
-
"response_language": st.session_state.language})
|
132 |
-
|
133 |
-
|
134 |
-
if __name__ == "__main__":
|
135 |
-
#st.set_page_config(page_title="Sticky toolbar test", layout="wide")
|
136 |
-
nest_asyncio.apply()
|
137 |
-
asyncio.run(launch_bot())
|
138 |
-
|
139 |
-
for i in range(100):
|
140 |
-
st.write(f"This is scrollable content line {i}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|