Add Amplitude Analytics Tracking

app.py

@@ -29,25 +29,25 @@ amp_api_key = os.getenv('AMPLITUDE_TOKEN')
 def thumbs_feedback(feedback, **kwargs):
     print(f'Debug: Feedback Received {feedback["score"]} FROM user question {kwargs.get("prompt", "No user input")} AND chat response {kwargs.get("response", "No chat response")}')
     
-    # Send feedback to Amplitude Analytics
-    # data = {
-    #         "api_key": amp_api_key,
-    #         "events": [{
-    #             "device_id": st.session_state.device_id,
-    #             "event_type": "provided_feedback",
-    #             "event_properties": {
-    #                 "Space Name": cfg["demo_name"],
-    #                 "query": kwargs.get("prompt", "No user input"),
-    #                 "response": kwargs.get("response", "No chat response"),
-    #                 "feedback": feedback["score"]
-    #             }
-    #         }]
-    #     }
-    # response = requests.post('https://api2.amplitude.com/2/httpapi', headers=headers, data=json.dumps(data))
-    # if response.status_code == 200:
-    #     print(f"Request successfully sent: {response.json()}")
-    # else:
-    #     print(f"Request failed with status code {response.status_code}. Response Text: {response.text}")
+    Send feedback to Amplitude Analytics
+    data = {
+            "api_key": amp_api_key,
+            "events": [{
+                "device_id": st.session_state.device_id,
+                "event_type": "provided_feedback",
+                "event_properties": {
+                    "Space Name": cfg["demo_name"],
+                    "query": kwargs.get("prompt", "No user input"),
+                    "response": kwargs.get("response", "No chat response"),
+                    "feedback": feedback["score"]
+                }
+            }]
+        }
+    response = requests.post('https://api2.amplitude.com/2/httpapi', headers=headers, data=json.dumps(data))
+    if response.status_code == 200:
+        print(f"Request successfully sent: {response.json()}")
+    else:
+        print(f"Request failed with status code {response.status_code}. Response Text: {response.text}")
     
     st.session_state.feedback_key += 1
 
@@ -156,24 +156,24 @@ def launch_bot():
             st.session_state.messages.append(message)
             st.markdown(res)
 
-        # Send query and response to Amplitude Analytics
-        # data = {
-        #     "api_key": amp_api_key,
-        #     "events": [{
-        #         "device_id": st.session_state.device_id,
-        #         "event_type": "submitted_query",
-        #         "event_properties": {
-        #             "Space Name": cfg['demo_name'],
-        #             "query": st.session_state.messages[-2]["content"],
-        #             "response": st.session_state.messages[-1]["content"]
-        #         }
-        #     }]
-        # }
-        # response = requests.post('https://api2.amplitude.com/2/httpapi', headers=headers, data=json.dumps(data))
-        # if response.status_code == 200:
-        #     print(f"Request successfully sent: {response.json()}")
-        # else:
-        #     print(f"Request failed with status code {response.status_code}. Response Text: {response.text}")
+        Send query and response to Amplitude Analytics
+        data = {
+            "api_key": amp_api_key,
+            "events": [{
+                "device_id": st.session_state.device_id,
+                "event_type": "submitted_query",
+                "event_properties": {
+                    "Space Name": cfg['demo_name'],
+                    "query": st.session_state.messages[-2]["content"],
+                    "response": st.session_state.messages[-1]["content"]
+                }
+            }]
+        }
+        response = requests.post('https://api2.amplitude.com/2/httpapi', headers=headers, data=json.dumps(data))
+        if response.status_code == 200:
+            print(f"Request successfully sent: {response.json()}")
+        else:
+            print(f"Request failed with status code {response.status_code}. Response Text: {response.text}")
 
         st.session_state.ex_prompt = None
         st.session_state.prompt = None
@@ -191,8 +191,6 @@ def launch_bot():
                 st.button("Show Logs", on_click=toggle_logs)
 
     if (st.session_state.messages[-1]["role"] == "assistant") & (st.session_state.messages[-1]["content"] != "How can I help you today?"):
-        # for i, message in enumerate(st.session_state.messages):
-        #     print(f"DEBUG: Message {i}, {message}")
         streamlit_feedback(feedback_type="thumbs", on_submit = thumbs_feedback, key = st.session_state.feedback_key,
                                       kwargs = {"prompt": st.session_state.messages[-2]["content"], "response": st.session_state.messages[-1]["content"]})