Refactor Namma Metro main application to fetch ridership data from API

- Replaced local JSONL file reading with an API call to retrieve ridership data.
- Added a refresh button to update the displayed data dynamically.
- Updated the FastAPI title and welcome message for consistency across the application.

Author: traffic-kowshik

apps/namma_metro/main.py

@@ -1,26 +1,29 @@
+import os
+import requests
 import streamlit as st
-import json
 import pandas as pd
-from pathlib import Path
-
-from traffic_data_bengaluru.utils import *
-
-data_directory = get_data_directory()
-ridership_filepath = data_directory / "namma_metro" / "raw" / "ridership.jsonl"
+from dotenv import load_dotenv
+load_dotenv()
 
 st.title("🚇 Namma Metro")
 st.set_page_config(page_title="Namma Metro Ridership", layout="wide")
 
-records = []
-with open(ridership_filepath, "r") as f:
-    for line in f:
-        item = json.loads(line)
-        results = item.get("results", [])
-        if results:
-            records.append(results[0])
+base_url = os.environ.get("TDB_API_BASE_URL")
+url = f"{base_url}/metro/ridership"
+
+# Function to fetch data
+def fetch_data():
+    response = requests.get(url)
+    response.raise_for_status()
+    return pd.DataFrame(response.json())
+
+# Initialize session state
+if "df" not in st.session_state:
+    st.session_state.df = fetch_data()
+
+# Refresh button
+if st.button("Refresh"):
+    st.session_state.df = fetch_data()
 
-if not records:
-    st.warning("No ridership data found.")
-else:
-    df = pd.DataFrame(records).sort_values(by='RidershipDate', ascending=False)
-    st.dataframe(df)
+# Display the table
+st.dataframe(st.session_state.df)

nbs/apis/01_app.ipynb

@@ -39,12 +39,12 @@
    "source": [
     "#| export\n",
     "\n",
-    "app = FastAPI(title=\"Namma Traffic API\")\n",
+    "app = FastAPI(title=\"Traffic Data Bengaluru API\")\n",
     "app.include_router(metro_router)\n",
     "\n",
     "@app.get(\"/\")\n",
     "def root():\n",
-    "    return {\"message\": \"Welcome to Namma Traffic API\"}"
+    "    return {\"message\": \"Welcome to Traffic Data Bengaluru API\"}"
    ]
   },
   {

nbs/namma_metro/01_ridership.ipynb

@@ -378,9 +378,13 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "python3",
+   "display_name": ".venv",
    "language": "python",
    "name": "python3"
+  },
+  "language_info": {
+   "name": "python",
+   "version": "3.12.3"
   }
  },
  "nbformat": 4,

traffic_data_bengaluru/apis/app.py

@@ -8,9 +8,9 @@
 from .namma_metro import router as metro_router
 
 # %% ../../nbs/apis/01_app.ipynb 3
-app = FastAPI(title="Namma Traffic API")
+app = FastAPI(title="Traffic Data Bengaluru API")
 app.include_router(metro_router)
 
 @app.get("/")
 def root():
-    return {"message": "Welcome to Namma Traffic API"}
+    return {"message": "Welcome to Traffic Data Bengaluru API"}