import customtkinter as ctk from database import Database import matplotlib.pyplot as plt from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg import datetime import numpy as np from utils import parse_float, get_bmi_status, get_bmi_icon class DashboardFrame(ctk.CTkFrame): def __init__(self, master, db: Database): super().__init__(master) self.db = db self.profile = self.db.get_user_profile() self.setup_ui() def setup_ui(self): self.clear_ui() self.grid_columnconfigure(0, weight=1) self.grid_rowconfigure(2, weight=1) # Header self.header = ctk.CTkLabel(self, text="Status & Prognosis", font=ctk.CTkFont(size=24, weight="bold")) self.header.grid(row=0, column=0, pady=20, sticky="n") # Check if profile exists if not self.profile: self.show_setup_form() else: self.show_dashboard() def clear_ui(self): for widget in self.winfo_children(): widget.destroy() def show_setup_form(self): frame = ctk.CTkFrame(self) frame.grid(row=1, column=0, pady=20) ctk.CTkLabel(frame, text="Welcome! Let's set up your profile.", font=ctk.CTkFont(size=16)).pack(pady=10) ctk.CTkLabel(frame, text="Height (cm):").pack(pady=5) self.height_entry = ctk.CTkEntry(frame, placeholder_text="e.g. 180,0") self.height_entry.pack(pady=5) ctk.CTkLabel(frame, text="Goal Weight (kg):").pack(pady=5) self.goal_entry = ctk.CTkEntry(frame, placeholder_text="e.g. 75,0") self.goal_entry.pack(pady=5) ctk.CTkButton(frame, text="Save Profile", command=self.save_profile).pack(pady=20) def save_profile(self): h = parse_float(self.height_entry.get()) g = parse_float(self.goal_entry.get()) if h is not None and g is not None: self.db.save_user_profile(h, g) self.profile = self.db.get_user_profile() self.setup_ui() else: ctk.CTkLabel(self, text="Invalid input. Use numbers (e.g. 180.5 or 180,5).", text_color="red").grid(row=2, column=0) def show_dashboard(self): # 1. Fetch Data logs = self.db.get_log_history(90) # Last 90 days for trends logs.reverse() # Oldest first latest_log = logs[-1] if logs else None current_weight = latest_log['weight'] if latest_log and latest_log['weight'] else None # --- Top Cards: BMI & Goal --- cards_frame = ctk.CTkFrame(self, fg_color="transparent") cards_frame.grid(row=1, column=0, sticky="ew", padx=20) cards_frame.grid_columnconfigure((0, 1, 2), weight=1) # BMI Card bmi_frame = ctk.CTkFrame(cards_frame) bmi_frame.grid(row=0, column=0, padx=10, sticky="ew") bmi_val_str = "N/A" bmi_status_text = "" bmi_color = "white" bmi_icon_img = None if current_weight: height_m = self.profile['height_cm'] / 100 bmi = current_weight / (height_m ** 2) bmi_val_str = f"{bmi:.1f}" bmi_status_text, _, bmi_color = get_bmi_status(bmi) bmi_icon_img = get_bmi_icon(bmi, size=(80, 80)) # Layout BMI Card # Left: Icon, Right: Text bmi_inner = ctk.CTkFrame(bmi_frame, fg_color="transparent") bmi_inner.pack(pady=10, padx=10) if bmi_icon_img: icon_label = ctk.CTkLabel(bmi_inner, text="", image=bmi_icon_img) icon_label.pack(side="left", padx=(0, 15)) text_container = ctk.CTkFrame(bmi_inner, fg_color="transparent") text_container.pack(side="left") ctk.CTkLabel(text_container, text="BMI", font=ctk.CTkFont(size=14)).pack(anchor="w") ctk.CTkLabel(text_container, text=bmi_val_str, font=ctk.CTkFont(size=30, weight="bold"), text_color=bmi_color).pack(anchor="w") ctk.CTkLabel(text_container, text=bmi_status_text, font=ctk.CTkFont(size=12)).pack(anchor="w") # Weight Card weight_frame = ctk.CTkFrame(cards_frame) weight_frame.grid(row=0, column=1, padx=10, sticky="ew") w_text = f"{current_weight} kg" if current_weight else "-- kg" g_text = f"Goal: {self.profile['goal_weight_kg']} kg" ctk.CTkLabel(weight_frame, text="Current Weight", font=ctk.CTkFont(size=14)).pack(pady=(10,0)) ctk.CTkLabel(weight_frame, text=w_text, font=ctk.CTkFont(size=30, weight="bold")).pack() ctk.CTkLabel(weight_frame, text=g_text, font=ctk.CTkFont(size=12, slant="italic")).pack(pady=(0,10)) # Date Estimate Card date_frame = ctk.CTkFrame(cards_frame) date_frame.grid(row=0, column=2, padx=10, sticky="ew") estimated_date_str = "-1 (Insuff. Data)" # Calculate Projection valid_logs = [l for l in logs if l['weight'] is not None] if len(valid_logs) >= 3: dates_num = [(datetime.datetime.strptime(l['date'], "%Y-%m-%d") - datetime.datetime.strptime(valid_logs[0]['date'], "%Y-%m-%d")).days for l in valid_logs] weights = [l['weight'] for l in valid_logs] # Linear Regression slope, intercept = np.polyfit(dates_num, weights, 1) # Time to goal goal = self.profile['goal_weight_kg'] if slope < 0 and current_weight > goal: # Losing weight towards goal days_to_goal = (goal - intercept) / slope target_date = datetime.datetime.strptime(valid_logs[0]['date'], "%Y-%m-%d") + datetime.timedelta(days=days_to_goal) estimated_date_str = target_date.strftime("%b %d, %Y") elif slope > 0 and current_weight < goal: # Gaining weight towards goal days_to_goal = (goal - intercept) / slope target_date = datetime.datetime.strptime(valid_logs[0]['date'], "%Y-%m-%d") + datetime.timedelta(days=days_to_goal) estimated_date_str = target_date.strftime("%b %d, %Y") else: estimated_date_str = "Trend mismatch" # Not moving towards goal ctk.CTkLabel(date_frame, text="Est. Goal Date", font=ctk.CTkFont(size=14)).pack(pady=(10,0)) ctk.CTkLabel(date_frame, text=estimated_date_str, font=ctk.CTkFont(size=20, weight="bold")).pack(pady=5) ctk.CTkLabel(date_frame, text="Based on recent trend", font=ctk.CTkFont(size=10)).pack(pady=(0,10)) # --- Graph --- self.graph_frame = ctk.CTkFrame(self) self.graph_frame.grid(row=2, column=0, sticky="nsew", padx=20, pady=20) self.plot_prognosis(valid_logs if 'valid_logs' in locals() else []) def plot_prognosis(self, logs): if not logs: return dates = [datetime.datetime.strptime(l['date'], "%Y-%m-%d") for l in logs] weights = [l['weight'] for l in logs] fig, ax = plt.subplots(figsize=(6, 4), dpi=100) fig.patch.set_facecolor('#2b2b2b') ax.set_facecolor('#2b2b2b') # Plot History ax.plot(dates, weights, color='tab:blue', marker='o', label='History') # Plot Goal Line ax.axhline(y=self.profile['goal_weight_kg'], color='tab:green', linestyle=':', label='Goal') ax.set_ylabel('Weight (kg)', color='white') ax.tick_params(axis='x', rotation=45, colors='white') ax.tick_params(axis='y', colors='white') ax.spines['bottom'].set_color('white') ax.spines['top'].set_color('none') ax.spines['left'].set_color('white') ax.spines['right'].set_color('none') fig.tight_layout() canvas = FigureCanvasTkAgg(fig, master=self.graph_frame) canvas.draw() canvas.get_tk_widget().pack(fill="both", expand=True)