BankAi is a Python script designed to assist professional accountants in assigning accounts to bank transaction records extracted from an Excel file. It utilizes two AI models, Google's Gemini and Ollama, to suggest the best accounts for each transaction based on provided details.
- Flexible Model Selection : Choose between Google's Gemini and Ollama for account assignment.
- Robust Error Handling : Handles temporary failures gracefully and logs errors for troubleshooting.
- Excel Integration : Reads transaction data from an Excel file and outputs results with assigned accounts.
- Python 3.x
- Libraries: pandas, google.generativeai, ollama, dotenv
- Clone this repository to your local machine.
- Install the required Python libraries using
pip install -r requirements.txt. - Obtain API keys for Google's GenerativeAI and configure them in a
.envfile (refer to.env.example).
- Place your bank transaction Excel file (
accounts-transactions.xls) in the root directory. - Run the script
bank_transaction_assignment.py. - Choose the desired model (Gemini or Ollama) for account assignment.
- The script will process each transaction record, assign accounts, and output the results to the console.
- The processed data with assigned accounts will be saved as
transactions-with-accounts.xls.
Contributions are welcome! If you have any suggestions or improvements, please submit a pull request.