georad commited on
Commit
21164af
Β·
verified Β·
1 Parent(s): f464139

Delete app_test_toolbar_demo2.py

Browse files
Files changed (1) hide show
  1. 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}")