Update app.py

app.py

@@ -339,15 +339,16 @@ def generate(
     
     # Generate timestamp
     timestamp = datetime.datetime.now()
-    
-    # Check if the file already exists
+
     if os.path.exists(data_file):
         # If file exists, load existing data
         existing_data = pd.read_csv(data_file)
         
-        # Add timestamp column
+        # Add timestamp, message and a readable sentence column 
         conversation_df = pd.DataFrame(conversation)
+        conversation_df['message'] = message
         conversation_df['timestamp'] = timestamp
+        conversation_df['readable_sentence'] = readable_sentence
         
         # Append new conversation to existing data
         updated_data = pd.concat([existing_data, conversation_df], ignore_index=True)
@@ -355,7 +356,9 @@ def generate(
     else:
         # If file doesn't exist, create new file with conversation data
         conversation_df = pd.DataFrame(conversation)
+        conversation_df['message'] = message
         conversation_df['timestamp'] = timestamp
+        conversation_df['readable_sentence'] = readable_sentence
         conversation_df.to_csv(data_file, index=False)
 
     print("Updating .csv")