Update app.py

app.py

@@ -4,69 +4,35 @@ import streamlit as st
 import os
 from PIL import Image
 
-topics =  { 
-    'Standardized testing in education': '266',
-    'Ethical implications of genetic editing': '267',
-    'Social media and mental health': '268',
-    'discus': '45'
-}
-
 def launch_bot():
-    def generate_response(question, cfg):
+    def generate_response(question):
         response = vq.submit_query(question)
         return response
 
     def reset():
-        st.session_state.messages = [{"role": "assistant", "content": "Please make your opening statement.", "avatar": '🦖'}]
+        st.session_state.messages = [{"role": "assistant", "content": "Please ask your question about drink names.", "avatar": '🦖'}]
         st.session_state.vq = VectaraQuery(cfg.api_key, cfg.customer_id, cfg.corpus_id, cfg.prompt_name)
 
     if 'cfg' not in st.session_state:
         cfg = OmegaConf.create({
             'customer_id': str(os.environ['VECTARA_CUSTOMER_ID']),
-            'corpus_id': list(topics.values())[0],
+            'corpus_id': '45',  # Fixed corpus ID for drink names
             'api_key': str(os.environ['VECTARA_API_KEY']),
             'prompt_name': 'vectara-experimental-summary-ext-2023-12-11-large',
         })
         st.session_state.cfg = cfg
         st.session_state.vq = VectaraQuery(cfg.api_key, cfg.customer_id, cfg.corpus_id, cfg.prompt_name)
 
-        st.session_state.current_human_role = None
-        st.session_state.current_topic = None
-        st.session_state.style = None
-
     cfg = st.session_state.cfg
     vq = st.session_state.vq
-    st.set_page_config(page_title="Debate Bot", layout="wide")
-
+    st.set_page_config(page_title="Drink Name Query Bot", layout="wide")
 
-    # left side content
+    # Left side content
     with st.sidebar:
         image = Image.open('Vectara-logo.png')
         st.image(image, width=250)
-        st.markdown(f"## Welcome to Debate Bot.\n\n\n")
-
-        role_options = ['in opposition to', 'in support of']
-        cfg.human_role = st.selectbox('You (the human) are:', role_options)
-        cfg.bot_role = role_options[1] if cfg.human_role == role_options[0] else role_options[0]
-        if st.session_state.current_human_role != cfg.human_role:
-            st.session_state.current_human_role = cfg.human_role
-            reset()
+        st.markdown(f"## Welcome to Drink Name Query Bot.\n\n\n")
 
-        topic_options = list(topics.keys())
-        cfg.topic = st.selectbox('The topic:', topic_options)
-        vq.corpus_id = topics[cfg.topic]
-        if st.session_state.current_topic != cfg.topic:
-            st.session_state.current_topic = cfg.topic
-            reset()
-
-        st.markdown("\n")
-        debate_styles = ['Lincoln-Douglas', 'Spontaneous Argumentation', 'Parliamentary debates']
-        cfg.style = st.selectbox('Debate Style:', debate_styles)
-        if st.session_state.style != cfg.style:
-            st.session_state.style = cfg.style
-            reset()
-
-        st.markdown("\n\n")
         if st.button('Start Over'):
             reset()
 
@@ -78,7 +44,6 @@ def launch_bot():
         )
         st.markdown("---")
 
-
     if "messages" not in st.session_state.keys():
         reset()
 
@@ -96,9 +61,8 @@ def launch_bot():
     # Generate a new response if last message is not from assistant
     if st.session_state.messages[-1]["role"] != "assistant":
         with st.chat_message("assistant", avatar='🤖'):
-            stream = generate_response(prompt, cfg) 
-            with st.spinner('Thinking...'):
-                response = st.write_stream(stream) 
+            response = generate_response(prompt) 
+            st.write(response)
             message = {"role": "assistant", "content": response, "avatar": '🤖'}
             st.session_state.messages.append(message)