Update app.py

app.py

@@ -67,6 +67,12 @@ def analyze_text(user_input):
 
         corrected_text = response.choices[0].message.content  # ✅ fixed
         
+        if "✅ Correct" in response or "No errors" in response:
+        update_daily_stats(is_correct=True)
+    else:
+        update_daily_stats(is_correct=False)
+
+    return response
         # Save history
         with open(HISTORY_FILE, "r+") as f:
             history = json.load(f)
@@ -169,57 +175,58 @@ def save_plot_to_base64():
     buf.close()
     return "data:image/png;base64," + encoded
 
+import datetime
 
-# === Daily Tracking ===
-DAILY_STATS = {
-    "date": str(datetime.date.today()),
-    "attempts": 0,
-    "correct": 0,
-    "errors": 0,
-    "current_streak": 0,
-    "badge": None
+# keep progress data in memory (you can later persist to file/db if needed)
+progress_data = {
+    "today": str(datetime.date.today()),
+    "correct_count": 0,
+    "incorrect_count": 0,
+    "streak": 0,
+    "badges": []
 }
 
 def update_daily_stats(is_correct: bool):
-    """Update stats after each attempt."""
+    """Update today's progress based on correctness."""
     today = str(datetime.date.today())
     
-    # reset if date changes
-    if DAILY_STATS["date"] != today:
-        DAILY_STATS.update({
-            "date": today,
-            "attempts": 0,
-            "correct": 0,
-            "errors": 0,
-            "current_streak": 0,
-            "badge": None
-        })
-
-    DAILY_STATS["attempts"] += 1
-    
+    # Reset stats if a new day has started
+    if progress_data["today"] != today:
+        progress_data["today"] = today
+        progress_data["correct_count"] = 0
+        progress_data["incorrect_count"] = 0
+        progress_data["streak"] = 0
+        progress_data["badges"] = []
+
     if is_correct:
-        DAILY_STATS["correct"] += 1
-        DAILY_STATS["current_streak"] += 1
-        if DAILY_STATS["current_streak"] >= 5:
-            DAILY_STATS["badge"] = "🏅 5-in-a-row Streak!"
+        progress_data["correct_count"] += 1
+        progress_data["streak"] += 1
+        # Award badge if streak hits 5
+        if progress_data["streak"] == 5:
+            progress_data["badges"].append("🏅 5-in-a-row Champion")
     else:
-        DAILY_STATS["errors"] += 1
-        DAILY_STATS["current_streak"] = 0  # reset streak
-    
-    return get_progress_summary()
+        progress_data["incorrect_count"] += 1
+        progress_data["streak"] = 0
+
 
 def get_progress_summary():
-    """Return formatted progress text + badge."""
-    text = (
-        f"📅 Date: {DAILY_STATS['date']}\n"
-        f"🔢 Attempts: {DAILY_STATS['attempts']}\n"
-        f"✅ Correct: {DAILY_STATS['correct']}\n"
-        f"❌ Errors: {DAILY_STATS['errors']}\n"
-        f"🔥 Streak: {DAILY_STATS['current_streak']}\n"
+    """Return a formatted summary for Gradio UI."""
+    today = progress_data["today"]
+    correct = progress_data["correct_count"]
+    incorrect = progress_data["incorrect_count"]
+    streak = progress_data["streak"]
+    badges = ", ".join(progress_data["badges"]) if progress_data["badges"] else "None yet"
+
+    summary = (
+        f"📅 Date: {today}\n"
+        f"✅ Correct: {correct}\n"
+        f"❌ Incorrect: {incorrect}\n"
+        f"🔥 Current Streak: {streak}\n"
+        f"🏅 Badges: {badges}"
     )
-    if DAILY_STATS["badge"]:
-        text += f"\n🏆 Badge Earned: {DAILY_STATS['badge']}"
-    return text
+    return summary
+
+
 
 # # === Inside Gradio UI ===
 # with gr.Tab("Progress 📈"):
@@ -260,12 +267,19 @@ with gr.Blocks() as demo:
     # with gr.Tab("Progress 📈"):
     #     gr.Markdown("Error tracking & progress visualization coming soon!")
     with gr.Tab("Progress 📈"):
-        gr.Markdown("### 📊 Your Daily Progress")
-        progress_box = gr.Textbox(label="Today’s Stats", interactive=False, lines=8)
+    gr.Markdown("### 📊 Daily Progress Tracker")
 
-        refresh_progress = gr.Button("🔄 Refresh Progress")
-        refresh_progress.click(get_progress_summary, outputs=progress_box)
-    
+    progress_box = gr.Textbox(
+        label="Today's Progress",
+        interactive=False,
+        lines=8,
+        value="No progress yet. Try practicing!"
+    )
+
+    refresh_btn = gr.Button("🔄 Refresh Progress")
+
+    # ✅ bind refresh button here
+    refresh_btn.click(get_progress_summary, outputs=progress_box)
 
 # === Launch ===
 demo.launch()