Updated logging

app.py

@@ -13,10 +13,13 @@ load_dotenv()
 # #===========================================
 # updates = '''
 # Updates
+# +01/08/2026
+# - Updated logging info
+
 # +10/10/2025
 # - Update the model options since Gemma-2-9B-it 
 #    is no longer supported. Replaced with GPT-OSS-120B
-#
+
 # + 04/20/2025
 # - Changed the inference from HF b/c 
 #     API calls are not very limted.
@@ -36,6 +39,120 @@ load_dotenv()
 
 
 
+
+
+
+
+#==========================================================
+# Logging
+#  --------------------------------------------
+
+import requests
+from datetime import datetime
+
+
+try:
+    LOGGER_TOOL_WEBHOOK = os.environ.get("LOGGER_TOOL_URL")
+except Exception as e:
+    print(f"❌ Error in loading LOGGER_TOOL_WEBHOOK: {e}")
+
+    
+
+def log_to_webhook(
+    *,
+    session_info: dict,
+    model: str,
+    prompt: str,
+    response: str,
+    temperature: float,
+):
+    if not LOGGER_TOOL_WEBHOOK:
+        return
+
+    payload = {
+        #Session info
+        **session_info,
+
+        #Model info
+        "model": model,
+        "temperature": temperature,
+
+        #Content
+        "user_prompt": prompt,
+        "assistant_response": response,
+
+        #Usage
+        "api_call_count": st.session_state.api_call_count,
+        "api_call_limit": API_CALL_LIMIT,
+        "remaining_calls": API_CALL_LIMIT - st.session_state.api_call_count,
+
+        #Timestamp
+        "timestamp": datetime.utcnow().isoformat(),
+    }
+
+    try:
+        requests.post(LOGGER_TOOL_WEBHOOK, json=payload, timeout=3)
+    except Exception as e:
+        print("Logging failed:", e)
+
+#  --------------------------------------------
+
+
+#==========================================================
+# Unique Users / Session Info
+#  --------------------------------------------
+import uuid
+import time
+import hashlib
+import json
+import sys
+from datetime import datetime
+
+def get_session_info():
+    data = {
+        "timezone": time.tzname,
+        "platform": sys.platform,
+    }
+    raw = json.dumps(data, sort_keys=True)
+    return hashlib.sha256(raw.encode()).hexdigest()[:12]
+
+
+if "session_info" not in st.session_state:
+    st.session_state.session_info = {
+        "session_id": str(uuid.uuid4()),
+        "session_start": datetime.utcnow().isoformat(),
+        "conversation_id": str(uuid.uuid4()),
+        "run_count": 0,
+        "fingerprint": get_session_info(),
+        "platform": sys.platform,
+        "timezone": time.tzname,
+    }
+
+st.session_state.session_info["run_count"] += 1
+
+
+
+def reset_conversation():
+    st.session_state.conversation = []
+    st.session_state.messages = []
+    st.session_state.session_info["conversation_id"] = str(uuid.uuid4())
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
 API_CALL_LIMIT = 20 # Define the limit
 
 if 'api_call_count' not in st.session_state:
@@ -321,7 +438,20 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question "):
 
                     remaining_calls = (API_CALL_LIMIT) - st.session_state.api_call_count
                     st.markdown(f"\n\n <span style='float: right; font-size: 0.8em; color: gray;'>API calls:({remaining_calls}/{API_CALL_LIMIT})</span>", unsafe_allow_html=True)
-
+                    
+                    #Logging   
+                    try:
+                        log_to_webhook(
+                            session_info=st.session_state.session_info,
+                            model=selected_model,
+                            prompt=prompt,
+                            response=response,
+                            temperature=temp_values,
+                        )
+                    except Exception:
+                        pass
+
+                        
             except Exception as e:
                 response = "😵‍💫 Looks like someone unplugged something!\
                         \n Either the model space is being updated or something is down.\