Update written_module.py

written_module.py

@@ -2,41 +2,39 @@ import gradio as gr
 import uuid
 import os
 import matplotlib.pyplot as plt
+from openai import OpenAI
 from app_utils import (
     LANG_CODES, save_to_db, fetch_user_sessions,
     parse_scores_from_feedback, generate_progress_summary,
     build_score_comparison_data, render_score_chart,
     build_trend_data, render_trend_chart
 )
-from openai import OpenAI
 
 client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 def generate_written_feedback(text, language, goal="general improvement", focus_areas=None, previous_text=None):
-    focus_str = ", ".join(focus_areas) if focus_areas else "Clarity, Structure, Grammar, Style, Tone, and Creativity"
+    focus_str = ", ".join(focus_areas) if focus_areas else "Clarity, Structure, Grammar, Style, Tone, Creativity"
     history_section = f"\n\nCompare this to their previous writing:\n{previous_text}" if previous_text else ""
 
     prompt = f"""
-You are a supportive writing coach helping someone working on: **{goal}**.
+You are a writing coach helping a student improve their writing for the following goal: **{goal}**.
 
-First, return a JSON object of the scores (0–10) for these categories:
+First, return a JSON object with scores (0–10) for each of these:
 {focus_str}
 
-Then, explain each score briefly.
-
-End with:
-- A summary of strengths and improvement areas
-- One motivational sentence
+Then write:
+- A brief explanation per score
+- A summary of strengths and improvements
+- One motivational line
 
-Text:
+Current text:
 {text}
 {history_section}
 """
-
     response = client.chat.completions.create(
         model="gpt-4",
         messages=[
-            {"role": "system", "content": f"You are a thoughtful writing tutor responding in {language}."},
+            {"role": "system", "content": f"You are a helpful writing tutor responding in {language}."},
             {"role": "user", "content": prompt}
         ],
         temperature=0.7
@@ -44,7 +42,7 @@ Text:
     return response.choices[0].message.content
 
 def improve_written_text(text, language):
-    prompt = f"""Rewrite this piece to improve structure, grammar, clarity, and tone without changing its meaning.
+    prompt = f"""Improve this writing by refining structure, grammar, and clarity — without changing its meaning.
 
 Text:
 {text}
@@ -64,39 +62,24 @@ def written_dashboard(nickname):
         <div id="header" style="text-align: center;">
             <img src="file/images/chatter_owl.png" width="120">
             <h2>✍️ Written Communication with <strong>Chatter the Owl</strong></h2>
-            <p>Paste or upload writing, select your goal and focus, and receive tailored feedback and trends over time.</p>
+            <p>Paste or upload your writing. Select your goal and focus, and receive detailed feedback, improvement, and progress tracking.</p>
         </div>
         """)
 
-        language_dropdown = gr.Dropdown(label="🌍 Select Language", choices=list(LANG_CODES.keys()), value="English")
-
-        goal_dropdown = gr.Dropdown(
-            label="🎯 Writing Goal",
-            choices=[
-                "Essay for school",
-                "Job application",
-                "Public blog post",
-                "Creative writing",
-                "General improvement"
-            ],
-            value="General improvement"
-        )
-
-        focus_checkboxes = gr.CheckboxGroup(
-            label="🧠 Areas to Focus On",
-            choices=["Clarity", "Organization", "Grammar", "Vocabulary", "Tone", "Creativity"],
-            value=["Clarity", "Organization", "Grammar", "Vocabulary", "Tone", "Creativity"]
-        )
+        with gr.Row():
+            language_dropdown = gr.Dropdown(label="🌍 Language", choices=list(LANG_CODES.keys()), value="English")
+            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"])
 
         with gr.Row():
-            input_text = gr.Textbox(lines=7, label="📝 Your Writing", elem_id="input_text", elem_classes=["text_area"])
+            input_text = gr.Textbox(label="📝 Your Writing", lines=8, elem_id="input_text", elem_classes=["text_area"])
             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", "✍️ Writing Preview", "📋 Feedback Preview"])
-        score_chart = gr.Plot(label="📊 Current vs Previous Scores")
-        trend_chart = gr.Plot(label="📈 Progress on Clarity Over Time")
+        history_table = gr.Dataframe(headers=["🕒 Timestamp", "🌐 Language", "✍️ Text", "📋 Feedback"])
+        score_chart = gr.Plot(label="📊 Score Comparison")
+        trend_chart = gr.Plot(label="📈 Clarity Progress")
         milestone_box = gr.Markdown(visible=False)
 
         def extract_text(file):
@@ -117,27 +100,21 @@ def written_dashboard(nickname):
 
             feedback = 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, feedback, language)
             sessions = fetch_user_sessions(nickname)
             session_table = [[s.timestamp, s.language, s.transcript[:40], s.feedback[:40]] for s in sessions]
 
-            # Score chart
-            chart_data = build_score_comparison_data(feedback, previous_feedback) if previous_feedback else None
-            chart_plot = render_score_chart(chart_data) if chart_data else None
-
-            # Trend chart
-            dates, clarity_scores = build_trend_data(sessions, category="Clarity")
+            score_plot = render_score_chart(build_score_comparison_data(feedback, previous_feedback)) if previous_feedback else None
+            dates, clarity_scores = build_trend_data(sessions, "Clarity")
             trend_plot = render_trend_chart(dates, clarity_scores, "Clarity") if clarity_scores else None
 
-            # Milestone
-            session_count = len(sessions)
             milestone_msg = ""
-            if session_count in [3, 5, 10]:
-                milestone_msg = f"🎉 You've completed **{session_count} writing sessions**!"
+            if len(sessions) in [3, 5, 10]:
+                milestone_msg = f"🎉 You’ve completed **{len(sessions)} writing sessions**!"
                 feedback += f"\n\n{milestone_msg}"
 
-            return feedback, improved, session_table, chart_plot, trend_plot, gr.update(visible=bool(milestone_msg), value=milestone_msg)
+            return feedback, improved, session_table, score_plot, trend_plot, gr.update(visible=bool(milestone_msg), value=milestone_msg)
 
         input_text.change(
             fn=tutor_written,