Mike
125 lines
4.0 KiB
Python
125 lines
4.0 KiB
Python
"""
|
|
Main script for running the trading bot with a web interface.
|
|
|
|
This script initializes the MetaTrader 5 connection, runs the trading bot, and integrates with a web interface for user credentials.
|
|
|
|
Author: Mike Kiwalabye
|
|
"""
|
|
|
|
import time
|
|
from flask import Flask, render_template, request, redirect, globals
|
|
from src.connectors import mt5_connector
|
|
from src.models import neural_network_model
|
|
from src.strategies.trading_strategy import get_historical_data, calculate_indicators_and_detect_patterns, generate_trade_signals, execute_trade
|
|
from src.utils.visualization import plot_trade_signals
|
|
|
|
app = Flask(__name__)
|
|
|
|
# Define input shape for the neural network
|
|
input_shape = (10,) # Adjust the input shape based on your features and data
|
|
|
|
# Create the neural network model
|
|
neural_network_model = neural_network_model.create_neural_network_model(input_shape)
|
|
|
|
# Global state to track whether MT5 is initialized
|
|
globals.mt5_initialized = False
|
|
|
|
# Web Interface Routes
|
|
|
|
@app.route('/')
|
|
def index():
|
|
"""Render the main page with login form."""
|
|
return render_template('index.html')
|
|
|
|
@app.route('/login', methods=['POST'])
|
|
def login():
|
|
"""
|
|
Handle the login form submission.
|
|
|
|
If the credentials are valid, start the trading bot with the provided credentials.
|
|
|
|
Returns:
|
|
- str: HTML response.
|
|
"""
|
|
if request.method == 'POST':
|
|
credentials = {
|
|
'username': request.form['username'],
|
|
'password': request.form['password'],
|
|
'server': request.form['server'],
|
|
'path': request.form['path']
|
|
}
|
|
if mt5_connector.connect_to_mt5(credentials):
|
|
# Set MT5 initialization state to True
|
|
globals.mt5_initialized = True
|
|
# Redirect to the main dashboard or another page
|
|
return redirect('/dashboard')
|
|
else:
|
|
return render_template('index.html', error='Invalid credentials. Please try again.')
|
|
|
|
@app.route('/dashboard')
|
|
def dashboard():
|
|
# Replace these with the actual MetaTrader data retrieval logic
|
|
mt5_data = mt5_connector.get_account_info() # Replace with the actual method to get account info
|
|
user_data = {'username': mt5_data.name, 'account_balance': mt5_data.balance, 'currency': mt5_data.currency}
|
|
username = user_data.get('username', 'N/A')
|
|
account_balance = user_data.get('account_balance', 'N/A')
|
|
account_currency = user_data.get('currency', 'N/A')
|
|
|
|
return render_template('dashboard.html', username=username, account_balance=account_balance, account_currency=account_currency)
|
|
|
|
# Flask app routes
|
|
|
|
@app.route('/start_ml_bot', methods=['POST'])
|
|
def start_ml_bot():
|
|
"""
|
|
Handle the request to start the ML bot.
|
|
|
|
"""
|
|
|
|
# Start the ML bot
|
|
run_trading_bot_web_interface()
|
|
|
|
|
|
# Main Trading Bot Logic
|
|
|
|
def run_trading_bot_web_interface():
|
|
"""
|
|
Run the trading bot using MetaTrader 5 credentials from the web interface.
|
|
"""
|
|
|
|
while True:
|
|
try:
|
|
symbol = 'EURUSD'
|
|
lot_size = 0.01
|
|
stop_loss = 100
|
|
take_profit = 150
|
|
|
|
# Get the latest historical data
|
|
latest_data = get_historical_data(symbol).iloc[-1:]
|
|
|
|
# Calculate indicators and detect patterns for the latest data
|
|
df = calculate_indicators_and_detect_patterns(latest_data)
|
|
|
|
# Generate trade signals for the latest data
|
|
df = generate_trade_signals(df)
|
|
print(df)
|
|
# Execute trades
|
|
for i in range(len(latest_data)):
|
|
signal = df['signal'].iloc[i]
|
|
if signal != 'None':
|
|
print(df['signal'].array)
|
|
execute_trade(signal, df, symbol, lot_size, stop_loss, take_profit)
|
|
|
|
# Visualize data
|
|
# plot_trade_signals(df)
|
|
|
|
except Exception as e:
|
|
print(f"Error running trading bot: {str(e)}")
|
|
|
|
# Wait for the next iteration
|
|
time.sleep(60) # Adjust the time interval as needed
|
|
|
|
# Start the Flask app
|
|
if __name__ == '__main__':
|
|
app.run(debug=True)
|