some more changes

Author: carlosabadia

stock_graph_app/stock_graph_app/components/stock_chart_display.py

@@ -115,8 +115,7 @@ def stock_header_component() -> rx.Component:
                     " ",
                     rx.el.span(
                         StockState.after_hours_change_display_val,
-                        class_name=StockState.after_hours_change_color
-                        + " font-semibold",
+                        class_name="text-neutral-500 font-semibold",
                     ),
                     rx.el.span(
                         StockState.company_info.get("currency", ""),

stock_graph_app/stock_graph_app/states/stock_state.py

@@ -1,7 +1,14 @@
+from typing import List, TypedDict
+
 import reflex as rx
 import yfinance as yf
 
 
+class HistoricalDataPoint(TypedDict):
+    name: str
+    price: float
+
+
 def format_market_cap(cap):
     if cap is None:
         return "N/A"
@@ -19,7 +26,7 @@ class StockState(rx.State):
 
     search_ticker_input: str = "AAPL"
     company_info: dict = {}
-    historical_data: list[dict] = []
+    historical_data: List[HistoricalDataPoint] = []
 
     is_loading: bool = False
     error_message: str = ""
@@ -208,19 +215,6 @@ def show_after_hours_section(self) -> bool:
         price, change, _ = self._after_hours_data
         return price is not None and change is not None
 
-    @rx.var
-    def after_hours_change_color(self) -> str:
-        _, change, _ = self._after_hours_data
-        if change is None:
-            return "text-neutral-500"
-        return (
-            "text-green-500"
-            if change > 0
-            else "text-red-500"
-            if change < 0
-            else "text-neutral-500"
-        )
-
     @rx.var
     def currency_code(self) -> str:
         """Returns the upper-cased currency code, defaulting to USD."""
@@ -239,7 +233,7 @@ def y_axis_label_config(self) -> dict:
         }
 
     @rx.var
-    def current_stock_data_for_chart(self) -> list[dict]:
+    def current_stock_data_for_chart(self) -> List[HistoricalDataPoint]:
         if not self.historical_data:
             return [{"name": "N/A", "price": 0}]
         return self.historical_data