Update app.py

app.py

@@ -95,15 +95,17 @@ from pymongo import MongoClient
 import matplotlib.pyplot as plt
 import os
 import ssl
+import pytz  # Importing pytz for timezone handling
 
 
 
 # Fetch the secret key from environment variables
 Mongo_ip = os.getenv("Mongo_IP")
 
+# Setting up IST timezone
+ist_timezone = pytz.timezone("Asia/Kolkata")
 
 # Connect to MongoDB
-# client = MongoClient()
 client = MongoClient(
     Mongo_ip,
     ssl=True,
@@ -111,8 +113,6 @@ client = MongoClient(
 )
 db = client.GoldRates
 collection = db['GoldRates']
-db = client.GoldRates
-collection = db['GoldRates']
 
 # Backend functions
 def jina(url):
@@ -152,25 +152,24 @@ def insert_data_if_not_exists(city, date, value_24k, value_22k, value_18k):
             "GoldRate_18k": float(value_18k.replace(',', ''))
         }
         collection.insert_one(document)
-        # st.success(f"Gold rates for {city} on {date} have been saved to MongoDB.")
 
 # Function to fetch weekly data for chart
 def fetch_weekly_data(city):
-    today = datetime.datetime.today()
+    today = datetime.datetime.now(ist_timezone)
     start_date = today - datetime.timedelta(days=7)
     query = {"Place": city, "Date": {"$gte": start_date.strftime("%Y-%m-%d")}}
     return list(collection.find(query).sort("Date", -1))
 
-# Function to check if it's the first run of the day after 12:30 PM
+# Function to check if it's the first run of the day after 12:30 PM IST
 def is_first_run_after_1230():
-    today = datetime.datetime.today()
+    today = datetime.datetime.now(ist_timezone)
     time_check = today.replace(hour=12, minute=30, second=0, microsecond=0)
     date_check = today.strftime("%Y-%m-%d")
     return today >= time_check and not collection.find_one({"Date": date_check})
 
 # Fetch and save rates for all cities
 def fetch_and_save_all_cities():
-    date_today = datetime.datetime.today().strftime("%Y-%m-%d")
+    date_today = datetime.datetime.now(ist_timezone).strftime("%Y-%m-%d")
     for city in cities:
         city_url = f"https://www.goodreturns.in/gold-rates/{city}.html"
         try:
@@ -206,7 +205,7 @@ if st.button("Generate Gold Rates"):
 
     # Fetch and display gold rates for the selected city
     if selected_city:
-        date_today = datetime.datetime.today().strftime("%Y-%m-%d")
+        date_today = datetime.datetime.now(ist_timezone).strftime("%Y-%m-%d")
         city_url = f"https://www.goodreturns.in/gold-rates/{selected_city}.html"
 
         try: