Update app.py

app.py

@@ -84,6 +84,13 @@ def process_nlp(system_message):
     colorized_text = colorize_text(system_message['content'])
     return colorized_text
 
+
+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 colorize_text(text):
     colorized_text = ""
     lines = text.split("\n")
@@ -128,6 +135,13 @@ def transcribe(audio, text, submit_update=None):
     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")