Update app.py

app.py

@@ -1,8 +1,8 @@
 # =========================================================
 # TALib + mplfinance + Gradio
 # 1️⃣ Load Data (Internet only)
-# 2️⃣ Chart Type
-# 3️⃣ Pattern Marking
+# 2️⃣ Pattern Selection
+# 3️⃣ Build Candlestick Chart + Pattern Marking
 # =========================================================
 
 import yfinance as yf
@@ -22,7 +22,7 @@ PATTERN_MAP = {p.replace("CDL", ""): p for p in TALIB_PATTERNS}
 PATTERN_CHOICES = ["None"] + list(PATTERN_MAP.keys())
 
 # ---------------------------------------------------------
-# DATA CLEANING (WITH YOUR REQUESTED ADDITION)
+# DATA CLEANING (MultiIndex SAFE)
 # ---------------------------------------------------------
 def clean_ohlc(df):
     if df is None or df.empty:
@@ -30,14 +30,14 @@ def clean_ohlc(df):
 
     df = df.copy()
 
-    # ✅ YOUR REQUESTED FORMAT FIX
+    # ---- MultiIndex handling ----
     if isinstance(df.columns, pd.MultiIndex):
         df.columns = df.columns.get_level_values(0)
 
     df.columns = list(df.columns)
     df.index.name = None
 
-    # normalize names
+    # normalize column names
     df.columns = [str(c).lower().strip() for c in df.columns]
 
     # rename to mplfinance standard
@@ -69,7 +69,7 @@ def clean_ohlc(df):
     return df
 
 # ---------------------------------------------------------
-# 1️⃣ LOAD DATA (INTERNET ONLY)
+# 1️⃣ LOAD DATA (Internet only)
 # ---------------------------------------------------------
 def load_data(symbol, start_date, end_date):
     if not symbol:
@@ -87,18 +87,7 @@ def load_data(symbol, start_date, end_date):
         return None, f"❌ Load error: {e}"
 
 # ---------------------------------------------------------
-# 2️⃣ CHART TYPE CONFIG
-# ---------------------------------------------------------
-def chart_type_config(chart_type):
-    mapping = {
-        "Candlestick": ("candle", True),
-        "OHLC": ("ohlc", True),
-        "Line": ("line", False)
-    }
-    return mapping[chart_type]
-
-# ---------------------------------------------------------
-# 3️⃣ PATTERN MARKER BUILDER
+# 2️⃣ PATTERN MARKER BUILDER
 # ---------------------------------------------------------
 def build_pattern_addplots(df, pattern):
     if pattern == "None":
@@ -115,12 +104,13 @@ def build_pattern_addplots(df, pattern):
     )
 
     s = pd.Series(result, index=df.index)
-    aps = []
+    addplots = []
 
+    # bullish
     if (s > 0).any():
         bull = pd.Series(np.nan, index=df.index)
         bull[s > 0] = df["Low"][s > 0] * 0.98
-        aps.append(
+        addplots.append(
             mpf.make_addplot(
                 bull,
                 type="scatter",
@@ -130,10 +120,11 @@ def build_pattern_addplots(df, pattern):
             )
         )
 
+    # bearish
     if (s < 0).any():
         bear = pd.Series(np.nan, index=df.index)
         bear[s < 0] = df["High"][s < 0] * 1.02
-        aps.append(
+        addplots.append(
             mpf.make_addplot(
                 bear,
                 type="scatter",
@@ -143,40 +134,40 @@ def build_pattern_addplots(df, pattern):
             )
         )
 
-    return aps
+    return addplots
 
 # ---------------------------------------------------------
-# BUILD CHART (NO INTERNET CALL)
+# 3️⃣ BUILD CANDLESTICK CHART (NO INTERNET)
 # ---------------------------------------------------------
-def build_chart(df, symbol, chart_type, pattern):
+def build_chart(df, symbol, pattern):
     if df is None:
         return None, "❌ Load data first"
 
-    mpf_type, volume = chart_type_config(chart_type)
     addplots = build_pattern_addplots(df, pattern)
+    has_volume = "Volume" in df.columns
 
     os.makedirs("/tmp", exist_ok=True)
-    path = f"/tmp/{symbol}_{chart_type}_{pattern}.png"
+    path = f"/tmp/{symbol}_candlestick_{pattern}.png"
 
     fig, _ = mpf.plot(
         df,
-        type=mpf_type,
-        volume=volume,
-        addplot=addplots if mpf_type != "line" else None,
+        type="candle",
+        volume=has_volume,
+        addplot=addplots if addplots else None,
         style="yahoo",
-        title=f"{symbol} | {chart_type} | {pattern}",
+        title=f"{symbol} | Candlestick | {pattern}",
         figscale=1.8,
         returnfig=True
     )
 
     fig.savefig(path, dpi=150, bbox_inches="tight")
-    return path, "✅ Chart built successfully"
+    return path, "✅ Candlestick chart built successfully"
 
 # ---------------------------------------------------------
 # =================== GRADIO UI ===========================
 # ---------------------------------------------------------
 with gr.Blocks(fill_height=True) as app:
-    gr.Markdown("## 📊 Stock Chart Builder (3-Step Clean App)")
+    gr.Markdown("## 📊 Candlestick Pattern Analyzer (Clean 3-Step App)")
 
     state_df = gr.State(None)
 
@@ -196,13 +187,7 @@ with gr.Blocks(fill_height=True) as app:
     # ---------------- BUILD SECTION ----------------
     with gr.Row():
         with gr.Column(scale=1):
-            gr.Markdown("### 2️⃣ Chart Type")
-            chart_type = gr.Dropdown(
-                ["Candlestick", "OHLC", "Line"],
-                value="Candlestick"
-            )
-
-            gr.Markdown("### 3️⃣ Pattern")
+            gr.Markdown("### 2️⃣ Pattern")
             pattern = gr.Dropdown(
                 PATTERN_CHOICES,
                 value="None"
@@ -211,7 +196,7 @@ with gr.Blocks(fill_height=True) as app:
             build_btn = gr.Button("Build Chart", variant="primary")
 
         with gr.Column(scale=3):
-            chart = gr.Image(label="Chart Output", height=600)
+            chart = gr.Image(label="Candlestick Chart", height=600)
             build_status = gr.Textbox(label="Build Status")
 
     # ---------------- EVENTS ----------------
@@ -223,7 +208,7 @@ with gr.Blocks(fill_height=True) as app:
 
     build_btn.click(
         build_chart,
-        inputs=[state_df, symbol, chart_type, pattern],
+        inputs=[state_df, symbol, pattern],
         outputs=[chart, build_status]
     )