Update app.py

app.py

@@ -65,6 +65,28 @@ def parse_email(raw_email):
 
     return email_details
 
+def parse_whatsapp_chat(input_file_path, output_file_path):
+    with open(input_file_path, 'r', encoding='utf-8') as input_file:
+        lines = input_file.readlines()
+
+    messages = []
+    for line in lines:
+        try:
+            if line.startswith('['):
+                date_time, sender_message = line[1:].split('] ', 1)
+                date, time = date_time.split(', ')
+                sender, message = sender_message.split(': ', 1)
+                messages.append([date, time, sender, message])
+        except ValueError:
+            print(f"Skipping line due to error: {line.strip()} - not enough values to unpack (expected 2, got 1)")
+
+    with open(output_file_path, 'w', newline='', encoding='utf-8') as output_file:
+        writer = csv.writer(output_file)
+        writer.writerow(['Date', 'Time', 'Sender', 'Message'])
+        writer.writerows(messages)
+
+    print(f"File {output_file_path} written with {len(messages)} records")
+
 def dataManipulator(converter, filepath):
     global output_file
     output_file = filepath
@@ -193,42 +215,7 @@ def dataManipulator(converter, filepath):
             return output_file
 
         if converter == "WhatsApp Converter":
-            with open(filepath, 'r', errors='ignore') as file:
-                lines = file.readlines()
-
-            records = []
-            for line in lines:
-                try:
-                    if ', ' in line and ' - ' in line and ': ' in line:
-                        date_time, message = line.split(' - ', 1)
-                        date, time = date_time.split(', ')
-                        sender, message = message.split(': ', 1)
-                        records.append({
-                            "date": date,
-                            "time": time,
-                            "sender": sender,
-                            "message": message
-                        })
-                    else:
-                        print(f"Line doesn't match expected format: {line}")
-                except ValueError as e:
-                    print(f"Skipping line due to error: {line} - {str(e)}")
-                    continue
-
-            print("WhatsApp records parsed:", records)
-
-            output_file = os.path.basename(os.path.splitext(filepath)[0]) + '_converted.csv'
-
-            columns = ["date", "time", "sender", "message"]
-
-            with open(output_file, 'w', newline='') as csvfile:
-                writer = csv.DictWriter(csvfile, fieldnames=columns)
-
-                writer.writeheader()
-                for record in records:
-                    writer.writerow(record)
-
-            print(f"File {output_file} written with {len(records)} records")
+            parse_whatsapp_chat(input_file_path, output_file)
             return output_file
 
         if converter == "Google Foto Converter":
@@ -262,15 +249,15 @@ def dataManipulator(converter, filepath):
 
     return "Converter type not recognized."
 
-def process_input(converter, input_file):
-    if input_file is not None:
-        result = dataManipulator(converter, input_file.name)
-        if result.startswith("Error:"):
-            return None, result
+def process_input(input_file):
+    try:
+        if input_file is not None:
+            output_file = dataManipulator(None, input_file.name)
+            return gr.File.update(value=output_file, label="Download converted file")
         else:
-            return result, None
-    else:
-        return None, "No file uploaded."
+            return gr.Textbox.update(value="No file uploaded.")
+    except Exception as e:
+        return gr.Textbox.update(value=f"Error processing WhatsApp file: {str(e)}")
 
 with gr.Blocks() as demo:
     converter = gr.Dropdown(["Email Converter", "Calendar Converter", "AppleHealth Converter", "AppleHealth GPS Converter", "WhatsApp Converter", "Google Foto Converter"], label="Select Converter")