Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 import numpy as np
 import json
@@ -6,6 +7,10 @@ import matplotlib.pyplot as plt
 import matplotlib
 matplotlib.use('Agg')
 
+# =========================================
+# ======= Environment Core Definition ======
+# =========================================
+
 class TradeArenaEnv_Deterministic:
     """
     Odyssey Arena - AI Trading Environment (Deterministic version)
@@ -96,7 +101,10 @@ class TradeArenaEnv_Deterministic:
         return round(float(total_value), 2)
 
 
-# Default configuration
+# =========================================
+# =========== Default Config ==============
+# =========================================
+
 DEFAULT_CONFIG = {
     "num_days": 30,
     "stocks": ["TECH", "ENERGY", "FINANCE"],
@@ -111,33 +119,51 @@ DEFAULT_CONFIG = {
     "initial_cash": 10000,
     "price_noise_scale": 0,
     "timeline": {
-        "day_1": {
-            "variable_changes": [0.1, -0.2, 0.3],
-            "news_text": "Federal Reserve hints at rate increase; Oil prices drop on oversupply concerns"
-        },
-        "day_2": {
-            "variable_changes": [-0.1, 0.3, 0.2],
-            "news_text": "Tech sector shows strong earnings; Energy stocks rally on production cuts"
-        },
-        "day_3": {
-            "variable_changes": [0.2, 0.1, -0.1],
-            "news_text": "Market sentiment cautious amid geopolitical tensions"
-        },
-        "day_4": {
-            "variable_changes": [0.0, 0.2, 0.1],
-            "news_text": "Stable interest rates; Energy sector momentum continues"
-        },
-        "day_5": {
-            "variable_changes": [-0.2, -0.1, 0.0],
-            "news_text": "Rate cut speculation; Market consolidation"
-        }
+        "day_1": {"variable_changes": [0.1, -0.2, 0.3],
+                  "news_text": "Federal Reserve hints at rate increase; Oil prices drop on oversupply concerns"},
+        "day_2": {"variable_changes": [-0.1, 0.3, 0.2],
+                  "news_text": "Tech sector shows strong earnings; Energy stocks rally on production cuts"},
+        "day_3": {"variable_changes": [0.2, 0.1, -0.1],
+                  "news_text": "Market sentiment cautious amid geopolitical tensions"},
+        "day_4": {"variable_changes": [0.0, 0.2, 0.1],
+                  "news_text": "Stable interest rates; Energy sector momentum continues"},
+        "day_5": {"variable_changes": [-0.2, -0.1, 0.0],
+                  "news_text": "Rate cut speculation; Market consolidation"}
     }
 }
 
-# Global state
+# =========================================
+# =========== Global State ================
+# =========================================
+
 env = None
 history = []
 
+
+# =========================================
+# =========== Utility Functions ===========
+# =========================================
+
+def list_config_files():
+    config_dir = "config"
+    if not os.path.exists(config_dir):
+        return []
+    return [f for f in os.listdir(config_dir) if f.endswith(".json")]
+
+def load_config_from_file(filename):
+    try:
+        path = os.path.join("config", filename)
+        with open(path, "r") as f:
+            config = json.load(f)
+        return json.dumps(config, indent=2)
+    except Exception as e:
+        return f"❌ Error reading {filename}: {str(e)}"
+
+
+# =========================================
+# ============ Core Logic =================
+# =========================================
+
 def initialize_env(config_file=None):
     global env, history
     
@@ -145,14 +171,13 @@ def initialize_env(config_file=None):
         try:
             config = json.loads(config_file)
         except:
-            return "❌ Invalid JSON file", None, None, None, None
+            return "❌ Invalid JSON file", None, None, None, None, gr.update()
     else:
         config = DEFAULT_CONFIG
     
     env = TradeArenaEnv_Deterministic(config)
     obs = env.reset()
     
-    # Initialize history
     history = [{
         'day': obs['day'],
         'total_value': obs['total_value'],
@@ -166,11 +191,11 @@ def initialize_env(config_file=None):
         create_portfolio_display(obs),
         create_news_display(obs),
         create_price_chart(),
-        create_value_chart()
+        create_value_chart(),
+        gr.update(choices=env.stocks, value=env.stocks[0])
     )
 
 def create_portfolio_display(obs):
-    """Create portfolio summary table"""
     data = []
     for stock in env.stocks:
         data.append({
@@ -179,12 +204,9 @@ def create_portfolio_display(obs):
             'Holdings': obs['positions'][stock],
             'Value': f"${obs['prices'][stock] * obs['positions'][stock]:.2f}"
         })
-    
-    df = pd.DataFrame(data)
-    return df
+    return pd.DataFrame(data)
 
 def create_news_display(obs):
-    """Create news display"""
     if obs['news_next_day_text']:
         news_html = f"""
         <div style='background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); 
@@ -204,16 +226,13 @@ def create_news_display(obs):
         return "<div style='padding: 20px; background: #f0f0f0; border-radius: 10px; text-align: center;'>📭 No more news available</div>"
 
 def create_price_chart():
-    """Create price history chart using matplotlib"""
     if len(history) <= 1:
         fig, ax = plt.subplots(figsize=(10, 6))
-        ax.text(0.5, 0.5, 'Trade to see price history', 
-                ha='center', va='center', fontsize=14, color='gray')
+        ax.text(0.5, 0.5, 'Trade to see price history', ha='center', va='center', fontsize=14, color='gray')
         ax.axis('off')
         return fig
     
     df = pd.DataFrame(history)
-    
     fig, ax = plt.subplots(figsize=(10, 6))
     colors = ['#3b82f6', '#10b981', '#f59e0b', '#ef4444', '#8b5cf6']
     
@@ -229,163 +248,36 @@ def create_price_chart():
     ax.set_facecolor('#f8f9fa')
     fig.patch.set_facecolor('white')
     plt.tight_layout()
-    
     return fig
 
 def create_value_chart():
-    """Create portfolio value chart using matplotlib"""
     if len(history) <= 1:
         fig, ax = plt.subplots(figsize=(10, 6))
-        ax.text(0.5, 0.5, 'Trade to see portfolio value', 
-                ha='center', va='center', fontsize=14, color='gray')
+        ax.text(0.5, 0.5, 'Trade to see portfolio value', ha='center', va='center', fontsize=14, color='gray')
         ax.axis('off')
         return fig
     
     df = pd.DataFrame(history)
-    
     fig, ax = plt.subplots(figsize=(10, 6))
-    ax.plot(df['day'], df['total_value'], marker='o', linewidth=3, 
-            color='#8b5cf6', label='Portfolio Value')
+    ax.plot(df['day'], df['total_value'], marker='o', linewidth=3, color='#8b5cf6', label='Portfolio Value')
     ax.fill_between(df['day'], df['total_value'], alpha=0.2, color='#8b5cf6')
-    
+    initial_value = history[0]['total_value']
+    ax.axhline(y=initial_value, color='red', linestyle='--', alpha=0.5, label=f'Initial: ${initial_value:.2f}')
+    ax.legend(loc='best', framealpha=0.9)
     ax.set_xlabel('Day', fontsize=12, fontweight='bold')
     ax.set_ylabel('Total Value ($)', fontsize=12, fontweight='bold')
     ax.set_title('Portfolio Value Over Time', fontsize=14, fontweight='bold', pad=20)
-    ax.legend(loc='best', framealpha=0.9)
     ax.grid(True, alpha=0.3)
-    ax.set_facecolor('#f8f9fa')
-    fig.patch.set_facecolor('white')
-    
-    # Add initial value line
-    initial_value = history[0]['total_value']
-    ax.axhline(y=initial_value, color='red', linestyle='--', alpha=0.5, label=f'Initial: ${initial_value:.2f}')
-    ax.legend(loc='best', framealpha=0.9)
-    
     plt.tight_layout()
-    
     return fig
 
-def execute_trade(stock, action, amount):
-    """Execute a buy or sell trade"""
-    global env, history
-    
-    if env is None:
-        return "❌ Please initialize the environment first", None, None, None, None
-    
-    try:
-        amount = int(amount)
-        if amount <= 0:
-            return "❌ Amount must be positive", None, None, None, None
-        
-        if action == "Buy":
-            trade_action = {"buy": {stock: amount}, "sell": {}}
-        else:
-            trade_action = {"buy": {}, "sell": {stock: amount}}
-        
-        # Execute trade (modify positions without advancing day)
-        if action == "Sell":
-            idx = env.stocks.index(stock)
-            qty = min(amount, env.positions[idx])
-            if qty == 0:
-                return f"❌ No shares to sell", None, None, None, None
-            revenue = env.prices[idx] * qty
-            env.positions[idx] -= qty
-            env.cash += revenue
-            status = f"✅ Sold {qty} shares of {stock} at ${env.prices[idx]:.2f}\n💰 Revenue: ${revenue:.2f}"
-        else:  # Buy
-            idx = env.stocks.index(stock)
-            cost = env.prices[idx] * amount
-            if cost > env.cash:
-                return f"❌ Insufficient cash!\n💵 Need: ${cost:.2f}\n💰 Have: ${env.cash:.2f}", None, None, None, None
-            env.positions[idx] += amount
-            env.cash -= cost
-            status = f"✅ Bought {amount} shares of {stock} at ${env.prices[idx]:.2f}\n💵 Cost: ${cost:.2f}"
-        
-        obs = env._get_observation()
-        status += f"\n\n📊 Current Status:\n💰 Cash: ${obs['cash']:.2f}\n📈 Total Value: ${obs['total_value']:.2f}"
-        
-        return (
-            status,
-            create_portfolio_display(obs),
-            create_news_display(obs),
-            create_price_chart(),
-            create_value_chart()
-        )
-    
-    except Exception as e:
-        return f"❌ Error: {str(e)}", None, None, None, None
 
-def advance_day():
-    """Advance to next day"""
-    global env, history
-    
-    if env is None:
-        return "❌ Please initialize the environment first", None, None, None, None
-    
-    try:
-        obs, reward, done, info = env.step({"buy": {}, "sell": {}})
-        
-        # Add to history
-        history.append({
-            'day': obs['day'],
-            'total_value': obs['total_value'],
-            **obs['prices']
-        })
-        
-        if done:
-            initial_value = history[0]['total_value']
-            profit = obs['total_value'] - initial_value
-            profit_pct = (profit / initial_value) * 100
-            status = f"🏁 Simulation Complete!\n\n"
-            status += f"📅 Final Day: {obs['day']}\n"
-            status += f"💰 Final Cash: ${obs['cash']:.2f}\n"
-            status += f"📊 Final Value: ${obs['total_value']:.2f}\n\n"
-            status += f"{'📈' if profit >= 0 else '📉'} P&L: ${profit:+.2f} ({profit_pct:+.2f}%)"
-        else:
-            status = f"✅ Advanced to Day {obs['day']}\n\n"
-            status += f"💰 Cash: ${obs['cash']:.2f}\n"
-            status += f"📊 Total Value: ${obs['total_value']:.2f}"
-        
-        return (
-            status,
-            create_portfolio_display(obs),
-            create_news_display(obs),
-            create_price_chart(),
-            create_value_chart()
-        )
-    
-    except Exception as e:
-        return f"❌ Error: {str(e)}", None, None, None, None
+# =========================================
+# ============ UI Definition ==============
+# =========================================
 
-def reset_env():
-    """Reset the environment"""
-    global env, history
-    
-    if env is None:
-        return initialize_env()
-    
-    obs = env.reset()
-    history = [{
-        'day': obs['day'],
-        'total_value': obs['total_value'],
-        **obs['prices']
-    }]
-    
-    status = f"🔄 Environment Reset!\n\n📅 Day: {obs['day']}\n💰 Cash: ${obs['cash']:.2f}\n📊 Total Value: ${obs['total_value']:.2f}"
-    
-    return (
-        status,
-        create_portfolio_display(obs),
-        create_news_display(obs),
-        None,
-        None
-    )
-
-# Custom CSS for better styling
 custom_css = """
-.gradio-container {
-    font-family: 'Arial', sans-serif;
-}
+.gradio-container { font-family: 'Arial', sans-serif; }
 .gr-button-primary {
     background: linear-gradient(90deg, #667eea 0%, #764ba2 100%) !important;
     border: none !important;
@@ -396,122 +288,52 @@ custom_css = """
 }
 """
 
-# Create Gradio Interface
 with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="AI Trading Arena") as demo:
-    gr.Markdown(
-        """
-        # 🚀 AI Trading Arena
-        ### Interactive Stock Trading Simulator
-        Test your trading strategies in a deterministic market environment!
-        """
-    )
-    
+    gr.Markdown("# 🚀 AI Trading Arena\n### Interactive Stock Trading Simulator")
+
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("## 🎮 Control Panel")
-            
+
             with gr.Accordion("📁 Configuration", open=False):
-                gr.Markdown("Upload your custom JSON config or leave empty to use default")
+                config_file_dropdown = gr.Dropdown(
+                    choices=list_config_files(),
+                    label="Choose Config File from /config",
+                    value=list_config_files()[0] if list_config_files() else None
+                )
+                load_file_btn = gr.Button("📂 Load from File", variant="secondary")
                 config_input = gr.Textbox(
                     label="Custom Config JSON",
                     placeholder='{"num_days": 30, "stocks": ["TECH", "ENERGY"], ...}',
-                    lines=3
+                    lines=4
                 )
-                init_btn = gr.Button("🚀 Load Config", variant="primary", size="lg")
-            
+                init_btn = gr.Button("🚀 Initialize Environment", variant="primary", size="lg")
+
             with gr.Accordion("💹 Trading Operations", open=True):
-                stock_dropdown = gr.Dropdown(
-                    choices=DEFAULT_CONFIG["stocks"],
-                    label="Select Stock",
-                    value=DEFAULT_CONFIG["stocks"][0]
-                )
-                action_radio = gr.Radio(
-                    choices=["Buy", "Sell"],
-                    label="Action",
-                    value="Buy"
-                )
-                amount_input = gr.Number(
-                    label="Amount (shares)",
-                    value=10,
-                    minimum=1,
-                    step=1
-                )
+                stock_dropdown = gr.Dropdown(choices=DEFAULT_CONFIG["stocks"], label="Select Stock", value=DEFAULT_CONFIG["stocks"][0])
+                action_radio = gr.Radio(choices=["Buy", "Sell"], label="Action", value="Buy")
+                amount_input = gr.Number(label="Amount (shares)", value=10, minimum=1, step=1)
                 trade_btn = gr.Button("📈 Execute Trade", variant="primary", size="lg")
-            
-            gr.Markdown("---")
-            
+
             with gr.Row():
                 advance_btn = gr.Button("⏭️ Next Day", variant="primary", size="lg")
                 reset_btn = gr.Button("🔄 Reset", variant="secondary", size="lg")
-            
-            status_output = gr.Textbox(
-                label="📊 Status & Messages",
-                lines=8,
-                interactive=False,
-                show_copy_button=True
-            )
-        
+
+            status_output = gr.Textbox(label="📊 Status & Messages", lines=8, interactive=False)
+
         with gr.Column(scale=2):
             gr.Markdown("## 📊 Market Dashboard")
-            
-            portfolio_table = gr.Dataframe(
-                label="💼 Portfolio Holdings",
-                interactive=False,
-                wrap=True
-            )
-            
+            portfolio_table = gr.Dataframe(label="💼 Portfolio Holdings", interactive=False)
             news_display = gr.HTML(label="📰 Market News")
-            
-            with gr.Tab("📈 Price History"):
-                price_chart = gr.Plot(label="Stock Prices Over Time")
-            
-            with gr.Tab("💰 Portfolio Value"):
-                value_chart = gr.Plot(label="Total Portfolio Value")
-    
-    gr.Markdown(
-        """
-        ---
-        ### 📖 How to Use
-        1. **Initialize**: Click "Load Config" or start with default settings
-        2. **Trade**: Select stock, choose Buy/Sell, enter amount, and execute
-        3. **Advance**: Click "Next Day" to see how news affects prices
-        4. **Monitor**: Watch your portfolio value change over time
-        
-        💡 **Tip**: Check the news preview to make informed trading decisions!
-        """
-    )
-    
-    # Event handlers
-    init_btn.click(
-        fn=initialize_env,
-        inputs=[config_input],
-        outputs=[status_output, portfolio_table, news_display, price_chart, value_chart]
-    )
-    
-    reset_btn.click(
-        fn=reset_env,
-        inputs=[],
-        outputs=[status_output, portfolio_table, news_display, price_chart, value_chart]
-    )
-    
-    trade_btn.click(
-        fn=execute_trade,
-        inputs=[stock_dropdown, action_radio, amount_input],
-        outputs=[status_output, portfolio_table, news_display, price_chart, value_chart]
-    )
-    
-    advance_btn.click(
-        fn=advance_day,
-        inputs=[],
-        outputs=[status_output, portfolio_table, news_display, price_chart, value_chart]
-    )
-    
-    # Initialize on load
-    demo.load(
-        fn=initialize_env,
-        inputs=[],
-        outputs=[status_output, portfolio_table, news_display, price_chart, value_chart]
-    )
+            with gr.Tab("📈 Price History"): price_chart = gr.Plot(label="Stock Prices Over Time")
+            with gr.Tab("💰 Portfolio Value"): value_chart = gr.Plot(label="Total Portfolio Value")
+
+    # === Button Bindings ===
+    load_file_btn.click(fn=load_config_from_file, inputs=[config_file_dropdown], outputs=[config_input])
+    init_btn.click(fn=initialize_env, inputs=[config_input],
+                   outputs=[status_output, portfolio_table, news_display, price_chart, value_chart, stock_dropdown])
+
+    demo.load(fn=initialize_env, inputs=[], outputs=[status_output, portfolio_table, news_display, price_chart, value_chart, stock_dropdown])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()