Update app.py

app.py

@@ -1,3 +1,5 @@
+from datetime import datetime
+
 import streamlit as st
 import os
 from openai import OpenAI
@@ -25,6 +27,12 @@ class ChatBot:
         return self.history
 
 
+# Credit: Time
+def current_year():
+    now = datetime.now()
+    return now.year
+
+
 st.set_page_config(layout="wide")
 st.title("OpenAI GPT-4o 🤖")
 
@@ -49,6 +57,14 @@ with st.sidebar:
     st.success("Example: How to conduct an AI experiment?")
     st.success("Example: Write a tensorflow flow code with a 3-layer neural network model.")
 
+    # Credit:
+    current_year = current_year()  # This will print the current year
+    st.markdown(
+        f"""
+            <h6 style='text-align: left;'>Copyright © 2010-{current_year} Present Yiqiao Yin</h6>
+        """,
+        unsafe_allow_html=True,
+    )
 
     # Add a button to clear the session state
     if st.button("Clear Session"):
@@ -78,6 +94,7 @@ if prompt := st.chat_input("😉 Ask any question or feel free to use the exampl
     st.chat_message("user").markdown(prompt)
 
     # Add user message to chat history
+    st.session_state.messages.append({"role": "system", content: f"You are a helpful assistant. Year now is {current_year}"})
     st.session_state.messages.append({"role": "user", "content": prompt})
 
     # API Call