Update main.py

main.py

@@ -7,6 +7,7 @@ from datetime import datetime, time, timedelta
 from pytz import timezone
 import os
 import csv
+import json
 from streamlit_autorefresh import st_autorefresh
 
 # --- Caching for SuperTrend calculation ---
@@ -38,6 +39,85 @@ def get_current_time():
     """Returns the current time in the 'Asia/Kolkata' timezone."""
     return datetime.now(timezone("Asia/Kolkata"))
 
+def save_settings(config, filename="settings.json"):
+    """Save current settings to a JSON file."""
+    try:
+        # Convert time objects to string format for JSON serialization
+        config_to_save = config.copy()
+        if 'market_start' in config_to_save:
+            config_to_save['market_start'] = config_to_save['market_start'].strftime('%H:%M')
+        if 'market_end' in config_to_save:
+            config_to_save['market_end'] = config_to_save['market_end'].strftime('%H:%M')
+        if 'clear_start' in config_to_save:
+            config_to_save['clear_start'] = config_to_save['clear_start'].strftime('%H:%M')
+        if 'clear_end' in config_to_save:
+            config_to_save['clear_end'] = config_to_save['clear_end'].strftime('%H:%M')
+        
+        with open(filename, 'w') as f:
+            json.dump(config_to_save, f, indent=2)
+        return True
+    except Exception as e:
+        st.error(f"Error saving settings: {e}")
+        return False
+
+def load_settings(filename="settings.json"):
+    """Load settings from a JSON file."""
+    try:
+        if os.path.exists(filename):
+            with open(filename, 'r') as f:
+                config = json.load(f)
+            
+            # Convert string time formats back to time objects
+            if 'market_start' in config:
+                config['market_start'] = datetime.strptime(config['market_start'], '%H:%M').time()
+            if 'market_end' in config:
+                config['market_end'] = datetime.strptime(config['market_end'], '%H:%M').time()
+            if 'clear_start' in config:
+                config['clear_start'] = datetime.strptime(config['clear_start'], '%H:%M').time()
+            if 'clear_end' in config:
+                config['clear_end'] = datetime.strptime(config['clear_end'], '%H:%M').time()
+            
+            return config
+        return None
+    except Exception as e:
+        st.error(f"Error loading settings: {e}")
+        return None
+
+def get_default_settings():
+    """Get default settings configuration."""
+    return {
+        'interval': '5',
+        'limit': 3,
+        'price_from': 100,
+        'price_to': 2000,
+        'amount_per_trade': 5000,
+        'min_volume': 500000,
+        'market_start': time(9, 15),
+        'market_end': time(15, 30),
+        'clear_start': time(9, 10),
+        'clear_end': time(9, 12),
+        'min_conditions': 3,
+        'enable_profit_target': True,
+        'profit_target': 5.0,
+        'enable_atr_stop': True,
+        'atr_multiplier': 2.0,
+        'enable_adx_exit': True,
+        'adx_threshold': 20,
+        'enable_timeout_exit': True,
+        'timeout_minutes': 10,
+        'enable_rsi_exit_long': True,
+        'enable_ema_exit_long': True,
+        'enable_supertrend_exit_long': True,
+        'enable_macd_exit_long': True,
+        'enable_profit_target_short': True,
+        'profit_target_short': 5.0,
+        'enable_rsi_exit_short': True,
+        'enable_ema_exit_short': True,
+        'enable_supertrend_exit_short': True,
+        'enable_macd_exit_short': True,
+        'initial_balance': 100000
+    }
+
 def clear_csv_data_without_header(file_path):
     """
     Clears all data from a CSV file, leaving the header row intact.
@@ -1003,49 +1083,76 @@ if 'eod_positions_closed' not in st.session_state:
 st.sidebar.title("⚙️ Configuration")
 st.sidebar.success("Using hard-coded Indian market times (09:15 - 15:30).")
 
+# Load saved settings or use defaults
+saved_settings = load_settings()
+default_settings = get_default_settings()
+
+# Settings persistence buttons
+st.sidebar.subheader("💾 Settings Management")
+col1, col2, col3 = st.sidebar.columns(3)
+
+if col2.button("📂 Load Settings"):
+    if saved_settings:
+        st.session_state.loaded_settings = saved_settings
+        st.sidebar.success("Settings loaded! Refresh to apply.")
+    else:
+        st.sidebar.warning("No saved settings found.")
+
+if col3.button("🔄 Reset to Defaults"):
+    st.session_state.loaded_settings = default_settings
+    st.sidebar.success("Reset to defaults! Refresh to apply.")
+
+# Use loaded settings if available, otherwise use defaults
+settings_to_use = st.session_state.get('loaded_settings', saved_settings if saved_settings else default_settings)
+
 st.sidebar.subheader("Signal Parameters")
 app_config = {
-    'interval': st.sidebar.select_slider("Time Interval (minutes)", options=['1', '5', '15', '30', '60'], value='5'),
-    'limit': st.sidebar.slider("Scan Limit (stocks per signal)", 1, 10, 3),
-    'price_from': st.sidebar.number_input("Min Price", value=100),
-    'price_to': st.sidebar.number_input("Max Price", value=2000),
-    'amount_per_trade': st.sidebar.number_input("Amount per Trade (₹)", value=5000),
-    'min_volume': st.sidebar.number_input("Minimum Volume", value=500000),
+    'interval': st.sidebar.select_slider("Time Interval (minutes)", options=['1', '5', '15', '30', '60'], value=settings_to_use['interval']),
+    'limit': st.sidebar.slider("Scan Limit (stocks per signal)", 1, 10, settings_to_use['limit']),
+    'price_from': st.sidebar.number_input("Min Price", value=settings_to_use['price_from']),
+    'price_to': st.sidebar.number_input("Max Price", value=settings_to_use['price_to']),
+    'amount_per_trade': st.sidebar.number_input("Amount per Trade (₹)", value=settings_to_use['amount_per_trade']),
+    'min_volume': st.sidebar.number_input("Minimum Volume", value=settings_to_use['min_volume']),
     'market_start': time(9, 15),
     'market_end': time(15, 30),
-    'clear_start': st.sidebar.time_input("Clear Start Time", value=time(9, 10)),
-    'clear_end': st.sidebar.time_input("Clear End Time", value=time(9, 12)),
+    'clear_start': st.sidebar.time_input("Clear Start Time", value=settings_to_use['clear_start']),
+    'clear_end': st.sidebar.time_input("Clear End Time", value=settings_to_use['clear_end']),
 }
 
 # New setting for flexibility
-app_config['min_conditions'] = st.sidebar.select_slider("Minimum Conditions for Entry", options=[5, 4, 3, 2], value=3, help="Choose how many indicators must agree for a trade. 5 is strictest, 2 is most flexible.")
+app_config['min_conditions'] = st.sidebar.select_slider("Minimum Conditions for Entry", options=[5, 4, 3, 2], value=settings_to_use['min_conditions'], help="Choose how many indicators must agree for a trade. 5 is strictest, 2 is most flexible.")
 
 st.sidebar.subheader("Universal Exit Strategy")
-app_config['enable_profit_target'] = st.sidebar.checkbox("Enable Profit Target", value=True)
-app_config['profit_target'] = st.sidebar.number_input("Profit Target (%)", min_value=0.1, value=5.0, step=0.1)
-app_config['enable_atr_stop'] = st.sidebar.checkbox("Enable ATR Stop-Loss", value=True)
-app_config['atr_multiplier'] = st.sidebar.number_input("ATR Multiplier", min_value=0.5, value=2.0, step=0.1)
-app_config['enable_adx_exit'] = st.sidebar.checkbox("Enable ADX Trend Exit", value=True)
-app_config['adx_threshold'] = st.sidebar.slider("ADX Threshold", 10, 40, 20)
-app_config['enable_timeout_exit'] = st.sidebar.checkbox("Enable Timeout Exit", value=True)
-app_config['timeout_minutes'] = st.sidebar.number_input("Timeout (minutes)", min_value=1, value=10, step=1)
+app_config['enable_profit_target'] = st.sidebar.checkbox("Enable Profit Target", value=settings_to_use['enable_profit_target'])
+app_config['profit_target'] = st.sidebar.number_input("Profit Target (%)", min_value=0.1, value=settings_to_use['profit_target'], step=0.1)
+app_config['enable_atr_stop'] = st.sidebar.checkbox("Enable ATR Stop-Loss", value=settings_to_use['enable_atr_stop'])
+app_config['atr_multiplier'] = st.sidebar.number_input("ATR Multiplier", min_value=0.5, value=settings_to_use['atr_multiplier'], step=0.1)
+app_config['enable_adx_exit'] = st.sidebar.checkbox("Enable ADX Trend Exit", value=settings_to_use['enable_adx_exit'])
+app_config['adx_threshold'] = st.sidebar.slider("ADX Threshold", 10, 40, settings_to_use['adx_threshold'])
+app_config['enable_timeout_exit'] = st.sidebar.checkbox("Enable Timeout Exit", value=settings_to_use['enable_timeout_exit'])
+app_config['timeout_minutes'] = st.sidebar.number_input("Timeout (minutes)", min_value=1, value=settings_to_use['timeout_minutes'], step=1)
 
 st.sidebar.subheader("Long Technical Exit Strategy")
-app_config['enable_rsi_exit_long'] = st.sidebar.checkbox("Enable RSI Cross-Below 70 Exit (L)", value=True)
-app_config['enable_ema_exit_long'] = st.sidebar.checkbox("Enable EMA Cross Exit (L)", value=True)
-app_config['enable_supertrend_exit_long'] = st.sidebar.checkbox("Enable SuperTrend Exit (L)", value=True)
-app_config['enable_macd_exit_long'] = st.sidebar.checkbox("Enable MACD Cross Exit (L)", value=True)
+app_config['enable_rsi_exit_long'] = st.sidebar.checkbox("Enable RSI Cross-Below 70 Exit (L)", value=settings_to_use['enable_rsi_exit_long'])
+app_config['enable_ema_exit_long'] = st.sidebar.checkbox("Enable EMA Cross Exit (L)", value=settings_to_use['enable_ema_exit_long'])
+app_config['enable_supertrend_exit_long'] = st.sidebar.checkbox("Enable SuperTrend Exit (L)", value=settings_to_use['enable_supertrend_exit_long'])
+app_config['enable_macd_exit_long'] = st.sidebar.checkbox("Enable MACD Cross Exit (L)", value=settings_to_use['enable_macd_exit_long'])
 
 st.sidebar.subheader("Short Technical Exit Strategy")
-app_config['enable_profit_target_short'] = st.sidebar.checkbox("Enable Profit Target (S)", value=True)
-app_config['profit_target_short'] = st.sidebar.number_input("Profit Target (S) %", min_value=0.1, value=5.0, step=0.1)
-app_config['enable_rsi_exit_short'] = st.sidebar.checkbox("Enable RSI Cross-Above 30 Exit (S)", value=True)
-app_config['enable_ema_exit_short'] = st.sidebar.checkbox("Enable EMA Cross Exit (S)", value=True)
-app_config['enable_supertrend_exit_short'] = st.sidebar.checkbox("Enable SuperTrend Exit (S)", value=True)
-app_config['enable_macd_exit_short'] = st.sidebar.checkbox("Enable MACD Cross Exit (S)", value=True)
+app_config['enable_profit_target_short'] = st.sidebar.checkbox("Enable Profit Target (S)", value=settings_to_use['enable_profit_target_short'])
+app_config['profit_target_short'] = st.sidebar.number_input("Profit Target (S) %", min_value=0.1, value=settings_to_use['profit_target_short'], step=0.1)
+app_config['enable_rsi_exit_short'] = st.sidebar.checkbox("Enable RSI Cross-Above 30 Exit (S)", value=settings_to_use['enable_rsi_exit_short'])
+app_config['enable_ema_exit_short'] = st.sidebar.checkbox("Enable EMA Cross Exit (S)", value=settings_to_use['enable_ema_exit_short'])
+app_config['enable_supertrend_exit_short'] = st.sidebar.checkbox("Enable SuperTrend Exit (S)", value=settings_to_use['enable_supertrend_exit_short'])
+app_config['enable_macd_exit_short'] = st.sidebar.checkbox("Enable MACD Cross Exit (S)", value=settings_to_use['enable_macd_exit_short'])
 
 st.sidebar.subheader("Balance Management")
-app_config['initial_balance'] = st.sidebar.number_input("Initial Balance (₹)", min_value=1000, value=100000, step=1000)
+app_config['initial_balance'] = st.sidebar.number_input("Initial Balance (₹)", min_value=1000, value=settings_to_use['initial_balance'], step=1000)
+
+# Save settings button (after all configuration is complete)
+if col1.button("💾 Save Settings"):
+    if save_settings(app_config):
+        st.sidebar.success("Settings saved successfully!")
 
 # Show current balance and investment in sidebar
 def display_balance():