Update app.py

app.py

@@ -22,14 +22,49 @@ SCROLL_WRAP = """
 
 
 def wrap(html):
-    """Wrap returned HTML inside scrollable container"""
     if html is None:
         return "<h3>No Data</h3>"
     return SCROLL_WRAP.replace("{{HTML}}", html)
 
 
 # ======================================================
-# Data Fetcher
+# REQUEST TYPE OPTIONS
+# ======================================================
+STOCK_REQ = [
+    "info", "intraday", "daily", "qresult", "result", "balance",
+    "cashflow", "dividend", "split", "other"
+]
+
+INDEX_REQ = [
+    "nse_indices", "nse_open", "nse_preopen", "nse_fno",
+    "nse_future", "nse_bhav", "nse_highlow"
+]
+
+
+# ======================================================
+# UPDATE DROPDOWN + SYMBOL BASED ON MODE
+# ======================================================
+def update_on_mode(mode):
+    if mode == "stock":
+        return (
+            gr.update(choices=STOCK_REQ, value="info", visible=True),
+            gr.update(value="ITC")        # default stock symbol
+        )
+
+    elif mode == "index":
+        return (
+            gr.update(choices=INDEX_REQ, value="nse_indices", visible=True),
+            gr.update(value="NIFTY 50")   # default index name
+        )
+
+    return (
+        gr.update(visible=False),
+        gr.update(value="")
+    )
+
+
+# ======================================================
+# DATA FETCHER
 # ======================================================
 def fetch_data(mode, req_type, name):
     req_type = req_type.lower()
@@ -39,16 +74,18 @@ def fetch_data(mode, req_type, name):
 
         if req_type == "nse_indices":
             return wrap(nse_indices())
-
         elif req_type == "nse_open":
             return wrap(nse_open(name))
-
         elif req_type == "nse_preopen":
             return wrap(nse_preopen(name))
-
         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(name))
+        elif req_type == "nse_highlow":
+            return wrap(nse_highlow(name))
         else:
             return wrap(f"<h3>No handler for {req_type}</h3>")
 
@@ -56,34 +93,24 @@ def fetch_data(mode, req_type, name):
 
         if req_type == "daily":
             return wrap(fetch_daily(symbol))
-
         elif req_type == "intraday":
             return wrap(fetch_intraday(symbol))
-
         elif req_type == "info":
             return wrap(fetch_info(symbol))
-
         elif req_type == "qresult":
             return wrap(fetch_qresult(symbol))
-
         elif req_type == "result":
             return wrap(fetch_result(symbol))
-
         elif req_type == "balance":
             return wrap(fetch_balance(symbol))
-
         elif req_type == "cashflow":
             return wrap(fetch_cashflow(symbol))
-
         elif req_type == "dividend":
             return wrap(fetch_dividend(symbol))
-
         elif req_type == "split":
             return wrap(fetch_split(symbol))
-
         elif req_type == "other":
             return wrap(fetch_other(symbol))
-
         else:
             return wrap(f"<h3>No handler for {req_type}</h3>")
 
@@ -99,35 +126,39 @@ with gr.Blocks(title="Stock / Index App") as iface:
 
     with gr.Row():
 
-        mode_input = gr.Textbox(
+        mode_input = gr.SegmentedButton(
+            ["stock", "index"],
             label="Mode",
             value="stock",
-            placeholder="stock / index",
             scale=1
         )
 
         symbol = gr.Textbox(
             label="Symbol / Index Name",
-            value="PNB",
+            value="ITC",    # default for stock
             placeholder="Enter symbol",
             scale=2
         )
 
         req_type = gr.Dropdown(
             label="Request Type",
-            choices=[
-                "info","intraday","daily","qresult","result","balance","cashflow",
-                "dividend","split","nse_indices","nse_open","nse_preopen","nse_fno","nse_future","nse_bhav","nse_highlow"
-            ],
+            choices=STOCK_REQ,
             value="info",
             scale=2
         )
 
         btn = gr.Button("Fetch", scale=1)
 
-    # Output box (scroll handled by wrapper)
     output = gr.HTML(label="Output")
 
+    # Mode changes dropdown + symbol
+    mode_input.change(
+        update_on_mode,
+        inputs=mode_input,
+        outputs=[req_type, symbol]
+    )
+
+    # Fetch button
     btn.click(fetch_data, inputs=[mode_input, req_type, symbol], outputs=output)