added db analytics

app.py

@@ -1,9 +1,12 @@
+import datetime
 import hmac
 import os
+import uuid
 
 import openai
 import requests
 import streamlit as st
+from azure.cosmos import ContainerProxy, CosmosClient
 from bs4 import BeautifulSoup, NavigableString
 from dotenv import load_dotenv
 from st_copy_to_clipboard import st_copy_to_clipboard
@@ -265,6 +268,13 @@ def on_click_handler_generate_article(**kwargs):
             kwargs["webpage_option"],
         )
         headline = create_headline(created_article, kwargs["webpage_option"])
+    db_analytics_item = {
+        "id": str(uuid.uuid4()),
+        "oparation": "article_generation",
+        "timestamp": str(datetime.datetime.now()),
+    }
+    client: ContainerProxy = st.session_state["db_container"]
+    client.create_item(body=db_analytics_item)
     st.session_state["generated_article"] = created_article
     st.session_state["generated_headline"] = headline
     if st.session_state["process_step"] < 3:
@@ -281,6 +291,13 @@ def on_click_handler_generate_generate_article_keywords(**kwargs):
             kwargs["webpage_option"],
         )
         headline = create_headline(created_article, kwargs["webpage_option"])
+        db_analytics_item = {
+            "id": str(uuid.uuid4()),
+            "oparation": "article_generation",
+            "timestamp": str(datetime.datetime.now()),
+        }
+        client: ContainerProxy = st.session_state["db_container"]
+        client.create_item(body=db_analytics_item)
         st.session_state["generated_article"] = created_article
         st.session_state["generated_headline"] = headline
 
@@ -315,6 +332,19 @@ if "webpage_option" not in st.session_state:
     st.session_state["webpage_option"] = "Boulevard"
 if "studie_links" not in st.session_state:
     st.session_state["studie_links"] = []
+if "db_container" not in st.session_state:
+    client = (
+        CosmosClient(os.environ["DB_ENDPOINT"], os.environ["DB_KEY"])
+        .get_database_client(os.environ["DB_NAME"])
+        .get_container_client("tina-analytics")
+    )
+    db_analytics_item = {
+        "id": str(uuid.uuid4()),
+        "oparation": "page_load",
+        "timestamp": str(datetime.datetime.now()),
+    }
+    client.create_item(body=db_analytics_item)
+    st.session_state["db_container"] = client
 
 PROCESS_STEPS = [
     "Artikel Extraktion",

requirements.txt

@@ -3,4 +3,5 @@ openai==0.28.1
 python-dotenv
 requests
 beautifulsoup4
-st-copy-to-clipboard
+st-copy-to-clipboard
+azure-cosmos