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 brief explanation per score
+- A detailed explanation per score
 - A summary of strengths and improvements
-- One motivational line
+- One motivational line, do not add the motivational keyword, just add it
 
 Current text:
 {text}
@@ -39,7 +39,16 @@ Current text:
         ],
         temperature=0.7
     )
-    return response.choices[0].message.content
+    feedback = response.choices[0].message.content
+
+    # Strip the JSON block from the feedback before returning to user
+    try:
+        split_idx = feedback.index('}') + 1
+        feedback_clean = feedback[split_idx:].strip()
+    except:
+        feedback_clean = feedback
+
+    return feedback, feedback_clean
 
 def improve_written_text(text, language):
     prompt = f"""Improve this writing by refining structure, grammar, and clarity — without changing its meaning.
@@ -77,22 +86,18 @@ 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"])
 
-        input_text = gr.Textbox(
-            label="📝 Your Writing",
-            lines=10,
-            elem_id="input_text",
-            elem_classes=["text_area"],
-            interactive=True
-        )
-
+        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")
         history_table = gr.Dataframe(headers=["🕒 Timestamp", "🌐 Language", "✍️ Text", "📋 Feedback"])
-        score_chart = gr.Plot(label="📊 Score Comparison")
-        trend_chart = gr.Plot(label="📈 Clarity Progress")
+
+        with gr.Row():
+            score_chart = gr.Plot(label="📊 Score Comparison")
+            trend_chart = gr.Plot(label="📈 Clarity Progress")
+
         milestone_box = gr.Markdown(visible=False)
 
         def extract_text(file):
@@ -113,23 +118,23 @@ def written_dashboard(nickname_input):
             previous_text = sessions[-1].transcript if sessions else None
             previous_feedback = sessions[-1].feedback if sessions else None
 
-            feedback = generate_written_feedback(text, language, goal, focus_areas, previous_text)
+            full_feedback, feedback_clean = generate_written_feedback(text, language, goal, focus_areas, previous_text)
             improved = improve_written_text(text, language)
 
-            save_to_db(nickname, text, feedback, language)
+            save_to_db(nickname, text, full_feedback, language)
             sessions = fetch_user_sessions(nickname)
             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(feedback, previous_feedback)) if previous_feedback else render_empty_chart("📊 Score Comparison")
+            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")
 
             milestone_msg = ""
             if len(sessions) in [3, 5, 10]:
                 milestone_msg = f"🎉 You’ve completed **{len(sessions)} writing sessions**!"
-                feedback += f"\n\n{milestone_msg}"
+                feedback_clean += f"\n\n{milestone_msg}"
 
-            return feedback, improved, session_table, score_plot, trend_plot, gr.update(visible=bool(milestone_msg), value=milestone_msg)
+            return feedback_clean, improved, session_table, score_plot, trend_plot, gr.update(visible=bool(milestone_msg), value=milestone_msg)
 
         submit_button.click(
             fn=tutor_written,
@@ -137,4 +142,4 @@ def written_dashboard(nickname_input):
             outputs=[feedback_box, improved_box, history_table, score_chart, trend_chart, milestone_box]
         )
 
-    return written_panel
+    return written_panel