Update app.py

app.py

@@ -189,14 +189,9 @@ def handle_custom_question(question):
         lowest_year = liabilities_by_year.idxmin()
         lowest_value = liabilities_by_year.min()
         return f"The year with the least amount of liabilities was {lowest_year} with ${lowest_value:,.2f}.", None
-    if "assets" in q and "graph" in q:
-        assets = df[df["tlabel"].astype(str).str.contains("total assets", case=False, na=False)]
-        grouped = assets.groupby("quarter")["value"].sum().sort_index().tail(8)
-        fig, ax = plt.subplots()
-        grouped.plot(kind="bar", ax=ax)
-        ax.set_title("Total Assets by Quarter (Last 2 Years)")
-        ax.set_ylabel("USD ($)")
-        return "Here is the total assets by quarter for the last 2 years.", to_img(fig)
+    if "asset" in q and ("graph" in q or "plot" in q or "quarter" in q):
+        image = plot_assets_by_quarter_last_two_years()
+        return "Here is the total assets by quarter for the last 2 years.", image
 
     # === Free Cash Flow trend (last 8 quarters) ===
     if "free cash flow" in question.lower() or "fcf" in question.lower():