Update app.py

app.py

@@ -76,6 +76,12 @@ def colorize_and_update(system_message, submit_update):
 
 def update_text_output(system_message, submit_update):
     submit_update(system_message['content'], None)
+    
+def train(text):
+    now_et = datetime.now(timezone(timedelta(hours=-4)))
+    published_date = now_et.strftime('%m-%d-%y %H:%M')
+    df = pd.DataFrame([text])
+    notion_df.upload(df, 'https://www.notion.so/US-62e861a0b35f43da8ef9a7789512b8c2?pvs=4', title=str(published_date), api_key=API_KEY)
 
 
 def transcribe(audio, text):
@@ -86,6 +92,7 @@ def transcribe(audio, text):
     
     # Check if the first word of the first line is "COLORIZE"
     if text and text.split("\n")[0].split(" ")[0].strip().upper() == "COLORIZE":
+        train(text)
         colorized_input = colorize_text(text)
         return text, colorized_input