app.py and connection.py are updated to save the data as CSV as well,if asked

Author: PayalLakra

app.py

@@ -82,7 +82,6 @@ def connect_device():
     data = request.get_json()
     protocol = data.get('protocol')
     device_address = data.get('device_address')
-    csv_logging = data.get('csv_logging', False)
 
     # Reset stream status
     stream_active = False
@@ -94,7 +93,7 @@ def connect_device():
             connection_thread.join()
 
     # Create new connection
-    connection_manager = Connection(csv_logging=csv_logging)
+    connection_manager = Connection()
 
     def run_connection():
         try:
@@ -136,5 +135,27 @@ def disconnect_device():
         return jsonify({'status': 'disconnected'})
     return jsonify({'status': 'no active connection'})
 
+@app.route('/start_recording', methods=['POST'])
+def start_recording():
+    global connection_manager
+    if connection_manager and connection_manager.stream_active:
+        try:
+            connection_manager.start_csv_recording()
+            return jsonify({'status': 'recording_started'})
+        except Exception as e:
+            return jsonify({'status': 'error', 'message': str(e)}), 500
+    return jsonify({'status': 'error', 'message': 'No active stream'}), 400
+
+@app.route('/stop_recording', methods=['POST'])
+def stop_recording():
+    global connection_manager
+    if connection_manager:
+        try:
+            connection_manager.stop_csv_recording()
+            return jsonify({'status': 'recording_stopped'})
+        except Exception as e:
+            return jsonify({'status': 'error', 'message': str(e)}), 500
+    return jsonify({'status': 'error', 'message': 'No active connection'}), 400
+
 if __name__ == "__main__":
     app.run(debug=True)

connection.py

@@ -10,7 +10,7 @@
 from datetime import datetime
 
 class Connection:
-    def __init__(self, csv_logging=False):
+    def __init__(self):
         self.ble_connection = None
         self.wifi_connection = None
         self.lsl_connection = None
@@ -21,12 +21,12 @@ def __init__(self, csv_logging=False):
         self.last_sample = None
         self.ble_samples_received = 0
         self.ble_start_time = time.time()
-        self.csv_logging = csv_logging
         self.csv_file = None
         self.csv_writer = None
         self.sample_counter = 0
         self.num_channels = 0
         self.stream_active = False
+        self.recording_active = False
 
     async def get_ble_device(self):
         devices = await Chords_BLE.scan_devices()
@@ -53,27 +53,45 @@ def setup_lsl(self, num_channels, sampling_rate):
         self.lsl_connection = StreamOutlet(info)
         print(f"LSL stream started: {num_channels} channels at {sampling_rate}Hz")
         self.stream_active = True
-        print("Flag is set to True")
         self.num_channels = num_channels
 
-    def setup_csv(self):
-        if not self.csv_logging or self.csv_file:
-            return  # Already set up or logging disabled
-            
+    def start_csv_recording(self):
+        if self.recording_active:
+            return False
+        
         try:
             timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
             filename = f"ChordsPy_{timestamp}.csv"
             self.csv_file = open(filename, 'w', newline='')
             headers = ['Counter'] + [f'Channel{i+1}' for i in range(self.num_channels)]
             self.csv_writer = csv.writer(self.csv_file)
             self.csv_writer.writerow(headers)
-            print(f"CSV logging started: {filename}")
+            self.recording_active = True
+            self.sample_counter = 0
+            print(f"CSV recording started: {filename}")
+            return True
         except Exception as e:
-            print(f"Error setting up CSV logging: {str(e)}")
-            self.csv_logging = False
+            print(f"Error starting CSV recording: {str(e)}")
+            return False
+
+    def stop_csv_recording(self):
+        if not self.recording_active:
+            return False
+        
+        try:
+            if self.csv_file:
+                self.csv_file.close()
+                self.csv_file = None
+                self.csv_writer = None
+            self.recording_active = False
+            print("CSV recording stopped")
+            return True
+        except Exception as e:
+            print(f"Error stopping CSV recording: {str(e)}")
+            return False
 
     def log_to_csv(self, sample_data):
-        if not self.csv_logging or not self.csv_writer:
+        if not self.recording_active or not self.csv_writer:
             return
 
         try:
@@ -82,7 +100,7 @@ def log_to_csv(self, sample_data):
             self.csv_writer.writerow(row)
         except Exception as e:
             print(f"Error writing to CSV: {str(e)}")
-            self.csv_logging = False
+            self.stop_csv_recording()
 
     def connect_ble(self, device_address=None):
         self.ble_connection = Chords_BLE()
@@ -92,7 +110,6 @@ def notification_handler(sender, data):
             if len(data) == self.ble_connection.NEW_PACKET_LEN:
                 if not self.lsl_connection:
                     self.setup_lsl(num_channels=3, sampling_rate=500)
-                    self.setup_csv()
 
                 original_notification_handler(sender, data)
 
@@ -109,7 +126,8 @@ def notification_handler(sender, data):
 
                         if self.lsl_connection:             # Push to LSL
                             self.lsl_connection.push_sample(channels)
-                        self.log_to_csv(channels)           # Log to CSV
+                        if self.recording_active:
+                            self.log_to_csv(channels)
 
         self.ble_connection.notification_handler = notification_handler
 
@@ -125,8 +143,10 @@ def notification_handler(sender, data):
                 self.ble_connection.connect(selected_device.address)
 
             print("BLE connection established. Waiting for data...")
+            return True
         except Exception as e:
             print(f"BLE connection failed: {str(e)}")
+            return False
 
     def connect_usb(self):
         serial_connection = Chords_USB()
@@ -135,18 +155,20 @@ def connect_usb(self):
             sampling_rate = serial_connection.supported_boards[serial_connection.board]["sampling_rate"]
 
             self.setup_lsl(self.num_channels, sampling_rate)
-            self.setup_csv()
 
             original_read_data = serial_connection.read_data
             def wrapped_read_data():
                 original_read_data()
                 if hasattr(serial_connection, 'data') and self.lsl_connection:
                     sample = serial_connection.data[:, -1]
                     self.lsl_connection.push_sample(sample)
-                    self.log_to_csv(sample.tolist())
+                    if self.recording_active:
+                        self.log_to_csv(sample.tolist())
 
             serial_connection.read_data = wrapped_read_data
             serial_connection.start_streaming()
+            return True
+        return False
 
     def connect_wifi(self):
         self.wifi_connection = Chords_WIFI()
@@ -157,8 +179,6 @@ def connect_wifi(self):
 
         if not self.lsl_connection:
             self.setup_lsl(self.num_channels, sampling_rate)
-            if self.csv_logging:
-                self.setup_csv()
 
         try:
             print("\nConnected! (Press Ctrl+C to stop)")
@@ -179,28 +199,35 @@ def connect_wifi(self):
 
                         if self.lsl_connection:                   # Push to LSL
                             self.lsl_connection.push_sample(channel_data)
-                        if self.csv_logging and self.csv_writer:  # Only log if CSV is set up
+                        if self.recording_active:
                             self.log_to_csv(channel_data)
 
         except KeyboardInterrupt:
             self.wifi_connection.disconnect()
             print("\nDisconnected")
         finally:
-            if self.csv_file:
-                self.csv_file.close()
+            self.stop_csv_recording()
+
+    def cleanup(self):
+        self.stop_csv_recording()
+        self.stream_active = False
+        if self.ble_connection:
+            self.ble_connection.disconnect()
+        if self.wifi_connection:
+            self.wifi_connection.disconnect()
+        if self.lsl_connection:
+            self.lsl_connection = None
 
     def __del__(self):
-        if self.csv_file:
-            self.csv_file.close()
+        self.cleanup()
 
 def main():
     parser = argparse.ArgumentParser(description='Connect to device')
     parser.add_argument('--protocol', choices=['usb', 'wifi', 'ble'], required=True, help='Connection protocol')
     parser.add_argument('--ble-address', help='Direct BLE device address')
-    parser.add_argument('--csv', action='store_true', help='Enable CSV logging')
     args = parser.parse_args()
 
-    manager = Connection(csv_logging=args.csv)
+    manager = Connection()
 
     try:
         if args.protocol == 'usb':