u

app.py

@@ -7,7 +7,7 @@ import tiktoken
 import re
 import openai
 # import firebase_stripe_auth
-
+from utils.firebase_logging import init_firebase, log_to_firestore
 EMBEDDINGS_MODEL = 'text-embedding-3-small'
 groq_api_key = st.secrets["GROQ_API_KEY"]
 pinecone_api_key = st.secrets["PINECONE_API_KEY"]
@@ -55,7 +55,8 @@ about = """
 - Enter your question in the chat box on the right.
 - Choose how many pages of "Results" you would like to get from the knowledge base and added to the AI prompt. Increase the number to extract more information from the knowledge base.
 - After the first prompt, choose whether you want another knowledge base search for the next prompt using the button below.
-- All the PDF pages from the knowledge base are published online and citable. Each PDF page has a link at the bottom going directly the the PDF source page so it can be seen in context.
+- The PDF pages from the knowledge base are published online and citable. Each PDF page has a link at the bottom going directly the the PDF source page so it can be seen in context.
+- This app works best on desktop browsers and is not yet optimized for mobile devices.
 - hudbot version 0.1 (c) 2024 DocMind AI
     """
 
@@ -200,6 +201,9 @@ def generate_response(user_input):
     except Exception as e:
         # Handle the exception
         error_message = f"An error occurred while generating the response: {str(e)}"
+        message = f"hud,anon,generate_response,{e}"
+        level = "ERROR" 
+        log_to_firestore("docmind-421208", message, level)
         yield error_message
         # You can also log the error or take any other necessary action
         print(error_message)
@@ -266,7 +270,9 @@ prompt = ChatPromptTemplate.from_messages([("system", system), ("human", human)]
 
 if user_prompt := st.chat_input("What can I help you with about HUD?"):
     original_user_prompt = user_prompt
-
+    message = f"hud,anon,p,{user_prompt[:200]}"
+    level = "INFO" 
+    log_to_firestore("docmind-421208", message, level)
     if st.session_state['query_pinecone_toggle']:
         user_prompt = get_db_results_new_prompt(user_prompt)
     else:
@@ -277,11 +283,18 @@ if user_prompt := st.chat_input("What can I help you with about HUD?"):
     with st.spinner("Generating response..."):
         try:
             response = st.write_stream(generate_response(user_prompt))
+            message = f"hud,anon,r,{response[:200]}"
+            print(message)
+            level = "INFO" 
+            log_to_firestore("docmind-421208", message, level)
             # If no exception occurred, append the response to the messages
             st.session_state.messages.append(("assistant", response))
         except Exception as e:
             error_message = f"An error occurred while generating the response: {str(e)}"
             st.error(error_message)
+            message = f"hud,anon,show_response,{e}"
+            level = "ERROR" 
+            log_to_firestore("docmind-421208", message, level)
             response = ""
             # You can also log the error or take any other necessary action
             print(error_message)

utils/firebase_logging.py

@@ -0,0 +1,28 @@
+import firebase_admin
+from firebase_admin import credentials, firestore
+import streamlit as st
+def init_firebase(project_id):
+    if not firebase_admin._apps:
+        firebase_creds = credentials.Certificate({
+            "type": "service_account",
+            "project_id": project_id,
+            "private_key_id": st.secrets["private_key_id"],
+            "private_key": st.secrets["private_key"].replace('\\n', '\n'),
+            "client_email": st.secrets["client_email"],
+            "client_id": st.secrets["client_id"],
+            "auth_uri": "https://accounts.google.com/o/oauth2/auth",
+            "token_uri": "https://oauth2.googleapis.com/token",
+            "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
+            "client_x509_cert_url": st.secrets["client_x509_cert_url"]
+        })
+        firebase_admin.initialize_app(firebase_creds)
+    return firestore.client()
+
+def log_to_firestore(project_id, message, level="INFO"):
+    db = init_firebase(project_id)
+    doc_ref = db.collection('logs').document()
+    doc_ref.set({
+        'message': message,
+        'level': level,
+        'timestamp': firestore.SERVER_TIMESTAMP
+    })