main.py

import websocket
import json
import pandas as pd
from datetime import datetime
from config import API_TOKEN, APP_ID, MARKET
from indicators import compute_indicators, generate_signal

# Variables to hold price data
prices = []
timestamps = []

def on_open(ws):
    print("🔌 Connected to Deriv...")
    ws.send(json.dumps({"authorize": API_TOKEN}))

def on_message(ws, message):
    global prices, timestamps

    data = json.loads(message)

    # Handle authorization
    if "authorize" in data:
        print("✅ Authorized. Subscribing to market ticks...")
        ws.send(json.dumps({
            "ticks": MARKET,
            "subscribe": 1
        }))

    # Handle tick data
    elif "tick" in data:
        tick = data["tick"]
        price = float(tick["quote"])
        timestamp = tick["epoch"]

        prices.append(price)
        timestamps.append(datetime.fromtimestamp(timestamp))

        if len(prices) >= 30:
            df = pd.DataFrame({
                "close": prices[-30:],
                "timestamp": timestamps[-30:]
            })
            df = compute_indicators(df)
            signal = generate_signal(df)

            print(f"[{timestamp}] Signal: {signal} | Price: {price}")

            # Log to CSV
            with open("trades.csv", "a") as file:
                file.write(f"{timestamp},{signal},{price}\n")

def on_error(ws, error):
    print("❌ Error:", error)

def on_close(ws, close_status_code, close_msg):
    print("🔌 Disconnected from Deriv")

# Run the WebSocket connection
socket = f"wss://ws.deriv.com/websockets/v3?app_id={APP_ID}"
ws = websocket.WebSocketApp(socket, 
                            on_open=on_open, 
                            on_message=on_message,
                            on_error=on_error, 
                            on_close=on_close)

ws.run_forever()