Update app.py

app.py

@@ -10,10 +10,12 @@ from datetime import datetime
 # Initialize hotdog classification pipeline
 hotdog_pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 
+# Global state for API credentials (updated by user input)
+api_key = gr.State(os.getenv('BINGX_API_KEY', ""))
+api_secret = gr.State(os.getenv('BINGX_API_SECRET', ""))
+
 # Configuration
 API_BASE_URL = "https://open-api.bingx.com"
-API_KEY = os.getenv('BINGX_API_KEY', "")
-API_SECRET = os.getenv('BINGX_API_SECRET', "")
 
 # Generate Signature
 def generate_signature(api_secret, params_str):
@@ -27,32 +29,32 @@ def parse_params(params):
     return params_str + "&timestamp=" + str(int(time.time() * 1000)) if params_str else "timestamp=" + str(int(time.time() * 1000))
 
 # Fetch from BingX API
-def fetch_from_api(endpoint, params=None):
-    if not API_KEY or not API_SECRET:
-        raise ValueError("BingX API Key and Secret are not set. Please configure them in Hugging Face Space Secrets.")
+def fetch_from_api(endpoint, params=None, api_key=api_key, api_secret=api_secret):
+    if not api_key.value or not api_secret.value:
+        raise ValueError("BingX API Key and Secret are not set. Please enter them in the Manage API Credentials section.")
     params = params or {}
     params['recvWindow'] = params.get('recvWindow', '5000')
     params_str = parse_params(params)
-    signature = generate_signature(API_SECRET, params_str)
+    signature = generate_signature(api_secret.value, params_str)
     url = f"{API_BASE_URL}{endpoint}?{params_str}&signature={signature}"
     try:
-        response = requests.get(url, headers={'X-BX-APIKEY': API_KEY, 'Content-Type': 'application/json'})
+        response = requests.get(url, headers={'X-BX-APIKEY': api_key.value, 'Content-Type': 'application/json'})
         response.raise_for_status()
         return response.json()
     except requests.exceptions.RequestException as e:
         raise Exception(f"API Error: {str(e)}")
 
 # Fetch Specific Data
-def fetch_balance():
-    data = fetch_from_api('/openApi/swap/v3/user/balance')
+def fetch_balance(api_key=api_key, api_secret=api_secret):
+    data = fetch_from_api('/openApi/swap/v3/user/balance', api_key=api_key, api_secret=api_secret)
     return data['data'][0]['walletBalance'] if data['data'] and data['data'][0]['asset'] == 'USDT' else 0
 
-def fetch_open_positions():
-    data = fetch_from_api('/openApi/swap/v2/user/positions', {'symbol': 'BTC-USDT'})
+def fetch_open_positions(api_key=api_key, api_secret=api_secret):
+    data = fetch_from_api('/openApi/swap/v2/user/positions', {'symbol': 'BTC-USDT'}, api_key=api_key, api_secret=api_secret)
     return data['data'] or []
 
-def fetch_trade_history():
-    data = fetch_from_api('/openApi/swap/v2/user/income', {'limit': 100})
+def fetch_trade_history(api_key=api_key, api_secret=api_secret):
+    data = fetch_from_api('/openApi/swap/v2/user/income', {'limit': 100}, api_key=api_key, api_secret=api_secret)
     return data['data'] or []
 
 # Helper Functions
@@ -82,7 +84,7 @@ def calculate_portfolio_allocation(positions):
     data = [val / total_value * 100 if total_value else 0 for val in by_symbol.values()]
     return {'labels': labels, 'data': data}
 
-# Update UI Functions (as Gradio components)
+# Update UI Functions
 def update_trading_table(positions, trades):
     table_rows = []
     for pos in positions:
@@ -139,7 +141,7 @@ def update_performance_chart(trades):
     return gr.Chart(value=chart_data, type="line", options={
         'responsive': True,
         'plugins': {'legend': {'display': False}},
-        'scales': {'y': {'grid': {'color': 'rgba(0, 0, 0, 0.05)', 'borderDash': [5]}, 'ticks': {'callback': lambda value: f'${value}'}}, 'x': {'grid': {'display': False}}}
+        'scales': {'y': {'grid': {'color': 'rgba(0, 0, 0, 0.05)', 'borderDash': [5]}, 'ticks': {'callback': lambda value: f'${value}'}}, 'x': {'grid': {'display': false}}}
     })
 
 def update_allocation_chart(allocation):
@@ -166,7 +168,7 @@ def update_allocation_chart(allocation):
     }), gr.HTML(legend_html)
 
 # Main Data Fetch and Update Function
-def update_dashboard():
+def update_dashboard(api_key, api_secret):
     try:
         balance = fetch_balance()
         positions = fetch_open_positions()
@@ -191,10 +193,16 @@ def update_dashboard():
         )
     except Exception as e:
         return (
-            "$0.00", 0, "0 Long • 0 Short", "$0.00", "Low", "0%", gr.HTML("<tr><td colspan='8' class='py-4 text-center'>Error: Failed to sync with BingX API</td></tr>"),
+            "$0.00", 0, "0 Long • 0 Short", "$0.00", "Low", "0%", gr.HTML("<tr><td colspan='8' class='py-4 text-center'>Error: Failed to sync with BingX API - Check credentials</td></tr>"),
             gr.HTML("<div>Error loading stats</div>"), gr.Chart(), (gr.Chart(), gr.HTML("")), "Just now", "Just now"
         )
 
+# Save Credentials Function
+def save_credentials(new_api_key, new_api_secret, api_key, api_secret):
+    if not new_api_key or not new_api_secret:
+        return gr.Warning("Both API Key and Secret are required."), api_key, api_secret
+    return gr.Info("Credentials saved successfully! Syncing data..."), gr.State(value=new_api_key), gr.State(value=new_api_secret)
+
 # Gradio Interface
 with gr.Blocks(title="Nakhoda4X Pro") as demo:
     gr.Markdown("""
@@ -209,7 +217,6 @@ with gr.Blocks(title="Nakhoda4X Pro") as demo:
                 gr.Markdown("Connected to BingX via API")
             with gr.Column():
                 refresh_btn = gr.Button("Refresh", variant="secondary")
-                new_trade_btn = gr.Button("New Trade", variant="primary")
         
         with gr.Row():
             with gr.Column(scale=1):
@@ -244,7 +251,12 @@ with gr.Blocks(title="Nakhoda4X Pro") as demo:
                         with api_connection:
                             gr.HTML("<div class='bg-gradient-to-r from-primary to-secondary rounded-2xl p-6'>")
                             gr.HTML("<div class='flex items-center mb-4'><div class='mr-4'><div class='h-12 w-12 rounded-xl bg-white/30 flex items-center justify-center'><i class='fas fa-plug text-white text-xl'></i></div></div><div><h3 class='text-lg font-bold text-white'>BingX API Connected</h3><p id='last-sync' class='text-blue-100'>Last synced: Just now</p></div></div>")
+                            with gr.Row():
+                                api_key_input = gr.Textbox(label="API Key", value=api_key.value, type="password")
+                                api_secret_input = gr.Textbox(label="API Secret", value=api_secret.value, type="password")
+                            save_btn = gr.Button("Save Credentials", variant="primary")
                             sync_now = gr.Button("Sync Now", elem_classes="w-full py-3 bg-white rounded-xl text-primary font-bold flex items-center justify-center")
+                            gr.Markdown("**Note:** Credentials are stored in memory for this session only.")
                     with gr.Column():
                         portfolio_allocation = gr.Blocks()
                         with portfolio_allocation:
@@ -254,19 +266,23 @@ with gr.Blocks(title="Nakhoda4X Pro") as demo:
                                 allocation_chart = gr.Chart()
                                 allocation_legend = gr.HTML()
 
-        # Initial update
-        stats_cards.update(value=gr.HTML("<div class='grid grid-cols-1 md:grid-cols-2 lg:grid-cols-4 gap-6'>"))
-        demo.load(fn=update_dashboard, inputs=None, outputs=[
+        # Event Handlers
+        save_btn.click(
+            fn=save_credentials,
+            inputs=[api_key_input, api_secret_input, api_key, api_secret],
+            outputs=[gr.Markdown(), api_key, api_secret]
+        )
+        demo.load(fn=update_dashboard, inputs=[api_key, api_secret], outputs=[
             total_balance, open_trades, trade_types, today_profit,
             risk_exposure, exposure_percent, trading_activity, advanced_stats,
             performance_chart, (allocation_chart, allocation_legend), gr.State(value=None), gr.State(value=None)
         ])
-        refresh_btn.click(fn=update_dashboard, inputs=None, outputs=[
+        refresh_btn.click(fn=update_dashboard, inputs=[api_key, api_secret], outputs=[
             total_balance, open_trades, trade_types, today_profit,
             risk_exposure, exposure_percent, trading_activity, advanced_stats,
             performance_chart, (allocation_chart, allocation_legend), gr.State(value=None), gr.State(value=None)
         ])
-        sync_now.click(fn=update_dashboard, inputs=None, outputs=[
+        sync_now.click(fn=update_dashboard, inputs=[api_key, api_secret], outputs=[
             total_balance, open_trades, trade_types, today_profit,
             risk_exposure, exposure_percent, trading_activity, advanced_stats,
             performance_chart, (allocation_chart, allocation_legend), gr.State(value=None), gr.State(value=None)