app.py

import gradio as gr
from stock import *
from indices_html import *
from index_live_html import *
from preopen_html import *
from eq_html import *
import pandas as pd

# ======================================================
# Scrollable HTML wrapper
# ======================================================
SCROLL_WRAP = """
<div style="
    max-height: 80vh;
    overflow-y: auto;
    overflow-x: auto;
    padding: 10px;
    border: 1px solid #ccc;
    border-radius: 6px;
">
{{HTML}}
</div>
"""

def wrap(html):
    if html is None:
        return "<h3>No Data</h3>"
    return SCROLL_WRAP.replace("{{HTML}}", html)


# ======================================================
# Request Type Options
# ======================================================
STOCK_REQ = [
    "info", "intraday", "daily","nse_eq", "qresult", "result", "balance",
    "cashflow", "dividend", "split", "other"
]

INDEX_REQ = [
    "indices", "nse_open", "nse_preopen", "nse_fno",
    "nse_future", "nse_bhav", "nse_highlow"
]


# ======================================================
# Update Dropdown based on mode
# ======================================================
def update_on_mode(mode):
    if mode == "stock":
        return gr.update(choices=STOCK_REQ, value="info")
    elif mode == "index":
        return gr.update(choices=INDEX_REQ, value="indices")
    return gr.update(choices=[], value="")


# ======================================================
# Data Fetcher (no defaults, use exactly frontend input)
# ======================================================
def fetch_data(mode, req_type, name, date_str):
    req_type = req_type.lower()
    name = name.strip()
    date_str = date_str.strip()

    if mode == "index":
        if req_type == "indices":
            return build_indices_html()

        elif req_type == "nse_open":
            return build_index_live_html(name)
        elif req_type == "nse_preopen":
            return build_preopen_html()
        elif req_type == "nse_fno":
            return wrap(nse_fno(name))
        elif req_type == "nse_future":
            return wrap(nse_future(name))
        elif req_type == "nse_bhav":
            return wrap(nse_bhav(date_str))  # no default
        elif req_type == "nse_highlow":
            return wrap(nse_highlow())
        else:
            return wrap(f"<h3>No handler for {req_type}</h3>")

    elif mode == "stock":
        if req_type == "daily":
            return wrap(fetch_daily(name))
        elif req_type == "nse_eq":
            return build_eq_html(name)
        elif req_type == "intraday":
            return wrap(fetch_intraday(name))
        elif req_type == "info":
            return wrap(fetch_info(name))
        elif req_type == "qresult":
            return wrap(fetch_qresult(name))
        elif req_type == "result":
            return wrap(fetch_result(name))
        elif req_type == "balance":
            return wrap(fetch_balance(name))
        elif req_type == "cashflow":
            return wrap(fetch_cashflow(name))
        elif req_type == "dividend":
            return wrap(fetch_dividend(name))
        elif req_type == "split":
            return wrap(fetch_split(name))
        elif req_type == "other":
            return wrap(fetch_other(name))
        else:
            return wrap(f"<h3>No handler for {req_type}</h3>")

    return wrap(f"<h3>No valid mode: {mode}</h3>")


# ======================================================
# UI
# ======================================================
with gr.Blocks(title="Stock / Index App") as iface:

    gr.Markdown("### **Stock / Index Data Fetcher**")

    with gr.Row():
        mode_input = gr.Radio(
            ["stock", "index"],
            label="Mode",
            value="stock",
            scale=1
        )

        name_input = gr.Textbox(
            label="Symbol / Index Name",
            value="ITC",
            placeholder="Enter stock symbol or index",
            scale=2
        )

        req_type_input = gr.Dropdown(
            label="Request Type",
            choices=STOCK_REQ + INDEX_REQ,
            value="info",
            allow_custom_value=True,
            scale=2
        )

        date_input = gr.Textbox(
            label="Date",
            value="",
            placeholder="DD-MM-YYYY",
            scale=1
        )

        fetch_btn = gr.Button("Fetch", scale=1)

    output = gr.HTML(label="Output")

    # Update dropdown choices when mode changes
    mode_input.change(
        update_on_mode,
        inputs=mode_input,
        outputs=req_type_input
    )

    # Fetch button
    fetch_btn.click(
        fetch_data,
        inputs=[mode_input, req_type_input, name_input, date_input],
        outputs=output
    )


# ======================================================
# Launch
# ======================================================
if __name__ == "__main__":
    iface.launch(server_name="0.0.0.0", server_port=7860)