Update app.py

app.py

@@ -145,17 +145,18 @@ def main_app():
     
             # Display label for the nearest goal
             st.write(f"Amount needed to reach goal '{closest_goal_name}': {closest_goal_amount:,} (Deadline: {closest_goal_deadline})")
-    
-        # Pie chart for income breakdown
-        labels = ['Expense', 'Savings', 'Unallocated Income']
-        sizes = [total_expense, allocated_savings, cash_in_hand]
-        sizes = [0 if np.isnan(value) else value for value in sizes]
-        colors = ['#ff9999','#66b3ff','#99ff99']
-        
-        fig, ax = plt.subplots()
-        ax.pie(sizes, labels=labels, autopct='%1.1f%%', startangle=90, colors=colors)
-        ax.axis('equal')  # Equal aspect ratio ensures that pie chart is drawn as a circle.
-        st.pyplot(fig)
+
+        if total_income > 0 and total_expense > 0:
+            # Pie chart for income breakdown
+            labels = ['Expense', 'Savings', 'Unallocated Income']
+            sizes = [total_expense, allocated_savings, cash_in_hand]
+            sizes = [0 if np.isnan(value) else value for value in sizes]
+            colors = ['#ff9999','#66b3ff','#99ff99']
+            
+            fig, ax = plt.subplots()
+            ax.pie(sizes, labels=labels, autopct='%1.1f%%', startangle=90, colors=colors)
+            ax.axis('equal')  # Equal aspect ratio ensures that pie chart is drawn as a circle.
+            st.pyplot(fig)
     
         # Progress bar for each goal
         for idx, goal in st.session_state["saving_goals"].iterrows():