Update app.py

app.py

@@ -1,3 +1,5 @@
+# app.py
+
 # Import necessary libraries
 import streamlit as st
 import requests
@@ -6,7 +8,6 @@ import uuid
 import os
 import hashlib
 from datetime import datetime, date, timedelta
-from dotenv import load_dotenv
 import pandas as pd
 from sklearn.ensemble import IsolationForest  # For anomaly detection
 from cryptography.fernet import Fernet  # For encryption
@@ -24,9 +25,18 @@ from huggingface_hub import HfApi, HfFolder, Repository  # For Hugging Face API
 from requests.auth import HTTPBasicAuth
 from oauthlib.oauth1 import Client  # For MasterCard OAuth1
 import json
-import cv2
-from pyzbar import pyzbar
-from streamlit_webrtc import webrtc_streamer, VideoTransformerBase
+
+# Optional imports for barcode scanning
+try:
+    import cv2
+    from pyzbar import pyzbar
+    from streamlit_webrtc import webrtc_streamer, VideoTransformerBase
+except ImportError:
+    st.warning("OpenCV and pyzbar are not available. Barcode scanning will be disabled.")
+    cv2 = None
+    pyzbar = None
+    webrtc_streamer = None
+    VideoTransformerBase = object
 
 # Set page config
 st.set_page_config(
@@ -37,7 +47,6 @@ st.set_page_config(
 
 # Load environment variables securely
 def load_env_vars():
-    load_dotenv()
     env_vars = os.environ
     return env_vars
 
@@ -90,27 +99,30 @@ if huggingface_token:
 else:
     huggingface_token = HfFolder.get_token()
 
-# Initialize Plaid client
-from plaid.api import plaid_api
-from plaid.model import *
-from plaid.api_client import ApiClient
-from plaid.configuration import Configuration
-
-plaid_environment = {
-    'sandbox': plaid.Environment.Sandbox,
-    'development': plaid.Environment.Development,
-    'production': plaid.Environment.Production,
-}
-
-plaid_config = Configuration(
-    host=plaid_environment.get(plaid_env, plaid.Environment.Sandbox),
-    api_key={
-        'clientId': plaid_client_id,
-        'secret': plaid_secret,
+# Initialize Plaid client if credentials are provided
+if plaid_client_id and plaid_secret:
+    from plaid.api import plaid_api
+    from plaid.model import *
+    from plaid.api_client import ApiClient
+    from plaid.configuration import Configuration
+
+    plaid_environment = {
+        'sandbox': plaid.Environment.Sandbox,
+        'development': plaid.Environment.Development,
+        'production': plaid.Environment.Production,
     }
-)
-plaid_api_client = ApiClient(plaid_config)
-plaid_client = plaid_api.PlaidApi(plaid_api_client)
+
+    plaid_config = Configuration(
+        host=plaid_environment.get(plaid_env, plaid.Environment.Sandbox),
+        api_key={
+            'clientId': plaid_client_id,
+            'secret': plaid_secret,
+        }
+    )
+    plaid_api_client = ApiClient(plaid_config)
+    plaid_client = plaid_api.PlaidApi(plaid_api_client)
+else:
+    plaid_client = None
 
 # JWT Authentication
 def authenticate_user():
@@ -161,109 +173,112 @@ else:
     if service_choice == "In-Store Scanner & Payment":
         st.header("In-Store Item Scanner and Payment")
 
-        # Mock database of items
-        ITEM_DATABASE = {
-            "123456789012": {"name": "Product A", "price": 9.99},
-            "987654321098": {"name": "Product B", "price": 19.99},
-            # Add more items as needed
-        }
-
-        # Function to get item details
-        def get_item_details(code):
-            return ITEM_DATABASE.get(code)
-
-        # Video transformer for barcode scanning
-        class BarcodeScanner(VideoTransformerBase):
-            def __init__(self):
-                self.result = None
-
-            def transform(self, frame):
-                image = frame.to_ndarray(format="bgr24")
-                barcodes = pyzbar.decode(image)
-
-                for barcode in barcodes:
-                    (x, y, w, h) = barcode.rect
-                    cv2.rectangle(image, (x, y), (x + w, y + h), (0, 255, 0), 2)
-
-                    barcode_data = barcode.data.decode("utf-8")
-                    barcode_type = barcode.type
-
-                    text = f"{barcode_data} ({barcode_type})"
-                    cv2.putText(image, text, (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
-                    self.result = barcode_data
-
-                return image
-
-        st.write("Scan an item to view details and purchase.")
-
-        # Start barcode scanner
-        barcode_scanner = BarcodeScanner()
-        webrtc_ctx = webrtc_streamer(
-            key="barcode-scanner",
-            video_transformer_factory=lambda: barcode_scanner,
-            async_transform=True,
-        )
+        if cv2 and pyzbar and webrtc_streamer and VideoTransformerBase:
+            # Mock database of items
+            ITEM_DATABASE = {
+                "123456789012": {"name": "Product A", "price": 9.99},
+                "987654321098": {"name": "Product B", "price": 19.99},
+                # Add more items as needed
+            }
 
-        if webrtc_ctx.video_transformer:
-            barcode_data = barcode_scanner.result
-            if barcode_data:
-                item_details = get_item_details(barcode_data)
-                if item_details:
-                    st.success(f"Item Scanned: {item_details['name']}")
-                    st.write(f"Price: ${item_details['price']:.2f}")
+            # Function to get item details
+            def get_item_details(code):
+                return ITEM_DATABASE.get(code)
+
+            # Video transformer for barcode scanning
+            class BarcodeScanner(VideoTransformerBase):
+                def __init__(self):
+                    self.result = None
+
+                def transform(self, frame):
+                    image = frame.to_ndarray(format="bgr24")
+                    barcodes = pyzbar.decode(image)
+
+                    for barcode in barcodes:
+                        (x, y, w, h) = barcode.rect
+                        cv2.rectangle(image, (x, y), (x + w, y + h), (0, 255, 0), 2)
+
+                        barcode_data = barcode.data.decode("utf-8")
+                        barcode_type = barcode.type
+
+                        text = f"{barcode_data} ({barcode_type})"
+                        cv2.putText(image, text, (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
+                        self.result = barcode_data
+
+                    return image
+
+            st.write("Scan an item to view details and purchase.")
+
+            # Start barcode scanner
+            barcode_scanner = BarcodeScanner()
+            webrtc_ctx = webrtc_streamer(
+                key="barcode-scanner",
+                video_transformer_factory=lambda: barcode_scanner,
+                async_transform=True,
+            )
+
+            if webrtc_ctx.video_transformer:
+                barcode_data = barcode_scanner.result
+                if barcode_data:
+                    item_details = get_item_details(barcode_data)
+                    if item_details:
+                        st.success(f"Item Scanned: {item_details['name']}")
+                        st.write(f"Price: ${item_details['price']:.2f}")
+
+                        if st.button("Purchase"):
+                            # Process payment
+                            try:
+                                # In a real application, collect payment details securely
+                                # For demonstration, we'll simulate a successful payment
+                                st.write("Payment processed successfully.")
+
+                                # Update Modern Treasury ledger
+                                mt_response = create_mt_ledger_transaction(
+                                    amount=item_details['price'],
+                                    description=f"Purchase of {item_details['name']}"
+                                )
+                                if mt_response:
+                                    st.write("Ledger updated successfully.")
+                                else:
+                                    st.write("Failed to update ledger.")
+                            except Exception as e:
+                                st.error(f"Error processing payment: {e}")
+                    else:
+                        st.error("Item not found in database.")
 
-                    if st.button("Purchase"):
-                        # Process payment
-                        try:
-                            # In a real application, collect payment details securely
-                            # For demonstration, we'll simulate a successful payment
-                            st.write("Payment processed successfully.")
-
-                            # Update Modern Treasury ledger
-                            mt_response = create_mt_ledger_transaction(
-                                amount=item_details['price'],
-                                description=f"Purchase of {item_details['name']}"
-                            )
-                            if mt_response:
-                                st.write("Ledger updated successfully.")
-                            else:
-                                st.write("Failed to update ledger.")
-                        except Exception as e:
-                            st.error(f"Error processing payment: {e}")
+            def create_mt_ledger_transaction(amount, description):
+                url = "https://app.moderntreasury.com/api/ledger_transactions"
+                credentials = f"{mt_organization_id}:{mt_api_key}"
+                base64_credentials = base64.b64encode(credentials.encode()).decode('utf-8')
+                headers = {
+                    "Authorization": f"Basic {base64_credentials}",
+                    "Content-Type": "application/json",
+                    "Accept": "application/json"
+                }
+                data = {
+                    "description": description,
+                    "status": "posted",
+                    "ledger_entries": [
+                        {
+                            "amount": int(amount * 100),  # Convert to cents
+                            "direction": "debit",
+                            "ledger_account_id": "ledger_account_id_debit",  # Replace with your ledger account ID
+                        },
+                        {
+                            "amount": int(amount * 100),
+                            "direction": "credit",
+                            "ledger_account_id": "ledger_account_id_credit",  # Replace with your ledger account ID
+                        },
+                    ]
+                }
+                response = requests.post(url, headers=headers, json=data)
+                if response.status_code in [200, 201]:
+                    return response.json()
                 else:
-                    st.error("Item not found in database.")
-
-        def create_mt_ledger_transaction(amount, description):
-            url = "https://app.moderntreasury.com/api/ledger_transactions"
-            credentials = f"{mt_organization_id}:{mt_api_key}"
-            base64_credentials = base64.b64encode(credentials.encode()).decode('utf-8')
-            headers = {
-                "Authorization": f"Basic {base64_credentials}",
-                "Content-Type": "application/json",
-                "Accept": "application/json"
-            }
-            data = {
-                "description": description,
-                "status": "posted",
-                "ledger_entries": [
-                    {
-                        "amount": int(amount * 100),  # Convert to cents
-                        "direction": "debit",
-                        "ledger_account_id": "ledger_account_id_debit",  # Replace with your ledger account ID
-                    },
-                    {
-                        "amount": int(amount * 100),
-                        "direction": "credit",
-                        "ledger_account_id": "ledger_account_id_credit",  # Replace with your ledger account ID
-                    },
-                ]
-            }
-            response = requests.post(url, headers=headers, json=data)
-            if response.status_code in [200, 201]:
-                return response.json()
-            else:
-                st.error(f"Modern Treasury Error: {response.text}")
-                return None
+                    st.error(f"Modern Treasury Error: {response.text}")
+                    return None
+        else:
+            st.error("Barcode scanning features are not available in this environment.")
 
     # Citibank Services
     elif service_choice == "Citibank Services":
@@ -384,8 +399,8 @@ else:
     # Plaid Services
     elif service_choice == "Plaid Services":
         st.header("Plaid Services")
-        if not plaid_client_id or not plaid_secret:
-            st.error("Please provide Plaid API credentials.")
+        if not plaid_client:
+            st.error("Plaid API credentials are not configured.")
         else:
             plaid_action = st.selectbox("Choose an Action", [
                 "Link Account",
@@ -769,7 +784,6 @@ else:
                         st.error(f"Error: {response.text}")
             elif visa_action == "Initiate Payment":
                 st.subheader("Initiate a Payment")
-                # Implement payment initiation with Visa API
                 st.info("Payment initiation with Visa API is not implemented in this example.")
 
     # MasterCard Services
@@ -810,7 +824,6 @@ else:
                         st.error(f"Error: {response.text}")
             elif mastercard_action == "Process Payment":
                 st.subheader("Process a Payment")
-                # Implement payment processing with MasterCard API
                 st.info("Payment processing with MasterCard API is not implemented in this example.")
 
     # Pipedream Integration
@@ -926,4 +939,4 @@ else:
 
     # Display footer
     st.sidebar.markdown("Powered by Streamlit")
-    st.sidebar.markdown("Made by Citibank Demo Business Inc.") 
+    st.sidebar.markdown("Made by Citibank Demo Business Inc.")