Update app.py

app.py

@@ -122,9 +122,12 @@ def infer_and_log(text_input):
             print(f"Uploaded log {submission_id}")
         except Exception as e:
             print(f"Error uploading log: {e}")
-    tokens = text_input.split()
+    # tokens = text_input.split()
+    # Replace \n with <br> in the input text for HTML rendering
+    tokens = text_input.replace("\n", " \n ").split()
     
-    formatted_output = " ".join(f'<span style= "color:{color}">{token}</span>' for token, color in zip(tokens, word_colors))
+    formatted_output = " ".join(f'<span style="color:{color}">{token}</span>' if token != "\n" else "<br>" for token, color in zip(tokens, word_colors))
+    # formatted_output = " ".join(f'<span style= "color:{color}">{token}</span>' for token, color in zip(tokens, word_colors))
     return formatted_output, word_probs
 
 def clear_fields():