Update app.py

app.py

@@ -84,6 +84,11 @@ def transcribe(audio, text):
     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":
+        colorized_input = colorize_text(text)
+        return text, colorized_input
+    
     # Transcribe the audio if provided
     if audio is not None:
         audio_file = open(audio, "rb")