Update app.py

app.py

@@ -1,22 +1,94 @@
+import ccxt
+import pandas as pd
+import matplotlib.pyplot as plt
+import matplotlib.dates as mdates
+from io import BytesIO
 import gradio as gr
-from data import fetch_ohlcv
-from indicators import generate_signal
-from charts import plot_candlestick
 
+# Fetch MEXC OHLCV data
+def fetch_ohlcv(symbol='BTC/USDT', timeframe='1h', limit=100):
+    exchange = ccxt.mexc()
+    ohlcv = exchange.fetch_ohlcv(symbol, timeframe=timeframe, limit=limit)
+    df = pd.DataFrame(ohlcv, columns=['timestamp', 'open', 'high', 'low', 'close', 'volume'])
+    df['timestamp'] = pd.to_datetime(df['timestamp'], unit='ms')
+    return df
+
+# Calculate indicators
+def calculate_rsi(df, period=14):
+    delta = df['close'].diff()
+    gain = delta.clip(lower=0)
+    loss = -delta.clip(upper=0)
+
+    avg_gain = gain.rolling(window=period).mean()
+    avg_loss = loss.rolling(window=period).mean()
+
+    rs = avg_gain / avg_loss
+    rsi = 100 - (100 / (1 + rs))
+    return rsi
+
+def calculate_macd(df):
+    ema12 = df['close'].ewm(span=12, adjust=False).mean()
+    ema26 = df['close'].ewm(span=26, adjust=False).mean()
+    macd = ema12 - ema26
+    signal = macd.ewm(span=9, adjust=False).mean()
+    return macd, signal
+
+def generate_signal(df):
+    rsi = calculate_rsi(df)
+    macd, signal = calculate_macd(df)
+
+    if len(rsi) < 1 or len(macd) < 1 or len(signal) < 1:
+        return "Not enough data"
+
+    latest_rsi = rsi.iloc[-1]
+    latest_macd = macd.iloc[-1]
+    latest_signal = signal.iloc[-1]
+
+    if latest_rsi < 30 and latest_macd > latest_signal:
+        return "BUY"
+    elif latest_rsi > 70 and latest_macd < latest_signal:
+        return "SELL"
+    else:
+        return "HOLD"
+
+# Plot candlestick chart
+def plot_candlestick(df):
+    df = df.tail(50).copy()
+    df['timestamp'] = pd.to_datetime(df['timestamp'])
+
+    fig, ax = plt.subplots(figsize=(8, 4))
+
+    for _, row in df.iterrows():
+        color = 'green' if row['close'] >= row['open'] else 'red'
+        ax.plot([row['timestamp'], row['timestamp']], [row['low'], row['high']], color='black')
+        ax.plot([row['timestamp'], row['timestamp']], [row['open'], row['close']], color=color, linewidth=5)
+
+    ax.xaxis.set_major_formatter(mdates.DateFormatter('%H:%M'))
+    ax.xaxis.set_major_locator(mdates.AutoDateLocator())
+    plt.xticks(rotation=45)
+    plt.title('Candlestick Chart (MEXC)')
+    plt.tight_layout()
+
+    buf = BytesIO()
+    plt.savefig(buf, format='png')
+    buf.seek(0)
+    plt.close(fig)
+    return buf
+
+# Main function
 def analyze(pair, timeframe):
     try:
         df = fetch_ohlcv(symbol=pair, timeframe=timeframe)
-        if df is None or df.empty:
-            return "❌ Failed to fetch data", None
+        if df.empty:
+            return "❌ No data returned from MEXC", None
 
         signal = generate_signal(df)
-        chart_img = plot_candlestick(df)
-        return f"📊 Signal: {signal}", chart_img
-
+        chart = plot_candlestick(df)
+        return f"📊 Signal: {signal}", chart
     except Exception as e:
         return f"❌ Error: {str(e)}", None
 
-# MEXC pairs only
+# Gradio UI
 pairs = ['BTC/USDT', 'ETH/USDT', 'MX/USDT', 'SOL/USDT']
 timeframes = ['1m', '5m', '15m', '1h', '4h']
 
@@ -31,5 +103,5 @@ gr.Interface(
         gr.Image(label="Candlestick Chart")
     ],
     title="🧠 Crypto Signal Generator (MEXC)",
-    description="Generates BUY / SELL / HOLD signals using RSI + MACD indicators."
+    description="Live BUY / SELL / HOLD signals based on RSI & MACD indicators using MEXC data."
 ).launch()