Update app.py

app.py

@@ -3,15 +3,12 @@ import requests
 import base64
 import uuid
 import os
-from dotenv import load_dotenv
+from dotenv import load_dotenv, dotenv_values
 from datetime import datetime, date
 import plaid
 from plaid import Client as PlaidClient
 import stripe
 
-# Load environment variables from .env file
-load_dotenv()
-
 # Set page config
 st.set_page_config(
     page_title="TransactPro: Unified Financial Platform",
@@ -26,26 +23,40 @@ st.subheader("Made by Citibank Demo Business Inc.")
 # Sidebar for API keys and configuration
 st.sidebar.header("API Configuration")
 
+# File uploader for .env file
+st.sidebar.subheader("Upload .env File")
+uploaded_env = st.sidebar.file_uploader("Choose a .env file", type="env")
+
+if uploaded_env:
+    env_data = dotenv_values(stream=uploaded_env)
+    st.sidebar.success("Environment variables loaded!")
+else:
+    st.sidebar.info("You can upload your .env file to automatically load environment variables.")
+
+# Load environment variables either from uploaded .env or default system
+def get_env_var(key, default=""):
+    return env_data.get(key) if uploaded_env else os.getenv(key, default)
+
 # Modern Treasury Configuration
 st.sidebar.subheader("Modern Treasury")
-mt_api_key = st.sidebar.text_input("Modern Treasury API Key", type="password", value=os.getenv('MT_API_KEY'))
-mt_organization_id = st.sidebar.text_input("Organization ID", value=os.getenv('MT_ORG_ID'))
+mt_api_key = st.sidebar.text_input("Modern Treasury API Key", type="password", value=get_env_var('MT_API_KEY'))
+mt_organization_id = st.sidebar.text_input("Organization ID", value=get_env_var('MT_ORG_ID'))
 
 # Citibank API Configuration
 st.sidebar.subheader("Citibank API")
-citibank_client_id = st.sidebar.text_input("Citibank Client ID", value=os.getenv('CITIBANK_CLIENT_ID'))
-citibank_client_secret = st.sidebar.text_input("Citibank Client Secret", type="password", value=os.getenv('CITIBANK_CLIENT_SECRET'))
-citibank_api_key = st.sidebar.text_input("Citibank API Key", value=os.getenv('CITIBANK_API_KEY'))
+citibank_client_id = st.sidebar.text_input("Citibank Client ID", value=get_env_var('CITIBANK_CLIENT_ID'))
+citibank_client_secret = st.sidebar.text_input("Citibank Client Secret", type="password", value=get_env_var('CITIBANK_CLIENT_SECRET'))
+citibank_api_key = st.sidebar.text_input("Citibank API Key", value=get_env_var('CITIBANK_API_KEY'))
 
 # Plaid API Configuration
 st.sidebar.subheader("Plaid API")
-plaid_client_id = st.sidebar.text_input("Plaid Client ID", value=os.getenv('PLAID_CLIENT_ID'))
-plaid_secret = st.sidebar.text_input("Plaid Secret", type="password", value=os.getenv('PLAID_SECRET'))
+plaid_client_id = st.sidebar.text_input("Plaid Client ID", value=get_env_var('PLAID_CLIENT_ID'))
+plaid_secret = st.sidebar.text_input("Plaid Secret", type="password", value=get_env_var('PLAID_SECRET'))
 plaid_env = st.sidebar.selectbox("Plaid Environment", ["sandbox", "development", "production"], index=0)
 
 # Stripe API Configuration
 st.sidebar.subheader("Stripe API")
-stripe_api_key = st.sidebar.text_input("Stripe API Key", type="password", value=os.getenv('STRIPE_API_KEY'))
+stripe_api_key = st.sidebar.text_input("Stripe API Key", type="password", value=get_env_var('STRIPE_API_KEY'))
 if stripe_api_key:
     stripe.api_key = stripe_api_key
 
@@ -60,11 +71,8 @@ endpoint_choice = st.selectbox("Choose Service", [
 # Helper functions
 
 def create_basic_auth_header(org_id, api_key):
-    # Combine Organization ID and API Key, separated by a colon
     credentials = f"{org_id}:{api_key}"
-    # Encode the credentials in base64
     base64_credentials = base64.b64encode(credentials.encode()).decode('utf-8')
-    # Return the full Authorization header
     return f"Basic {base64_credentials}"
 
 def fetch_modern_treasury_data(api_key, org_id, endpoint):