Add Amplitude Analytics Tracking

app.py

@@ -12,7 +12,10 @@ from datasets import load_dataset
 from vectara_agent.agent import AgentStatusType
 from agent import initialize_agent, get_agent_config
 
+from amplitude import Amplitude, BaseEvent
+
 initial_prompt = "How can I help you today?"
+client = Amplitude("2a706490a2e8d62cdabf007d25408d34")
 
 def toggle_logs():
     st.session_state.show_logs = not st.session_state.show_logs
@@ -76,7 +79,7 @@ def launch_bot():
 
     if "messages" not in st.session_state.keys():
         reset()
-    
+
     # Display chat messages
     for message in st.session_state.messages:
         with st.chat_message(message["role"], avatar=message["avatar"]):
@@ -97,6 +100,14 @@ def launch_bot():
     if prompt:
         st.session_state.messages.append({"role": "user", "content": prompt, "avatar": '🧑‍💻'})
         st.session_state.prompt = prompt  # Save the prompt in session state
+        client.track(
+            BaseEvent(
+                event_type="submitted_query",
+                event_properties={
+                    "query": prompt
+                }
+            )
+        )
         st.session_state.log_messages = []
         st.session_state.show_logs = False
         with st.chat_message("user", avatar='🧑‍💻'):