Update app.py

app.py

@@ -5,8 +5,12 @@ import uuid
 import os
 from dotenv import load_dotenv, dotenv_values
 from datetime import datetime, date
-import plaid
-from plaid import Client as PlaidClient
+from plaid.model.products import Products
+from plaid.model.country_code import CountryCode
+from plaid.model.link_token_create_request import LinkTokenCreateRequest
+from plaid.configuration import Configuration
+from plaid.api_client import ApiClient
+from plaid.api.plaid_api import PlaidApi
 import stripe
 
 # Set page config
@@ -165,37 +169,42 @@ class CitibankService:
 # Plaid Service Class
 class PlaidService:
     def __init__(self, client_id, secret, environment):
-        env = getattr(plaid.Environment, environment.capitalize())
-        self.client = PlaidClient(
-            client_id=client_id,
-            secret=secret,
-            environment=env
+        configuration = Configuration(
+            host=f"https://{environment}.plaid.com",
+            api_key={
+                'clientId': client_id,
+                'secret': secret,
+            }
         )
+        api_client = ApiClient(configuration)
+        self.client = PlaidApi(api_client)
 
     def create_link_token(self):
-        configs = {
-            'user': {
-                'client_user_id': str(uuid.uuid4())
-            },
-            'client_name': 'TransactPro',
-            'products': ['auth', 'transactions'],
-            'country_codes': ['US'],
-            'language': 'en',
-        }
-        response = self.client.LinkToken.create(configs)
-        return response['link_token']
+        request = LinkTokenCreateRequest(
+            products=[Products('auth')],
+            client_name='TransactPro',
+            country_codes=[CountryCode('US')],
+            user={'client_user_id': str(uuid.uuid4())},
+            language='en'
+        )
+        response = self.client.link_token_create(request)
+        return response.link_token
 
     def exchange_public_token(self, public_token):
-        response = self.client.Item.public_token.exchange(public_token)
-        return response['access_token']
+        response = self.client.item_public_token_exchange({'public_token': public_token})
+        return response.access_token
 
     def get_accounts(self, access_token):
-        response = self.client.Accounts.get(access_token)
-        return response['accounts']
+        response = self.client.accounts_get({'access_token': access_token})
+        return response.accounts
 
     def get_transactions(self, access_token, start_date, end_date):
-        response = self.client.Transactions.get(access_token, start_date, end_date)
-        return response['transactions']
+        response = self.client.transactions_get({
+            'access_token': access_token,
+            'start_date': start_date,
+            'end_date': end_date
+        })
+        return response.transactions
 
 # Main application logic