Update written_module.py

written_module.py

@@ -1,45 +1,49 @@
 import gradio as gr
 import uuid
+import matplotlib.pyplot as plt
 from app_utils import (
-    LANG_CODES, client, save_to_db, fetch_user_sessions
+    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):
-    prompt = f"""You are a writing tutor. Please respond in [language={language}].
-Evaluate the user's writing on the following 6 criteria:
-1. Clarity & Conciseness
-2. Organization & Flow
-3. Grammar & Mechanics
-4. Vocabulary & Style
-5. Tone Appropriateness
-6. Originality & Creativity
+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"
+    history_section = f"\n\nCompare this to their previous writing:\n{previous_text}" if previous_text else ""
 
-For each:
-- Give a score out of 10
-- Write a short explanation
+    prompt = f"""
+You are a supportive writing coach helping someone working on: **{goal}**.
 
-Then provide:
-- An overall summary
-- One motivational closing line
+First, return a JSON object of the scores (0–10) for these categories:
+{focus_str}
 
-Writing:
+Then, explain each score briefly.
+
+End with:
+- A summary of strengths and improvement areas
+- One motivational sentence
+
+Text:
 {text}
+{history_section}
 """
+
     response = client.chat.completions.create(
         model="gpt-4",
         messages=[
-            {"role": "system", "content": f"You are a supportive writing coach responding in {language}."},
+            {"role": "system", "content": f"You are a thoughtful writing tutor responding in {language}."},
             {"role": "user", "content": prompt}
         ],
         temperature=0.7
     )
     return response.choices[0].message.content
 
-
 def improve_written_text(text, language):
-    prompt = f"""You are a writing coach. Rewrite this piece to improve structure, grammar, and clarity.
-Make it more compelling while keeping the original meaning.
+    prompt = f"""Rewrite this piece to improve structure, grammar, clarity, and tone without changing its meaning.
 
 Text:
 {text}
@@ -53,29 +57,46 @@ Text:
     )
     return response.choices[0].message.content
 
-
-def written_dashboard():
+def written_dashboard(nickname):
     with gr.Column() as written_panel:
         gr.Markdown("""
-        <div id=\"header\" style=\"text-align: center;\">
-            <img src=\"file/images/chatter_owl.png\" width=\"120\">
+        <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 your writing or upload a file, select language, and I’ll help you sharpen your message.</p>
+            <p>Paste or upload writing, select your goal and focus, and receive tailored feedback and trends over time.</p>
         </div>
         """)
 
-        nickname_box = gr.Textbox(label="👤 Your Nickname", placeholder="Enter your name…")
         language_dropdown = gr.Dropdown(label="🌍 Select Language", choices=list(LANG_CODES.keys()), value="English")
-        font_size_slider = gr.Slider(12, 24, value=16, step=1, label="🔠 Font Size")
-        contrast_toggle = gr.Checkbox(label="🌓 High Contrast Text")
+
+        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():
             input_text = gr.Textbox(lines=7, label="📝 Your Writing", elem_id="input_text", elem_classes=["text_area"])
-            file_input = gr.File(label="📄 Upload Text File", file_types=[".txt", ".md"]) 
+            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")
+        milestone_box = gr.Markdown(visible=False)
 
         def extract_text(file):
             if file is None:
@@ -85,30 +106,42 @@ def written_dashboard():
 
         file_input.change(fn=extract_text, inputs=file_input, outputs=input_text)
 
-        def tutor_written(text, language, nickname):
+        def tutor_written(text, language, goal, focus_areas):
             if not text.strip():
-                return "", "", []
+                return "", "", [], None, None, gr.update(visible=False)
 
-            feedback = generate_written_feedback(text, language)
-            improved = improve_written_text(text, language)
+            sessions = fetch_user_sessions(nickname)
+            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)
+            improved = improve_written_text(text, 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]
 
-            return feedback, improved, session_table
+            # 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")
+            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**!"
+                feedback += f"\n\n{milestone_msg}"
+
+            return feedback, improved, session_table, chart_plot, trend_plot, gr.update(visible=bool(milestone_msg), value=milestone_msg)
 
         input_text.change(
             fn=tutor_written,
-            inputs=[input_text, language_dropdown, nickname_box],
-            outputs=[feedback_box, improved_box, history_table]
+            inputs=[input_text, language_dropdown, goal_dropdown, focus_checkboxes],
+            outputs=[feedback_box, improved_box, history_table, score_chart, trend_chart, milestone_box]
         )
 
-        def update_styles(font_size, high_contrast):
-            # Dummy update to trigger refresh, can't update style directly in gr.update
-            return "", "", ""
-
-        font_size_slider.change(update_styles, [font_size_slider, contrast_toggle], [input_text, feedback_box, improved_box])
-        contrast_toggle.change(update_styles, [font_size_slider, contrast_toggle], [input_text, feedback_box, improved_box])
-
-        return written_panel
+    return written_panel