Update app.py

app.py

@@ -38,20 +38,25 @@ if paste_data:
     messages = [row[0] for row in reader]
 
     row_count = 0
-    for i, message in enumerate(messages):
-        if show_notes:
-            note_delimiter = "//note//"
+    note_flag = show_notes  # flag to check if we should add to note or message
 
-            if note_delimiter in message:
-                split_message = message.split(note_delimiter)
-                note, message = split_message[1].strip(), split_message[2].strip()
-            else:
-                note = message
+    for i, message in enumerate(messages):
+        note_delimiter = "//note//"
 
+        if note_delimiter in message:
+            split_message = message.split(note_delimiter)
+            note, message = split_message[1].strip(), split_message[2].strip()
             st.session_state[f"note{row_count}"] = note
+            st.session_state[f"message{row_count}"] = message
+            row_count += 1
+        else:
+            if note_flag:
+                st.session_state[f"note{row_count}"] = message
+            else:
+                st.session_state[f"message{row_count}"] = message
+                row_count += 1
 
-        st.session_state[f"message{row_count}"] = message
-        row_count += 1
+            note_flag = not note_flag  # flip the flag for next iteration
 
     st.session_state.row_count = row_count