update app.py

app.py

@@ -21,23 +21,23 @@ import utils
 # Initialize configuration
 config = get_config()
 
-def initialize_wb_client():
+def initialize_wb_client(api_token=None):
     """Initialize Wildberries API client with error handling"""
     try:
-        api_token = os.getenv("WILDBERRIES_API_TOKEN")
-        if not api_token:
-            gr.Warning("⚠️ Wildberries API token not configured. Using demo mode.")
+        # Use provided token or fallback to environment variable
+        token = api_token or os.getenv("WILDBERRIES_API_TOKEN")
+        if not token:
             return None
         
-        client = WildberriesAPI(api_token)
+        client = WildberriesAPI(token)
         return client
     except Exception as e:
         gr.Error(f"Failed to initialize Wildberries client: {str(e)}")
         return None
 
-def get_sales_data(period, start_date=None, end_date=None):
+def get_sales_data(period, api_token=None, start_date=None, end_date=None):
     """Fetch sales data with fallback to demo data"""
-    wb_client = initialize_wb_client()
+    wb_client = initialize_wb_client(api_token)
     
     if wb_client is None:
         # Use demo data when API is not available
@@ -58,10 +58,10 @@ def get_sales_data(period, start_date=None, end_date=None):
         gr.Warning(f"API error: {str(e)}. Using demo data.")
         return utils.load_demo_sales_data(period)
 
-def analyze_sales_performance(period):
+def analyze_sales_performance(period, api_token):
     """Analyze sales performance for the specified period"""
     try:
-        data = get_sales_data(period)
+        data = get_sales_data(period, api_token)
         
         if data.empty:
             return "No sales data available for the selected period.", None
@@ -91,11 +91,11 @@ def analyze_sales_performance(period):
         gr.Error(error_msg)
         return error_msg, None
 
-def calculate_stockout_forecast(method="simple"):
+def calculate_stockout_forecast(method, api_token):
     """Calculate days until stockout for products"""
     try:
         # Get current inventory (demo data if API unavailable)
-        wb_client = initialize_wb_client()
+        wb_client = initialize_wb_client(api_token)
         
         if wb_client:
             inventory_data = wb_client.get_stocks()
@@ -103,7 +103,7 @@ def calculate_stockout_forecast(method="simple"):
             inventory_data = utils.load_demo_inventory_data()
         
         # Get sales data for forecasting
-        sales_data = get_sales_data("month")
+        sales_data = get_sales_data("month", api_token)
         
         # Initialize forecaster
         forecaster = InventoryForecaster()
@@ -185,6 +185,13 @@ def calculate_stockout_forecast(method="simple"):
         gr.Error(error_msg)
         return error_msg, None, None
 
+def update_status(api_token):
+    """Update API status based on token"""
+    if not api_token or api_token.strip() == "":
+        return "🔴 Demo Mode - No API token provided"
+    else:
+        return "🟢 API Token Configured - Ready for live data"
+
 # Create Gradio interface
 def create_interface():
     """Create the main Gradio interface"""
@@ -207,15 +214,29 @@ def create_interface():
         - 📈 Interactive dashboards
         """)
         
-        # API Status indicator
+        # API Token Configuration
         with gr.Row():
-            with gr.Column():
+            with gr.Column(scale=3):
+                api_token_input = gr.Textbox(
+                    value="",
+                    label="🔑 Wildberries API Token",
+                    placeholder="Enter your Wildberries API token here (optional - leave empty to use demo mode)",
+                    type="password"
+                )
+            with gr.Column(scale=1):
                 api_status = gr.Textbox(
-                    value="🟡 Checking API connection..." if os.getenv("WILDBERRIES_API_TOKEN") else "🔴 Demo Mode - No API token configured",
+                    value="🔴 Demo Mode - No API token provided",
                     label="Status",
                     interactive=False
                 )
         
+        # Update status when token changes
+        api_token_input.change(
+            fn=update_status,
+            inputs=[api_token_input],
+            outputs=[api_status]
+        )
+        
         with gr.Tabs():
             # Sales Analytics Tab
             with gr.TabItem("📊 Sales Analytics"):
@@ -229,13 +250,13 @@ def create_interface():
                         analyze_btn = gr.Button("📈 Analyze Sales", variant="primary")
                     
                     with gr.Column(scale=3):
-                        sales_summary = gr.Markdown("Select a period and click 'Analyze Sales' to get started.")
+                        sales_summary = gr.Markdown("Enter your API token above and select a period, then click 'Analyze Sales' to get started.")
                         sales_chart = gr.Plot(label="Sales Performance")
                 
                 # Event handlers
                 analyze_btn.click(
                     fn=analyze_sales_performance,
-                    inputs=[period_selector],
+                    inputs=[period_selector, api_token_input],
                     outputs=[sales_summary, sales_chart]
                 )
             
@@ -256,7 +277,7 @@ def create_interface():
                         forecast_btn = gr.Button("🔮 Calculate Forecast", variant="primary")
                     
                     with gr.Column(scale=3):
-                        forecast_summary = gr.Markdown("Select a forecasting method and click 'Calculate Forecast'.")
+                        forecast_summary = gr.Markdown("Enter your API token above and select a forecasting method, then click 'Calculate Forecast'.")
                         forecast_table = gr.DataFrame(
                             headers=["Product", "Current Stock", "Days Until Stockout", "Risk Level"],
                             label="Inventory Forecast Results"
@@ -266,7 +287,7 @@ def create_interface():
                 # Event handlers
                 forecast_btn.click(
                     fn=calculate_stockout_forecast,
-                    inputs=[forecast_method],
+                    inputs=[forecast_method, api_token_input],
                     outputs=[forecast_summary, forecast_table, forecast_chart]
                 )
             
@@ -276,8 +297,9 @@ def create_interface():
                 ## How to Use This Dashboard
                 
                 ### 🔧 Setup
-                1. **API Token**: Configure your Wildberries API token in the Space settings as `WILDBERRIES_API_TOKEN`
+                1. **API Token**: Enter your Wildberries API token in the field above
                 2. **Permissions**: Ensure your token has access to Analytics and Statistics APIs
+                3. **Demo Mode**: Leave token field empty to use demo data
                 
                 ### 📊 Sales Analytics
                 - **Week Analysis**: Shows sales data for the last 7 days
@@ -309,7 +331,7 @@ def create_interface():
         
         gr.Markdown("""
         ---
-        💡 **Tip**: This dashboard works in demo mode even without an API token. Real-time data requires a valid Wildberries API token.
+        💡 **Tip**: This dashboard works in demo mode even without an API token. Enter your Wildberries API token above for real-time data.
         """)
     
     return demo