Update app.py

app.py

@@ -1,8 +1,8 @@
 """
-Robust TALib + mplfinance + Gradio example
-- UI shows pattern names WITHOUT 'CDL'
-- Internal TALib still uses CDL* functions
-- Vertical layout to maximize chart space
+TALib + mplfinance + Gradio
+- Clean dashboard layout
+- Pattern names without CDL
+- Max chart space
 """
 
 import yfinance as yf
@@ -13,29 +13,25 @@ import talib
 import gradio as gr
 from datetime import date
 import os
-from typing import Optional, Tuple, List
+from typing import Optional, List
 
-# ===============================
-# TALib pattern handling
-# ===============================
+# =====================================================
+# TALib pattern mapping (UI clean, backend intact)
+# =====================================================
 
 TALIB_PATTERNS = sorted([n for n in dir(talib) if n.startswith("CDL")])
-
-# UI name → TALib name mapping
 PATTERN_DISPLAY_MAP = {n.replace("CDL", ""): n for n in TALIB_PATTERNS}
 DISPLAY_PATTERNS = list(PATTERN_DISPLAY_MAP.keys())
 
-
-# ===============================
-# Utilities
-# ===============================
+# =====================================================
+# Data utilities
+# =====================================================
 
 def _normalize_col_name(col) -> str:
     if isinstance(col, (tuple, list)):
         return "_".join(str(c) for c in col if c).lower()
     return str(col).strip().lower()
 
-
 def _find_best_col(key: str, columns: List[str]) -> Optional[str]:
     if key in columns:
         return key
@@ -47,10 +43,9 @@ def _find_best_col(key: str, columns: List[str]) -> Optional[str]:
             return c
     return None
 
-
 def clean_ohlc(df: pd.DataFrame) -> pd.DataFrame:
     if df is None or not isinstance(df, pd.DataFrame):
-        raise ValueError("Invalid DataFrame")
+        raise ValueError("Invalid dataframe")
 
     df = df.copy()
     df.columns = [_normalize_col_name(c) for c in df.columns]
@@ -86,16 +81,12 @@ def clean_ohlc(df: pd.DataFrame) -> pd.DataFrame:
 
     return df
 
-
-# ===============================
+# =====================================================
 # Pattern detection
-# ===============================
+# =====================================================
 
 def find_candlestick_patterns(df, talib_pattern):
-    try:
-        func = getattr(talib, talib_pattern)
-    except AttributeError:
-        return [], f"TALib pattern not found: {talib_pattern}"
+    func = getattr(talib, talib_pattern)
 
     df = clean_ohlc(df)
 
@@ -115,28 +106,31 @@ def find_candlestick_patterns(df, talib_pattern):
         bull = pd.Series(np.nan, index=df.index)
         bull[s > 0] = df.loc[s > 0, "Low"] * 0.98
         apds.append(
-            mpf.make_addplot(bull, type="scatter", marker="^",
-                             markersize=80, color="green")
+            mpf.make_addplot(
+                bull, type="scatter", marker="^",
+                markersize=90, color="green", alpha=0.85
+            )
         )
 
     if (s < 0).any():
         bear = pd.Series(np.nan, index=df.index)
         bear[s < 0] = df.loc[s < 0, "High"] * 1.02
         apds.append(
-            mpf.make_addplot(bear, type="scatter", marker="v",
-                             markersize=80, color="red")
+            mpf.make_addplot(
+                bear, type="scatter", marker="v",
+                markersize=90, color="red", alpha=0.85
+            )
         )
 
     return apds, None
 
-
-# ===============================
+# =====================================================
 # Main plot function
-# ===============================
+# =====================================================
 
 def plot_stock_with_patterns(symbol, start, end, selected):
     if not symbol:
-        return None, "Symbol required"
+        return None, "Stock symbol required"
 
     try:
         df = yf.download(symbol, start=start, end=end, progress=False)
@@ -144,7 +138,7 @@ def plot_stock_with_patterns(symbol, start, end, selected):
         return None, str(e)
 
     if df.empty:
-        return None, "No data"
+        return None, "No data found"
 
     try:
         df_clean = clean_ohlc(df)
@@ -169,36 +163,67 @@ def plot_stock_with_patterns(symbol, start, end, selected):
         volume="Volume" in df_clean.columns,
         addplot=addplots if addplots else None,
         style="yahoo",
-        title=f"{symbol} Candlestick Chart",
-        returnfig=True,
-        figscale=1.6
+        title=f"{symbol} • Candlestick Chart",
+        figscale=1.7,
+        returnfig=True
     )
 
     fig.savefig(path, dpi=150, bbox_inches="tight")
-    return path, "Success"
-
-
-# ===============================
-# Gradio UI (VERTICAL)
-# ===============================
-
-with gr.Blocks(fill_height=True) as iface:
-    gr.Markdown("# 📊 TALib Candlestick Pattern Scanner")
-
-    symbol = gr.Textbox(label="Stock Symbol", value="MSFT")
-    start = gr.Textbox(label="Start Date", value="2024-01-01")
-    end = gr.Textbox(label="End Date", value=date.today().strftime("%Y-%m-%d"))
-
-    patterns = gr.CheckboxGroup(
-        label="Candlestick Pattern",
-        choices=DISPLAY_PATTERNS,
-        value=["HAMMER", "SHOOTINGSTAR"]
+    return path, "Chart generated successfully"
+
+# =====================================================
+# Gradio UI – Clean Dashboard Layout
+# =====================================================
+
+with gr.Blocks(fill_height=True, theme=gr.themes.Soft()) as iface:
+    gr.Markdown(
+        """
+        # 📊 TALib Candlestick Pattern Dashboard
+        **Bullish = Green ▲ | Bearish = Red ▼**
+        """
     )
 
-    run = gr.Button("Generate Chart", variant="primary")
-
-    chart = gr.Image(label="Chart", type="filepath", height=650)
-    status = gr.Textbox(label="Status")
+    with gr.Row():
+        with gr.Column(scale=1, min_width=320):
+            gr.Markdown("### ⚙️ Controls")
+
+            symbol = gr.Textbox(
+                label="Stock / Index Symbol",
+                value="MSFT",
+                placeholder="AAPL, MSFT, ^NSEI, ^GSPC"
+            )
+
+            start = gr.Textbox(
+                label="Start Date",
+                value="2024-01-01"
+            )
+
+            end = gr.Textbox(
+                label="End Date",
+                value=date.today().strftime("%Y-%m-%d")
+            )
+
+            patterns = gr.CheckboxGroup(
+                label="Candlestick Patterns",
+                choices=DISPLAY_PATTERNS,
+                value=["HAMMER", "SHOOTINGSTAR"],
+                interactive=True
+            )
+
+            run = gr.Button("📈 Generate Chart", variant="primary")
+
+            status = gr.Textbox(
+                label="Status",
+                interactive=False
+            )
+
+        with gr.Column(scale=3):
+            gr.Markdown("### 📉 Price Chart")
+            chart = gr.Image(
+                type="filepath",
+                height=700,
+                show_label=False
+            )
 
     run.click(
         plot_stock_with_patterns,