Update app.py

app.py

@@ -265,29 +265,32 @@ def handle_custom_question(question):
 
     if "financial health" in q:
         latest_year = df["year"].max()
+    
+        # Filter relevant rows
         revenue = df[df["tlabel"].astype(str).str.contains("revenue", case=False, na=False)]
         net_income = df[df["tag"].astype(str).isin(["NetIncomeLoss", "NetIncome"])]
         assets = df[df["tag"].astype(str).str.contains("Assets", case=False, na=False)]
-        equity = df[df["tag"].astype(str).str.contains("Equity", case=False, na=False) & ~df["tag"].astype(str).str.contains("Liabilities")]
-        liabilities = df[df["tlabel"].astype(str).str.contains("total liabilities", case=False, na=False)]
-
+        equity = df[df["tag"].astype(str).str.contains("Equity", case=False, na=False) & ~df["tag"].astype(str).str.contains("Liabilities", case=False, na=False)]
+    
+        # Get latest values
         rev_latest = revenue[revenue["year"] == latest_year]["value"].sum()
         net_income_latest = net_income[net_income["year"] == latest_year]["value"].sum()
         assets_latest = assets[assets["year"] == latest_year]["value"].sum()
-        liabilities_latest = liabilities[liabilities["year"] == latest_year]["value"].sum()
         equity_latest = equity[equity["year"] == latest_year]["value"].sum()
-
+    
+        # Calculate metrics
         profit_margin = (net_income_latest / rev_latest * 100) if rev_latest > 0 else 0
-        debt_to_equity = (liabilities_latest / equity_latest) if equity_latest > 0 else 0
+        debt_to_equity = ((assets_latest - equity_latest) / equity_latest) if equity_latest > 0 else 0
         roa = (net_income_latest / assets_latest * 100) if assets_latest > 0 else 0
-
+    
+        # Classification
         if profit_margin > 15 and debt_to_equity < 1 and roa > 5:
             health = "Healthy"
         elif profit_margin > 5 and debt_to_equity < 2:
             health = "Watchlist"
         else:
             health = "Risky"
-
+    
         return (
             f"For {latest_year}, Apple is classified as **{health}** based on:\n"
             f"- Profit Margin: {profit_margin:.2f}%\n"