Update app.py

app.py

@@ -3,6 +3,10 @@ import os
 import datetime
 import json
 import tempfile
+import matplotlib.pyplot as plt
+import io
+import base64
+
 
 # === Import Groq API ===
 from groq import Groq
@@ -111,6 +115,76 @@ def get_history():
     return "\n".join([f"[{h['timestamp']}] {h['input']} → {h['response']}" for h in history])
 
 
+# === Progress Functions ===
+def calculate_progress():
+    # Load history
+    with open(HISTORY_FILE, "r") as f:
+        history = json.load(f)
+
+    if not history:
+        return "No progress yet! Try practicing first.", None, None, "🔥 0-day streak"
+
+    # --- Error Trend (Mistakes over time) ---
+    dates = [h["timestamp"].split(" ")[0] for h in history]
+    mistake_counts = [len(h["response"].split()) - len(h["input"].split()) for h in history]  # rough proxy
+
+    plt.figure(figsize=(5,3))
+    plt.plot(dates, mistake_counts, marker="o")
+    plt.xticks(rotation=45)
+    plt.title("Mistakes Over Time")
+    plt.tight_layout()
+    trend_img = save_plot_to_base64()
+
+    # --- Error Categories ---
+    categories = {"Grammar":0, "Spelling":0, "Vocabulary":0}
+    for h in history:
+        resp = h["response"].lower()
+        if "grammar" in resp: categories["Grammar"] += 1
+        if "spelling" in resp: categories["Spelling"] += 1
+        if "word" in resp: categories["Vocabulary"] += 1
+
+    plt.figure(figsize=(4,3))
+    plt.bar(categories.keys(), categories.values(), color="orange")
+    plt.title("Error Categories")
+    plt.tight_layout()
+    cat_img = save_plot_to_base64()
+
+    # --- Streak Counter ---
+    practiced_days = sorted(set(dates))
+    streak = 1
+    for i in range(len(practiced_days)-1, 0, -1):
+        d1 = datetime.datetime.strptime(practiced_days[i], "%Y-%m-%d")
+        d2 = datetime.datetime.strptime(practiced_days[i-1], "%Y-%m-%d")
+        if (d1 - d2).days == 1:
+            streak += 1
+        else:
+            break
+
+    return "📊 Progress Overview", trend_img, cat_img, f"🔥 {streak}-day streak"
+
+
+def save_plot_to_base64():
+    buf = io.BytesIO()
+    plt.savefig(buf, format="png")
+    buf.seek(0)
+    encoded = base64.b64encode(buf.read()).decode("utf-8")
+    buf.close()
+    return "data:image/png;base64," + encoded
+
+
+# === Updated Progress Tab ===
+with gr.Tab("Progress 📈"):
+    gr.Markdown("### Your Learning Progress")
+
+    progress_label = gr.Label()
+    trend_plot = gr.Image(type="pil", label="Error Trend")
+    cat_plot = gr.Image(type="pil", label="Error Categories")
+    streak_label = gr.Label()
+
+    load_btn = gr.Button("Load Progress")
+    load_btn.click(calculate_progress, outputs=[progress_label, trend_plot, cat_plot, streak_label])
+
+
 # === Gradio Interface ===
 with gr.Blocks() as demo:
     gr.Markdown("# 🌐✍️ Language Buddy")
@@ -137,8 +211,20 @@ with gr.Blocks() as demo:
         refresh_btn = gr.Button("Refresh History")
         refresh_btn.click(get_history, outputs=history_box)
 
+    # with gr.Tab("Progress 📈"):
+    #     gr.Markdown("Error tracking & progress visualization coming soon!")
     with gr.Tab("Progress 📈"):
-        gr.Markdown("Error tracking & progress visualization coming soon!")
+    gr.Markdown("### 📊 Your Progress Overview")
+
+    with gr.Row():
+        error_plot = gr.Image(type="pil", label="Error Trend")
+        error_stats = gr.Textbox(label="Error Categories", interactive=False, lines=6)
+
+    streak_counter = gr.Textbox(label="🔥 Current Streak", interactive=False)
+
+    refresh_progress = gr.Button("Refresh Progress")
+    refresh_progress.click(calculate_progress, outputs=[error_plot, error_stats, streak_counter])
+
 
 
 # === Launch ===