Update written_module.py

written_module.py

@@ -41,7 +41,7 @@ Current text:
     )
     feedback = response.choices[0].message.content
 
-    # Strip the JSON block from the feedback before returning to user
+    # Strip JSON before returning
     try:
         split_idx = feedback.index('}') + 1
         feedback_clean = feedback[split_idx:].strip()
@@ -86,23 +86,26 @@ 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")
 
         file_input = gr.File(label="📄 Upload Text File", file_types=[".txt", ".md"])
-        feedback_box = gr.Textbox(label="💡 Feedback", interactive=False, elem_id="feedback_box")
-        improved_box = gr.Textbox(label="🎯 Improved Version", interactive=False, elem_id="improved_box")
+        feedback_box = gr.Textbox(label="💡 Feedback", interactive=False)
+        improved_box = gr.Textbox(label="🎯 Improved Version", interactive=False)
         history_table = gr.Dataframe(headers=["🕒 Timestamp", "🌐 Language", "✍️ Text", "📋 Feedback"])
 
+        # === Chart Area ===
         with gr.Row():
-            score_chart = gr.Plot(label="📊 Score Comparison")
-            trend_chart = gr.Plot(label="📈 Progress Over Time")
+            with gr.Column(scale=1):
+                gr.Markdown("### 📊 Score Comparison")
+                score_chart = gr.Plot()
+            with gr.Column(scale=1):
+                trend_category_dropdown = gr.Dropdown(
+                    label="📈 Track Progress In",
+                    choices=["Clarity", "Organization", "Grammar", "Vocabulary", "Tone", "Creativity"],
+                    value="Clarity"
+                )
+                trend_chart = gr.Plot(label="Progress Over Time")
 
         milestone_box = gr.Markdown(visible=False)
 
@@ -148,4 +151,19 @@ def written_dashboard(nickname_input):
             outputs=[feedback_box, improved_box, history_table, score_chart, trend_chart, milestone_box]
         )
 
-    return written_panel
+        def update_trend_chart(trend_category, nickname):
+            if hasattr(nickname, "value"):
+                nickname = nickname.value
+            sessions = fetch_user_sessions(nickname)
+            dates, trend_scores = build_trend_data(sessions, trend_category)
+            if trend_scores:
+                return render_trend_chart(dates, trend_scores, trend_category)
+            return render_empty_chart(f"📈 {trend_category} Progress")
+
+        trend_category_dropdown.change(
+            fn=update_trend_chart,
+            inputs=[trend_category_dropdown, nickname_input],
+            outputs=[trend_chart]
+        )
+
+    return written_panel