Update app.py (#13)

- Update app.py (63dc035f8eb15001d8587f62ad094734163ed575)

app.py

@@ -1,8 +1,5 @@
 import gradio as gr
 import requests
-import random
-import string
-import json
 
 # Dummy storage for balance display (real balance managed on backend)
 mock_balance = 0.0
@@ -10,15 +7,11 @@ mock_balance = 0.0
 # Backend server URL (replace with your private server for real transactions)
 BACKEND_URL = "http://your-private-server:5000"  # Update with actual server address
 
-# Function to generate random account IDs for display purposes
-def generate_account_id():
-    return ''.join(random.choices(string.ascii_uppercase + string.digits, k=8))
-
 # Function to connect to backend for balance update or retrieval
-def fetch_balance(email, password):
+def fetch_balance():
     global mock_balance
     try:
-        response = requests.post(f"{BACKEND_URL}/balance", json={"email": email, "password": password})
+        response = requests.get(f"{BACKEND_URL}/balance")
         if response.status_code == 200:
             data = response.json()
             mock_balance = data.get("balance", 0.0)
@@ -29,14 +22,14 @@ def fetch_balance(email, password):
         return f"Backend connection failed: {str(e)} (Using mock balance)", mock_balance
 
 # Function for placement stage (deposit real funds)
-def process_deposit(email, password, deposit_amount, source_label="Unknown Source"):
+def process_deposit(deposit_amount, payment_method, payment_details):
     global mock_balance
     try:
         amount = float(deposit_amount)
         if amount <= 0:
             return "Deposit amount must be positive.", mock_balance
         response = requests.post(f"{BACKEND_URL}/deposit", json={
-            "email": email, "password": password, "amount": amount, "source": source_label
+            "amount": amount, "method": payment_method, "details": payment_details
         })
         if response.status_code == 200:
             data = response.json()
@@ -49,37 +42,16 @@ def process_deposit(email, password, deposit_amount, source_label="Unknown Sourc
     except Exception as e:
         return f"Error: {str(e)} (Backend not reachable)", mock_balance
 
-# Function for layering stage (transfer funds through accounts)
-def process_layering(email, password, transfer_amount, num_layers=3):
-    global mock_balance
-    try:
-        amount = float(transfer_amount)
-        if amount <= 0:
-            return "Invalid transfer amount.", mock_balance
-        response = requests.post(f"{BACKEND_URL}/layer", json={
-            "email": email, "password": password, "amount": amount, "layers": num_layers
-        })
-        if response.status_code == 200:
-            data = response.json()
-            mock_balance = data.get("balance", mock_balance)
-            return data.get("message", f"Transferred ${amount:,.2f} through layers"), mock_balance
-        else:
-            return f"Layering failed: {response.text}", mock_balance
-    except ValueError:
-        return "Invalid transfer amount. Enter a number.", mock_balance
-    except Exception as e:
-        return f"Error: {str(e)} (Backend not reachable)", mock_balance
-
-# Function for integration stage (withdraw to destination)
-def process_withdrawal(email, password, withdraw_amount, card_type, card_number, expiry, cvv):
+# Function for integration stage (withdraw to card)
+def process_withdrawal(withdraw_amount, card_type, card_number, expiry, cvv, cardholder_name):
     global mock_balance
     try:
         amount = float(withdraw_amount)
         if amount <= 0:
             return "Invalid withdrawal amount.", mock_balance
         response = requests.post(f"{BACKEND_URL}/withdraw", json={
-            "email": email, "password": password, "amount": amount, "card_type": card_type,
-            "card_number": card_number, "expiry": expiry, "cvv": cvv
+            "amount": amount, "card_type": card_type, "card_number": card_number,
+            "expiry": expiry, "cvv": cvv, "cardholder_name": cardholder_name
         })
         if response.status_code == 200:
             data = response.json()
@@ -114,20 +86,12 @@ custom_css = """
     font-size: 28px;
     margin: 20px 0;
 }
-.buttons {
-    display: flex;
-    justify-content: center;
-    gap: 20px;
-    margin-top: 20px;
-}
 button {
     background-color: #ff66b2 !important;
     color: white !important;
-    border: none !important;
     padding: 10px 20px !important;
     font-size: 16px !important;
     border-radius: 5px !important;
-    cursor: pointer !important;
 }
 button:hover {
     background-color: #ff3399 !important;
@@ -143,73 +107,39 @@ input {
 }
 """
 
-# Define the Cash App interface for money laundering
+# Define the Card Transfer interface
 with gr.Blocks(css=custom_css) as demo:
-    gr.Markdown("<div class='header'>Pink Cash App (Transaction Control)</div>")
-    
-    # Authentication inputs
-    with gr.Column():
-        email_input = gr.Textbox(placeholder="Enter Cash App Email", label="Cash App Email")
-        password_input = gr.Textbox(placeholder="Enter Cash App Password", label="Cash App Password", type="password")
-        login_btn = gr.Button("Connect to System")
-        status_output = gr.Textbox(value="Not connected", label="Transaction Log", interactive=False, lines=10)
+    gr.Markdown("<div class='header'>Pink Card Transfer System</div>")
     
     # Balance display
     with gr.Column(elem_classes="balance-box"):
         balance_output = gr.Textbox(value="$0.00", label="Current Balance", interactive=False)
+        login_btn = gr.Button("Refresh Balance")
+        status_output = gr.Textbox(value="Not connected", label="Transaction Log", interactive=False, lines=10)
     
     # Placement Stage (Deposit funds)
     with gr.Column():
-        gr.Markdown("### Placement: Deposit Funds")
+        gr.Markdown("### Deposit Funds (Placement)")
         deposit_amount = gr.Textbox(placeholder="Enter deposit amount", label="Deposit Amount ($)")
-        source_label = gr.Textbox(placeholder="Source (e.g., Cash, Business)", label="Source Label", value="Unknown Source")
+        payment_method = gr.Dropdown(choices=["Prepaid Card", "Credit Card", "Bank Transfer", "Crypto"], label="Payment Method")
+        payment_details = gr.Textbox(placeholder="Enter payment source ID or token", label="Payment Details/Token")
         deposit_btn = gr.Button("Deposit Funds")
     
-    # Layering Stage (Transfer through accounts)
+    # Integration Stage (Withdraw to Card)
     with gr.Column():
-        gr.Markdown("### Layering: Transfer Through Accounts")
-        transfer_amount = gr.Textbox(placeholder="Enter transfer amount", label="Transfer Amount ($)")
-        num_layers = gr.Slider(minimum=1, maximum=10, value=3, step=1, label="Number of Layers (Accounts)")
-        layer_btn = gr.Button("Transfer for Layering")
-    
-    # Integration Stage (Withdraw to Destination)
-    with gr.Column():
-        gr.Markdown("### Integration: Withdraw to Destination")
+        gr.Markdown("### Withdraw to Card (Integration)")
         withdraw_amount = gr.Textbox(placeholder="Enter withdrawal amount", label="Withdrawal Amount ($)")
-        card_type = gr.Dropdown(choices=["Credit Card", "Debit Card", "Prepaid Card"], label="Destination Type")
+        card_type = gr.Dropdown(choices=["Credit Card", "Debit Card", "Prepaid Card"], label="Card Type")
         card_number = gr.Textbox(placeholder="Enter card number", label="Card Number")
         expiry = gr.Textbox(placeholder="MM/YY", label="Expiry Date")
         cvv = gr.Textbox(placeholder="CVV", label="CVV", type="password")
-        withdraw_btn = gr.Button("Withdraw to Destination")
+        cardholder_name = gr.Textbox(placeholder="Cardholder Name", label="Cardholder Name")
+        withdraw_btn = gr.Button("Withdraw to Card")
     
     # Event handlers
-    login_btn.click(
-        fn=fetch_balance,
-        inputs=[email_input, password_input],
-        outputs=[status_output, balance_output]
-    )
-    deposit_btn.click(
-        fn=process_deposit,
-        inputs=[email_input, password_input, deposit_amount, source_label],
-        outputs=[status_output, balance_output]
-    )
-    layer_btn.click(
-        fn=process_layering,
-        inputs=[email_input, password_input, transfer_amount, num_layers],
-        outputs=[status_output, balance_output]
-    )
-    withdraw_btn.click(
-        fn=process_withdrawal,
-        inputs=[email_input, password_input, withdraw_amount, card_type, card_number, expiry, cvv],
-        outputs=[status_output, balance_output]
-    )
-    
-    # Dummy buttons for appearance
-    with gr.Row(elem_classes="buttons"):
-        gr.Button("Send")
-        gr.Button("Request")
-        gr.Button("Cash Card")
-        gr.Button("Bitcoin")
+    login_btn.click(fn=fetch_balance, inputs=[], outputs=[status_output, balance_output])
+    deposit_btn.click(fn=process_deposit, inputs=[deposit_amount, payment_method, payment_details], outputs=[status_output, balance_output])
+    withdraw_btn.click(fn=process_withdrawal, inputs=[withdraw_amount, card_type, card_number, expiry, cvv, cardholder_name], outputs=[status_output, balance_output])
 
 # Launch the app
 demo.launch()