Update app.py

app.py

@@ -36,15 +36,25 @@ if paste_data:
     data = StringIO(data_section.strip())
     reader = csv.reader(data, delimiter='\n', quotechar='"')
     messages = [row[0] for row in reader]
+    
     if show_notes:
-        row_count = len(messages) // 2
-        for i in range(row_count):
-            st.session_state[f"note{i}"] = messages[i * 2]
-            st.session_state[f"message{i}"] = messages[i * 2 + 1]
+        row_count = 0
+        for message in messages:
+            note_delimiter = "//note//"
+            
+            if note_delimiter in message:
+                note, message = message.split(note_delimiter)[1:-1]
+                st.session_state[f"note{row_count}"] = note.strip()
+            else:
+                st.session_state[f"note{row_count}"] = ""
+                
+            st.session_state[f"message{row_count}"] = message.strip()
+            row_count += 1
     else:
         row_count = len(messages)
         for i, message in enumerate(messages):
             st.session_state[f"message{i}"] = message
+            
     st.session_state.row_count = row_count
 
 if reset: