Update preprocess.py

preprocess.py

@@ -38,9 +38,9 @@ def preprocess(data):
 
     df = pd.DataFrame({'date': dates, 'message': messages})
     
-    df['date'] = df['date'].astype(str).str.replace('[', '')
-    df['date'] = df['date'].astype(str).str.replace(']', '')
-    df['date'] = df['date'].astype(str).str.replace('-', '')
+    df['date'] = df['date'].astype(str).str.replace('[', '', regex=False)
+    df['date'] = df['date'].astype(str).str.replace(']', '', regex=False)
+    df['date'] = df['date'].astype(str).str.replace('-', '', regex=False)
 
 
 #     group_name = df.loc[df['message'].str.contains('added you')]['message'].str.split(':').str[0]
@@ -57,8 +57,11 @@ def preprocess(data):
     # df['sender'] = df.message.str.split(':').str[0]
     # df['message'] = df.message.str.split(':').str[1]
 
+    # df[['sender', 'message']] = df['message'].str.split(':', n=1, expand=True)
+    df['message'] = df['message'].astype(str)  # Convert column to string type
     df[['sender', 'message']] = df['message'].str.split(':', n=1, expand=True)
 
+
     # df['message'] = df['message'].str.encode('utf-8')
     df = df[['date', 'sender', 'message']]