Update spoken_module.py

spoken_module.py

@@ -1,12 +1,15 @@
 import gradio as gr
 import uuid
+import matplotlib.pyplot as plt
 from gtts import gTTS
 from datetime import datetime
 from sqlmodel import Session
 from app_utils import (
     model, LANG_CODES, client, save_to_db, fetch_user_sessions,
     convert_to_wav, transcribe_audio, generate_feedback, generate_example_response,
-    parse_scores_from_feedback, generate_progress_summary
+    parse_scores_from_feedback, generate_progress_summary,
+    build_score_comparison_data, render_score_chart,
+    build_trend_data, render_trend_chart
 )
 
 def spoken_dashboard(nickname):
@@ -53,15 +56,13 @@ def spoken_dashboard(nickname):
         example_box = gr.Textbox(label="🗣 Suggested Improvement", visible=True, placeholder="Click to generate improved speech...")
         history_table = gr.Dataframe(headers=["🕒 Timestamp", "🌐 Language", "📝 Transcript (Preview)", "📋 Feedback (Preview)"])
 
-        def fetch_last_transcript(n):
-            if not n:
-                return None
-            sessions = fetch_user_sessions(n)
-            return sessions[-1].transcript if sessions else None
+        comparison_plot = gr.Plot(label="📊 Current vs Previous Scores")
+        trend_plot = gr.Plot(label="📈 Tone Progress Over Time")
+        milestone_box = gr.Markdown(visible=False)
 
         def tutor_feedback(audio_file, language, goal, focus_areas):
             if not audio_file:
-                return "", "No audio received.", None, "", []
+                return "", "No audio received.", None, "", [], None, None, gr.update(visible=False)
 
             wav_path = convert_to_wav(audio_file)
             transcript = transcribe_audio(wav_path)
@@ -69,7 +70,7 @@ def spoken_dashboard(nickname):
             previous_transcript = previous_sessions[-1].transcript if previous_sessions else None
             previous_feedback = previous_sessions[-1].feedback if previous_sessions else None
 
-            # Get feedback from GPT
+            # Get new feedback
             feedback_text = generate_feedback(
                 transcript=transcript,
                 language=language,
@@ -78,39 +79,66 @@ def spoken_dashboard(nickname):
                 previous_transcript=previous_transcript
             )
 
-            # Append progress tracker if available
+            # Progress summary
             if previous_feedback:
                 progress_summary = generate_progress_summary(feedback_text, previous_feedback)
                 feedback_text += progress_summary
 
-            # Text-to-Speech
+            # Milestone message
+            session_count = len(previous_sessions) + 1
+            milestone_message = ""
+            if session_count in [3, 5, 10]:
+                milestone_message = f"🎉 Congrats! You’ve completed **{session_count} sessions**!"
+                feedback_text += f"\n\n{milestone_message}"
+
+            # TTS
             lang_code = LANG_CODES.get(language, "en")
             tts = gTTS(feedback_text, lang=lang_code)
             mp3_path = f"/tmp/{uuid.uuid4()}.mp3"
             tts.save(mp3_path)
 
-            # Save session
-            if nickname:
-                save_to_db(nickname, transcript, feedback_text, language)
-
+            # Save new session
+            save_to_db(nickname, transcript, feedback_text, language)
             sessions = fetch_user_sessions(nickname)
             session_table = [[s.timestamp, s.language, s.transcript[:40], s.feedback[:40]] for s in sessions]
 
-            return transcript, feedback_text, mp3_path, transcript, session_table
+            # Charts
+            score_chart, trend_chart = None, None
+            if previous_feedback:
+                chart_data = build_score_comparison_data(feedback_text, previous_feedback)
+                score_chart = render_score_chart(chart_data)
+
+            if sessions:
+                dates, tone_scores = build_trend_data(sessions, category="Tone")
+                if tone_scores:
+                    trend_chart = render_trend_chart(dates, tone_scores, "Tone")
+
+            return (
+                transcript, feedback_text, mp3_path,
+                transcript, session_table,
+                score_chart, trend_chart,
+                gr.update(visible=bool(milestone_message), value=milestone_message)
+            )
 
         audio_input.change(
             fn=tutor_feedback,
             inputs=[audio_input, language_dropdown, goal_dropdown, focus_checkboxes],
-            outputs=[transcript_box, feedback_box, audio_output, hidden_transcript, history_table],
+            outputs=[
+                transcript_box, feedback_box, audio_output, hidden_transcript,
+                history_table, comparison_plot, trend_plot, milestone_box
+            ],
             show_progress="minimal"
         )
 
-        try_again.click(fn=lambda: ("", "", None, "", "", []),
+        try_again.click(fn=lambda: ("", "", None, "", "", None, None, gr.update(visible=False)),
                         inputs=None,
-                        outputs=[transcript_box, feedback_box, audio_output, hidden_transcript, example_box, history_table])
+                        outputs=[
+                            transcript_box, feedback_box, audio_output, hidden_transcript,
+                            example_box, comparison_plot, trend_plot, milestone_box
+                        ])
 
         show_example.click(fn=generate_example_response,
                            inputs=[hidden_transcript, language_dropdown],
                            outputs=example_box)
 
-    return spoken_panel
+    return spoken_panel