Update app.py

app.py

@@ -78,11 +78,26 @@ 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, submit_update=None):
     global messages
     global answer_count
     transcript = {'text': ''} 
     input_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
+
     # Transcribe the audio if provided
     if audio is not None:
         audio_file = open(audio, "rb")