feat: add user prompt for sampling rate if info.rhd file is missing

[abuzarmahmood]

Co-authored-by: aider (gpt-4o) aider@aider.chat

[abuzarmahmood]

Generated edit command:

Summary of User's Issues and Requests

1. Issue Summary:

   • The blech_clust.py script throws a FileNotFoundError when it cannot find the info.rhd file. The user requests that the script should prompt for a manual input of the sampling rate via CLI if the file is missing. This change should also be applied to blech_reload_amp_digs.py and blech_channel_profile.py.

2. Files Identified for Modification:

   • blech_clust.py: Main script where the error occurs.
   • blech_reload_amp_digs.py: Utility script that may also access info.rhd.
   • blech_channel_profile.py: Another utility script that should handle missing info.rhd.

Overview of Plan to Address the Issues

• Implement error handling in the scripts to check for the existence of info.rhd.
• If the file is not found, prompt the user to manually input the sampling rate via CLI.
• Ensure consistency across all three scripts.

Specific Details of Changes to be Made

1. blech_clust.py

   • File: /home/exouser/Desktop/blech_github_bot/repos/katzlabbrandeis/blech_clust/blech_clust.py
   • Line: Around 290-291
   • Description of Change: Add a check for the info.rhd file and prompt for the sampling rate if missing.
   • Code Snippet:

     import os
     
     # Check if info.rhd file exists
     info_file_path = os.path.join(dir_name, 'info.rhd')
     if os.path.exists(info_file_path):
         info_file = np.fromfile(info_file_path, dtype=np.dtype('float32'))
         sampling_rate = int(info_file[2])
     else:
         print("info.rhd file not found. Please enter the sampling rate manually:")
         sampling_rate = int(input("Sampling rate (Hz): "))

2. blech_reload_amp_digs.py

   • File: /home/exouser/Desktop/blech_github_bot/repos/katzlabbrandeis/blech_clust/utils/blech_reload_amp_digs.py
   • Line: Around 84-85
   • Description of Change: Similar check and prompt as in blech_clust.py.
   • Code Snippet:

     import os
     
     # Check if info.rhd file exists
     info_file_path = os.path.join(dir_name, 'info.rhd')
     if os.path.exists(info_file_path):
         info_file = np.fromfile(info_file_path, dtype=np.dtype('float32'))
         sampling_rate = int(info_file[2])
     else:
         print("info.rhd file not found. Please enter the sampling rate manually:")
         sampling_rate = int(input("Sampling rate (Hz): "))

3. blech_channel_profile.py

   • File: /home/exouser/Desktop/blech_github_bot/repos/katzlabbrandeis/blech_clust/utils/blech_channel_profile.py
   • Line: Around 40-42
   • Description of Change: Implement the same error handling and user prompt for the sampling rate.
   • Code Snippet:

     import os
     
     # Check if info.rhd file exists
     info_file_path = os.path.join(dir_path, 'info.rhd')
     if os.path.exists(info_file_path):
         info_file = np.fromfile(info_file_path, dtype=np.dtype('float32'))
         sampling_rate = int(info_file[2])
     else:
         print("info.rhd file not found. Please enter the sampling rate manually:")
         sampling_rate = int(input("Sampling rate (Hz): "))

These changes will ensure that the scripts handle the absence of the info.rhd file gracefully and allow the user to proceed by manually entering the necessary sampling rate.

---

This response was automatically generated by blech_bot using model gpt-4o

Aider output:

View Aider Output

Aider v0.85.2
Main model: gpt-4o with diff edit format
Weak model: gpt-4o-mini
Git repo: .git with 115 files
Repo-map: using 4096 tokens, auto refresh


blech_clust.py

utils/blech_channel_profile.py

utils/blech_reload_amp_digs.py
To address the issue of handling missing info.rhd files, we will modify the three scripts to prompt the user for manual input of the sampling rate if
the file is not found. Here are the necessary changes:                                                                                               

Changes for blech_clust.py                                                                                                                           

blech_clust.py                                                                                                                                       

                                                                                                                                                     
<<<<<<< SEARCH                                                                                                                                       
    info_file = np.fromfile(dir_name + '/info.rhd', dtype=np.dtype('float32'))                                                                       
    sampling_rate = int(info_file[2])                                                                                                                
=======                                                                                                                                              
    info_file_path = os.path.join(dir_name, 'info.rhd')                                                                                              
    if os.path.exists(info_file_path):                                                                                                               
        info_file = np.fromfile(info_file_path, dtype=np.dtype('float32'))                                                                           
        sampling_rate = int(info_file[2])                                                                                                            
    else:                                                                                                                                            
        print("info.rhd file not found. Please enter the sampling rate manually:")                                                                   
        sampling_rate = int(input("Sampling rate (Hz): "))                                                                                           
>>>>>>> REPLACE                                                                                                                                      
                                                                                                                                                     

Changes for blech_reload_amp_digs.py                                                                                                                 

utils/blech_reload_amp_digs.py                                                                                                                       

                                                                                                                                                     
<<<<<<< SEARCH                                                                                                                                       
info_file = np.fromfile(dir_name + '/info.rhd', dtype=np.dtype('float32'))                                                                           
sampling_rate = int(info_file[2])                                                                                                                    
=======                                                                                                                                              
info_file_path = os.path.join(dir_name, 'info.rhd')                                                                                                  
if os.path.exists(info_file_path):                                                                                                                   
    info_file = np.fromfile(info_file_path, dtype=np.dtype('float32'))                                                                               
    sampling_rate = int(info_file[2])                                                                                                                
else:                                                                                                                                                
    print("info.rhd file not found. Please enter the sampling rate manually:")                                                                       
    sampling_rate = int(input("Sampling rate (Hz): "))                                                                                               
>>>>>>> REPLACE                                                                                                                                      
                                                                                                                                                     

Changes for blech_channel_profile.py                                                                                                                 

utils/blech_channel_profile.py                                                                                                                       

                                                                                                                                                     
<<<<<<< SEARCH                                                                                                                                       
        info_file = np.fromfile(os.path.join(                                                                                                        
            dir_path, 'info.rhd'), dtype=np.dtype('float32'))                                                                                        
        sampling_rate = int(info_file[2])                                                                                                            
=======                                                                                                                                              
        info_file_path = os.path.join(dir_path, 'info.rhd')                                                                                          
        if os.path.exists(info_file_path):                                                                                                           
            info_file = np.fromfile(info_file_path, dtype=np.dtype('float32'))                                                                       
            sampling_rate = int(info_file[2])                                                                                                        
        else:                                                                                                                                        
            print("info.rhd file not found. Please enter the sampling rate manually:")                                                               
            sampling_rate = int(input("Sampling rate (Hz): "))                                                                                       
>>>>>>> REPLACE                                                                                                                                      
                                                                                                                                                     

These changes will ensure that the scripts handle the absence of the info.rhd file gracefully and allow the user to proceed by manually entering the 
necessary sampling rate.                                                                                                                             

Tokens: 14k sent, 523 received. Cost: $0.04 message, $0.04 session.
Applied edit to utils/blech_reload_amp_digs.py
Applied edit to blech_clust.py
Applied edit to utils/blech_channel_profile.py
Commit c5c696b feat: add user prompt for sampling rate if info.rhd file is missing