ubi integration

pages/Semantic_Search.py

@@ -6,6 +6,7 @@ import os
 import sys
 import boto3
 import requests
+from datetime import datetime, timedelta
 from requests_aws4auth import AWS4Auth
 sys.path.insert(1, "/".join(os.path.realpath(__file__).split("/")[0:-2])+"/semantic_search")
 sys.path.insert(1, "/".join(os.path.realpath(__file__).split("/")[0:-2])+"/RAG")
@@ -38,6 +39,7 @@ from streamlit.components.v1 import html
 import llm_eval
 import all_search_execute
 import warnings
+import utilities.ubi_lambda as ubi
 
 warnings.filterwarnings("ignore", category=DeprecationWarning)
 st.set_page_config(
@@ -864,12 +866,30 @@ def render_answer(answer,index):
                                 if(sparse_[key]>=1.0):
                                     filtered_sparse[key] = round(sparse_[key], 2)
                             st.write(filtered_sparse)
+                    item_id =  ans['id']  # use a real ID if available
+                    expander_key = f"expander_{item_id}"
+                    opened_key = f"{item_id}_opened"
                     with st.expander("Document Metadata:",expanded = False):
                         st.write(":green[default:]")
                         st.json({"category:":ans['category'],"price":str(ans['price']),"gender_affinity":ans['gender_affinity'],"style":ans['style']},expanded = True)
                         if("rekog" in ans):
                             st.write(":green[enriched:]")
                             st.json(ans['rekog'],expanded = True)
+                        # Log event only on first open
+                        if not st.session_state.get(opened_key):
+                            st.session_state[opened_key] = True
+
+                            # ✅ Build event
+                            event_payload = {
+                                "action_name": "expander_open",
+                                "query_id": st.session_state.get("query_id"),
+                                "session_id": st.session_state.get("session_id"),
+                                "item_id": item_id,
+                                "timestamp": datetime.utcnow().isoformat() + "Z",
+                                "message_type": "INFO",
+                                "message": f"Expander opened for item {item_id}"
+                            }  
+                            status = ubi.send_to_lambda("ubi_events", query_payload)  
                 with inner_col_1:
                     
                     if(st.session_state.input_evaluate == "enabled"):