Update app.py

app.py

@@ -36,26 +36,23 @@ 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 = 0
-        for message in messages:
+
+    row_count = 0
+    for i, message in enumerate(messages):
+        if show_notes:
             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
             else:
-                st.session_state[f"note{row_count}"] = ""
-                
-            st.session_state[f"message{row_count}"] = message
-            row_count += 1
-    else:
-        row_count = len(messages)
-        for i, message in enumerate(messages):
-            st.session_state[f"message{i}"] = message
-            
+                note = message
+
+            st.session_state[f"note{row_count}"] = note
+
+        st.session_state[f"message{row_count}"] = message
+        row_count += 1
+
     st.session_state.row_count = row_count
 
 if reset: