Update app.py

app.py

@@ -12,7 +12,7 @@ from langchain.embeddings.openai import OpenAIEmbeddings
 
 
 from langchain.vectorstores import Chroma
-
+user_id = None
 
 chat = ChatOpenAI(model="gpt-4o-mini")
 OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
@@ -20,7 +20,62 @@ OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
 
 embedding_function = OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY, model="text-embedding-3-large")
 
+def track_user_interaction(user_input, action):
+    global user_id
+    print(user_input)
+    
+    user_id = str(user_id)
+    
 
+    # Construct connection string
+    from mysql.connector import errorcode
+
+    try:
+        connection = mysql.connector.connect(user=os.getenv("MYSQLUSER"), password= os.getenv("MYSQLPASSWORD"), host=os.getenv("DBHOST"), port=3306, database="user_interact")
+        print("Connection established")
+    except mysql.connector.Error as err:
+        if err.errno == errorcode.ER_ACCESS_DENIED_ERROR:
+            print("Something is wrong with the user name or password")
+        elif err.errno == errorcode.ER_BAD_DB_ERROR:
+            print("Database does not exist")
+        else:
+            print(err)
+    
+    cursor = connection.cursor()
+
+
+    # Create table if it doesn't exist
+    cursor.execute('''
+    CREATE TABLE IF NOT EXISTS user_interactions (
+    user_input LONGTEXT NOT NULL,
+    action TEXT NOT NULL,
+    timestamp TEXT NOT NULL,
+    user_id TEXT NOT NULL
+);
+''')
+    # Prepare data
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    user_input_str = str(user_input)
+    action_str = str(action)
+
+# Use a parameterized query to insert data
+    insert_query = """
+INSERT INTO user_interactions (user_input, action, timestamp, user_id)
+VALUES (%s, %s, %s, %s)
+"""
+    cursor.execute(insert_query, (user_input_str, action_str, timestamp, user_id))
+    # Commit changes and close connection
+    connection.commit()
+    connection.close()
+
+def profile_user(username, password):
+    global user_id
+    user_id = username
+    track_user_interaction("", "login")
+    if password == os.getenv("APP_PASSWORD"):
+        return True
+    else:
+        return False
 
 
 
@@ -66,7 +121,7 @@ def answer_query(message, chat_history):
 
 
 with gr.Blocks() as demo:
-    gr.HTML("<h1 align = 'center'>ChatGPT augmented with specific Cyber Security Knowledge</h1>")
+    gr.HTML("<h1 align = 'center'>ChatGPT augmented with Cyber Security Knowledge</h1>")
     gr.Markdown("""## 1. Input: 🔍
                 **Define Your Workshop Objective.**
                 Choose a topic that is timely and fills a skill gap relevant to your consulting firm’s strategic goals. 
@@ -84,9 +139,9 @@ with gr.Blocks() as demo:
     
     chatbot = gr.Chatbot()
     msg = gr.Textbox(label = "Enter your question here")
-    msg.submit(answer_query,[msg,chatbot],[msg,chatbot])
+    msg.submit(answer_query,[msg,chatbot],[msg,chatbot]).then(track_user_interaction, inputs=[msg, gr.Textbox("promptChatGPT", visible=False)].then(track_user_interaction, inputs=[chatbot, gr.Textbox("ChatGPTreply", visible=False)]
     submitChat = gr.Button("Submit message")
-    submitChat.click(answer_query,[msg,chatbot],[msg,chatbot])
+    submitChat.click(answer_query,[msg,chatbot],[msg,chatbot]).then(track_user_interaction, inputs=[msg, gr.Textbox("promptChatGPT", visible=False)].then(track_user_interaction, inputs=[chatbot, gr.Textbox("ChatGPTreply", visible=False)]
 
 
 if __name__ == "__main__":