Update app.py

app.py

@@ -14,6 +14,7 @@ import matplotlib.pyplot as plt
 from scipy.interpolate import make_interp_spline
 from transformers import AutoTokenizer
 import matplotlib.font_manager as fm
+import pytz
 
 # Load models and data (your existing code)
 autovectorizer = joblib.load('AutoVectorizer.pkl')
@@ -173,8 +174,12 @@ def process_data():
     ax.fill_between(xnew, pred_smooth, color='#244B48', alpha=0.4)
     ax.plot(xnew, pred_smooth, color='#244B48', lw=3, label='Forecast')
     ax.scatter(np.arange(7), pred, color='#244B48', s=100, zorder=5)
-
-    ax.set_title(f"7-Day Political Sentiment Forecast - {datetime.datetime.now()}", fontsize=22, fontweight='bold', pad=20, fontproperties=custom_font)
+    
+    est_timezone = pytz.timezone('America/New_York')
+    est_time = datetime.datetime.now(est_timezone)
+    ax.set_title(f"7-Day Political Sentiment Forecast - {est_time.strftime('%Y-%m-%d %H:%M:%S EST')}", 
+             fontsize=22, fontweight='bold', pad=20, fontproperties=custom_font)
+    # ax.set_title(f"7-Day Political Sentiment Forecast - {datetime.datetime.now()}", fontsize=22, fontweight='bold', pad=20, fontproperties=custom_font)
     ax.set_xlabel("Day", fontsize=16, fontproperties=custom_font)
     ax.set_ylabel("Negative Sentiment (0-1)", fontsize=16, fontproperties=custom_font)
     ax.set_xticks(np.arange(7))
@@ -209,35 +214,7 @@ def display_plot():
         return "Processing data..."
 
 
-# process_data()
-
-# # Schedule daily refresh
-# def run_daily():
-#     process_data()
-#     print("Data refreshed at:", datetime.datetime.now())
-
-# schedule.every().day.at("00:00").do(run_daily)
-
-# def run_schedule():
-#     while True:
-#         schedule.run_pending()
-#         time.sleep(60)
-
-# import threading
-# thread = threading.Thread(target=run_schedule)
-# thread.daemon = True
-# thread.start()
 
-# # Gradio Interface
-# if prediction_plot_base64:
-#     html_content = f'<img src="data:image/png;base64,{prediction_plot_base64}" alt="Prediction Plot">'
-# else:
-#     html_content = "Processing data..."
-
-# iface = gr.Interface(fn=None, inputs=None, outputs=gr.HTML(value=html_content))
-# iface.launch()
-
-# Initial data processing
 process_data()
 
 # Schedule daily refresh
@@ -258,30 +235,7 @@ thread = threading.Thread(target=run_schedule)
 thread.daemon = True
 thread.start()
 
-# # Gradio Interface
-# if prediction_plot_base64:
-#     html_content = (
-#     f'<img src="data:image/png;base64,{prediction_plot_base64}" '
-#     'alt="Prediction Plot" '
-#     'style="width: 100vw; height: 100vh; object-fit: contain;">'
-#     )
-# else:
-#     html_content = "Processing data..."
-
-# custom_css = """
-# body, .gradio-container {
-#     margin: 0;
-#     padding: 0;
-# }
-# """
-
-# with gr.Blocks(css=custom_css) as demo:
-#     gr.HTML(value=html_content)
-
-
-# print("Data refreshed at:", datetime.datetime.now())
 
-# demo.launch()
 
 
 custom_css = """