Update app.py

app.py

@@ -1,7 +1,6 @@
 from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
-
 import os
 import importlib
 
@@ -17,7 +16,7 @@ for file in os.listdir(current_dir):
         except Exception as e:
             print(f"Failed to import {module_name}: {e}")
 
-
+# ---------- FastAPI app ----------
 app = FastAPI(title="Stock / Index Backend")
 
 # ---------- CORS ----------
@@ -57,49 +56,55 @@ def health():
 
 # ---------- STOCK Handlers ----------
 def handle_stock(req: FetchRequest):
-    if req.req_type not in STOCK_REQ:
-        raise HTTPException(status_code=400, detail=f"Invalid stock req_type: {req.req_type}")
+    try:
+        if req.req_type not in STOCK_REQ:
+            raise HTTPException(status_code=400, detail=f"Invalid stock req_type: {req.req_type}")
 
-    t = req.req_type.lower()
-    if t == "info": return wrap(yahooinfo.fetch_info(req.name))
-    if t == "intraday": return wrap(stock.fetch_intraday(req.name, req.date_start, req.date_end))
-    if t == "daily": return wrap(stock.fetch_daily(req.name, req.date_start, req.date_end))
-    if t == "nse_eq": return eq_html.build_eq_html(req.name)
-    if t == "qresult": return wrap(stock.fetch_qresult(req.name))
-    if t == "result": return wrap(stock.fetch_result(req.name))
-    if t == "balance": return wrap(stock.fetch_balance(req.name))
-    if t == "cashflow": return wrap(stock.fetch_cashflow(req.name))
-    if t == "dividend": return wrap(stock.fetch_dividend(req.name))
-    if t == "split": return wrap(stock.fetch_split(req.name))
-    if t == "other": return wrap(stock.fetch_other(req.name))
-    if t == "stock_hist": return nsepython.nse_stock_hist(req.date_start, req.date_end, req.name).to_html()
-    return wrap(f"<h3>No handler for stock req_type: {t}</h3>")
+        t = req.req_type.lower()
+        if t == "info": return common.wrap(yahooinfo.fetch_info(req.name))
+        if t == "intraday": return common.wrap(stock.fetch_intraday(req.name, req.date_start, req.date_end))
+        if t == "daily": return common.wrap(stock.fetch_daily(req.name, req.date_start, req.date_end))
+        if t == "nse_eq": return eq_html.build_eq_html(req.name)
+        if t == "qresult": return common.wrap(stock.fetch_qresult(req.name))
+        if t == "result": return common.wrap(stock.fetch_result(req.name))
+        if t == "balance": return common.wrap(stock.fetch_balance(req.name))
+        if t == "cashflow": return common.wrap(stock.fetch_cashflow(req.name))
+        if t == "dividend": return common.wrap(stock.fetch_dividend(req.name))
+        if t == "split": return common.wrap(stock.fetch_split(req.name))
+        if t == "other": return common.wrap(stock.fetch_other(req.name))
+        if t == "stock_hist": return nsepython.nse_stock_hist(req.date_start, req.date_end, req.name).to_html()
+        return common.wrap(f"<h3>No handler for stock req_type: {t}</h3>")
+    except Exception as e:
+        return {"error": str(e)}
 
 # ---------- INDEX Handlers ----------
 def handle_index(req: FetchRequest):
-    if req.req_type not in INDEX_REQ:
-        raise HTTPException(status_code=400, detail=f"Invalid index req_type: {req.req_type}")
+    try:
+        if req.req_type not in INDEX_REQ:
+            raise HTTPException(status_code=400, detail=f"Invalid index req_type: {req.req_type}")
 
-    t = req.req_type.lower()
-    if t == "indices": return indices_html.build_indices_html()
-    if t == "nse_open": return index_live_html.build_index_live_html()
-    if t == "nse_preopen": return preopen_html.build_preopen_html()
-    if t == "nse_fno": return build_nse_fno.nse_fno_html(req.date_end, req.name)
-    if t == "nse_fiidii": return nsepython.nse_fiidii().to_html()
-    if t == "nse_events": return nsepython.nse_events().to_html()
-    if t == "nse_future": return wrap(nsepython.nse_future(req.name))
-    if t == "nse_highlow": return nsepython.nse_highlow(req.date_end).to_html()
-    if t == "stock_highlow": return nsepython.stock_highlow(req.date_end).to_html()
-    if t == "nse_bhav": return bhavcopy_html.build_bhavcopy_html(req.date_end)
-    if t == "nse_largedeals": return nsepython.nse_largedeals().to_html()
-    if t == "nse_bulkdeals": return nsepython.nse_bulkdeals().to_html()
-    if t == "nse_blockdeals": return nsepython.nse_blockdeals().to_html()
-    if t == "nse_most_active": return nsepython.nse_most_active().to_html()
-    if t == "index_history": return nsepython.index_history("NIFTY", req.date_start, req.date_end).to_html()
-    if t == "largedeals_historical": return nsepython.nse_largedeals_historical(req.date_start, req.date_end).to_html()
-    if t == "index_pe_pb_div": return nsepython.index_pe_pb_div("NIFTY", req.date_start, req.date_end).to_html()
-    if t == "index_total_returns": return nsepython.index_total_returns("NIFTY", req.date_start, req.date_end).to_html()
-    return wrap(f"<h3>No handler for index req_type: {t}</h3>")
+        t = req.req_type.lower()
+        if t == "indices": return indices_html.build_indices_html()
+        if t == "nse_open": return index_live_html.build_index_live_html()
+        if t == "nse_preopen": return preopen_html.build_preopen_html()
+        if t == "nse_fno": return build_nse_fno.nse_fno_html(req.date_end, req.name)
+        if t == "nse_fiidii": return nsepython.nse_fiidii().to_html()
+        if t == "nse_events": return nsepython.nse_events().to_html()
+        if t == "nse_future": return common.wrap(nsepython.nse_future(req.name))
+        if t == "nse_highlow": return nsepython.nse_highlow(req.date_end).to_html()
+        if t == "stock_highlow": return nsepython.stock_highlow(req.date_end).to_html()
+        if t == "nse_bhav": return bhavcopy_html.build_bhavcopy_html(req.date_end)
+        if t == "nse_largedeals": return nsepython.nse_largedeals().to_html()
+        if t == "nse_bulkdeals": return nsepython.nse_bulkdeals().to_html()
+        if t == "nse_blockdeals": return nsepython.nse_blockdeals().to_html()
+        if t == "nse_most_active": return nsepython.nse_most_active().to_html()
+        if t == "index_history": return nsepython.index_history("NIFTY", req.date_start, req.date_end).to_html()
+        if t == "largedeals_historical": return nsepython.nse_largedeals_historical(req.date_start, req.date_end).to_html()
+        if t == "index_pe_pb_div": return nsepython.index_pe_pb_div("NIFTY", req.date_start, req.date_end).to_html()
+        if t == "index_total_returns": return nsepython.index_total_returns("NIFTY", req.date_start, req.date_end).to_html()
+        return common.wrap(f"<h3>No handler for index req_type: {t}</h3>")
+    except Exception as e:
+        return {"error": str(e)}
 
 # ---------- Main POST Endpoint ----------
 @app.post("/api/fetch")