Update app.py

app.py

@@ -184,7 +184,7 @@ def handle_custom_question(question):
         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:
+    if "asset" in q and ("graph" in q or "plot" in q or "chart" in q)::
         assets = df[df["tlabel"].astype(str).str.contains("total assets", case=False, na=False)]
         grouped = assets.groupby("quarter")["value"].sum().last("8Q")
         fig, ax = plt.subplots()
@@ -335,6 +335,7 @@ def ask_question(history, question):
                 "embedding": embedding
             })
         time.sleep(2)
+        print("Current prompt history:", prompt_history)
         return history + [(question, custom_response)], custom_image if custom_image else gr.update()
 
     # === Fallback to generic plot if question mentions plot/graph/chart ===
@@ -372,6 +373,7 @@ def ask_question(history, question):
             })
 
         time.sleep(2)
+        print("Current prompt history:", prompt_history)
         return history + [(question, answer)], gr.update()
 
     except Exception as e: