ubi integration

pages/Semantic_Search.py

@@ -871,45 +871,31 @@ def render_answer(answer,index):
                     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):
-                        doc_id = ans['id']
+                    with st.expander(f"{ans['caption']}", 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)
-                        # Inject the hidden div for JS to hook into
-                        components.html(f"""
-                        <div id="{doc_id}" onclick="sendEvent('{doc_id}')"></div>
-
-                        <script>
-                        function sendEvent(itemId) {{
-                            const payload = {{
-                                target_index: "ubi_events",
-                                document: {{
-                                    action_name: "expander_open",
-                                    item_id: itemId,
-                                    session_id: "{st.session_state.get("session_id", "sess_unknown")}",
-                                    query_id: "{st.session_state.get("query_id", "query_unknown")}",
-                                    timestamp: new Date().toISOString(),
-                                    message_type: "INFO",
-                                    message: "Expander opened for item " + itemId
-                                }}
-                            }};
-
-                            fetch("{lambda_url}", {{
-                                method: "POST",
-                                headers: {{
-                                    "Content-Type": "application/json"
-                                }},
-                                body: JSON.stringify(payload)
-                            }}).then(response => console.log("Sent to Lambda:", response.status));
-                        }}
-
-                        document.getElementById("{doc_id}").click();
-                        </script>
-                        """, height=0)
+
+                        # Check and log only the first time it's expanded
+                        if not st.session_state.get(opened_key):
+                            st.session_state[opened_key] = True
+
+                            # Send the expander open event (only once per session)
+                            ubi.send_data(
+                                index="ubi_events",
+                                document={
+                                    "action_name": "expander_open",
+                                    "item_id": item_id,
+                                    "query_id": st.session_state.query_id,
+                                    "session_id": st.session_state.session_id,
+                                    "timestamp": datetime.utcnow().isoformat() + "Z",
+                                    "message_type": "INFO",
+                                    "message": f"Expander opened for item {item_id}"
+                                }
+                                    )
+                    
                 with inner_col_1:
                     
                     if(st.session_state.input_evaluate == "enabled"):