Update written_module.py

written_module.py

@@ -21,11 +21,11 @@ You are a writing coach helping a student improve their writing for the followin
 
 First, return a JSON object with scores (0–10) for each of these:
 {focus_str}
-Ensure that it is in a decent formatted style.
+
 Then write:
-- A detailed explanation per score
+- A brief explanation per score
 - A summary of strengths and improvements
-- One motivational line, do not add the motivational keyword, just add it
+- One motivational line
 
 Current text:
 {text}
@@ -86,6 +86,12 @@ def written_dashboard(nickname_input):
             goal_dropdown = gr.Dropdown(label="🎯 Writing Goal", choices=["Essay for school", "Job application", "Blog post", "Creative writing", "General improvement"], value="General improvement")
             focus_checkboxes = gr.CheckboxGroup(label="🧠 Focus Areas", choices=["Clarity", "Organization", "Grammar", "Vocabulary", "Tone", "Creativity"], value=["Clarity", "Organization", "Grammar", "Vocabulary", "Tone", "Creativity"])
 
+        trend_category_dropdown = gr.Dropdown(
+            label="📈 Track Progress In",
+            choices=["Clarity", "Organization", "Grammar", "Vocabulary", "Tone", "Creativity"],
+            value="Clarity"
+        )
+
         input_text = gr.Textbox(label="📝 Your Writing", lines=10, elem_id="input_text", elem_classes=["text_area"], interactive=True)
         submit_button = gr.Button("🚀 Submit", variant="primary", size="sm")
 
@@ -96,7 +102,7 @@ def written_dashboard(nickname_input):
 
         with gr.Row():
             score_chart = gr.Plot(label="📊 Score Comparison")
-            trend_chart = gr.Plot(label="📈 Clarity Progress")
+            trend_chart = gr.Plot(label="📈 Progress Over Time")
 
         milestone_box = gr.Markdown(visible=False)
 
@@ -108,11 +114,11 @@ def written_dashboard(nickname_input):
 
         file_input.change(fn=extract_text, inputs=file_input, outputs=input_text)
 
-        def tutor_written(text, language, goal, focus_areas, nickname):
+        def tutor_written(text, language, goal, focus_areas, trend_category, nickname):
             if hasattr(nickname, "value"):
                 nickname = nickname.value
             if not text.strip():
-                return "", "", [], render_empty_chart("📊 Score Comparison"), render_empty_chart("📈 Clarity Progress"), gr.update(visible=False)
+                return "", "", [], render_empty_chart("📊 Score Comparison"), render_empty_chart("📈 Progress Over Time"), gr.update(visible=False)
 
             sessions = fetch_user_sessions(nickname)
             previous_text = sessions[-1].transcript if sessions else None
@@ -126,8 +132,8 @@ def written_dashboard(nickname_input):
             session_table = [[s.timestamp, s.language, s.transcript[:40], s.feedback[:40]] for s in sessions]
 
             score_plot = render_score_chart(build_score_comparison_data(full_feedback, previous_feedback)) if previous_feedback else render_empty_chart("📊 Score Comparison")
-            dates, clarity_scores = build_trend_data(sessions, "Clarity")
-            trend_plot = render_trend_chart(dates, clarity_scores, "Clarity") if clarity_scores else render_empty_chart("📈 Clarity Progress")
+            dates, trend_scores = build_trend_data(sessions, trend_category)
+            trend_plot = render_trend_chart(dates, trend_scores, trend_category) if trend_scores else render_empty_chart(f"📈 {trend_category} Progress")
 
             milestone_msg = ""
             if len(sessions) in [3, 5, 10]:
@@ -138,7 +144,7 @@ def written_dashboard(nickname_input):
 
         submit_button.click(
             fn=tutor_written,
-            inputs=[input_text, language_dropdown, goal_dropdown, focus_checkboxes, nickname_input],
+            inputs=[input_text, language_dropdown, goal_dropdown, focus_checkboxes, trend_category_dropdown, nickname_input],
             outputs=[feedback_box, improved_box, history_table, score_chart, trend_chart, milestone_box]
         )