Update app.py

app.py

@@ -58,6 +58,10 @@ def analyze_text(user_input):
         2. Provide a formal version.
         3. Provide an informal version.
         4. Highlight difficult/new words.
+
+        
+        Finally, output a JSON object in this format (at the very end, nothing else):
+        <stats>{{"correct": true or false}}</stats>
         """
 
         response = client.chat.completions.create(
@@ -68,13 +72,26 @@ def analyze_text(user_input):
         corrected_text = response.choices[0].message.content  # ✅ fixed
         # corrected_text = response.choices[0].message.content
 
-        
-        if "✅ Correct" in response or "No errors" in response:
-            update_daily_stats(is_correct=True)
-        else: 
-            update_daily_stats(is_correct=False)
+        # ✅ Extract correctness
+        is_correct = False
+        if "<stats>" in corrected_text and "</stats>" in corrected_text:
+            try:
+                stats_str = corrected_text.split("<stats>")[1].split("</stats>")[0]
+                stats = json.loads(stats_str)
+                is_correct = stats.get("correct", False)
+            except Exception:
+                is_correct = False
 
-        return corrected_text  # not return response
+        update_daily_stats(is_correct=is_correct)
+
+        # ✅ Clean output for user (remove hidden stats)
+        cleaned_text = corrected_text.split("<stats>")[0].strip()
+        # if "✅ Correct" in response or "No errors" in response:
+        #     update_daily_stats(is_correct=True)
+        # else: 
+        #     update_daily_stats(is_correct=False)
+
+        # return corrected_text  # not return response
 
         # Save history
         with open(HISTORY_FILE, "r+") as f:
@@ -82,12 +99,12 @@ def analyze_text(user_input):
             history.append({
                 "timestamp": datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
                 "input": user_input,
-                "response": corrected_text
+                "response": cleaned_text
             })
             f.seek(0)
             json.dump(history, f, indent=2)
 
-        return corrected_text
+        return cleaned_text
     except Exception as e:
         return f"❌ Error in analyze_text: {str(e)}"
 
@@ -178,7 +195,6 @@ def save_plot_to_base64():
     buf.close()
     return "data:image/png;base64," + encoded
 
-import datetime
 
 # keep progress data in memory (you can later persist to file/db if needed)
 progress_data = {