Update app.py

app.py

@@ -2,7 +2,6 @@ import streamlit as st
 import streamlit_chat
 import json
 import os
-import datetime
 from pymongo import MongoClient
 from bson import ObjectId
 from dotenv import load_dotenv
@@ -14,6 +13,8 @@ import re
 st.set_page_config(layout="wide", page_title="IOCL Chatbot", page_icon="📄")
 load_dotenv()
 import logging
+from pytz import timezone, utc
+from datetime import datetime
 
 logging.basicConfig(
     level=logging.DEBUG,  # This is for your application logs
@@ -55,8 +56,13 @@ if 'regenerate' not in st.session_state:
 
 # Function to create a new chat session in MongoDB
 def create_new_chat_session():
+    # Get the current time in IST
+    ind_time = datetime.now(timezone("Asia/Kolkata"))
+    # Convert IST time to UTC for storing in MongoDB
+    utc_time = ind_time.astimezone(utc)
+
     new_session = {
-        "created_at": datetime.datetime.utcnow(),
+        "created_at": utc_time,  # Store in UTC
         "messages": []  # Empty at first
     }
     session_id = chat_sessions.insert_one(new_session).inserted_id
@@ -250,10 +256,13 @@ if st.sidebar.button("New Chat"):
 existing_sessions = chat_sessions.find().sort("created_at", -1)
 for session in existing_sessions:
     session_id = str(session['_id'])
-    session_date = session['created_at'].strftime("%Y-%m-%d %H:%M:%S")
+    
+    # Retrieve stored UTC time and convert it to IST for display
+    utc_time = session['created_at']
+    ist_time = utc_time.replace(tzinfo=utc).astimezone(timezone("Asia/Kolkata"))
+    session_date = ist_time.strftime("%Y-%m-%d %H:%M:%S")  # Format for display
+    
     col1, col2 = st.sidebar.columns([8, 1])
-
-    # Display session name
     with col1:
         if st.button(f"Session {session_date}", key=session_id):
             st.session_state['current_chat_id'] = session_id