Update spoken_module.py

spoken_module.py

@@ -40,7 +40,6 @@ Transcript:
     )
     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()
@@ -85,6 +84,12 @@ def spoken_dashboard(nickname_input):
             goal_dropdown = gr.Dropdown(label="🎯 Goal", choices=["Interview preparation", "Public speaking", "Class presentation", "General improvement"], value="General improvement")
             focus_checkboxes = gr.CheckboxGroup(label="🧠 Focus Areas", choices=["Clarity", "Structure", "Fluency", "Tone", "Content Relevance"], value=["Clarity", "Structure", "Fluency", "Tone", "Content Relevance"])
 
+        trend_category_dropdown = gr.Dropdown(
+            label="📈 Track Progress In",
+            choices=["Clarity", "Structure", "Fluency", "Tone", "Content Relevance"],
+            value="Tone"
+        )
+
         with gr.Row():
             audio_input = gr.Audio(type="filepath", label="🎙 Speak or Upload Audio")
             audio_output = gr.Audio(label="🔊 Chatter's Response", type="filepath")
@@ -102,13 +107,13 @@ def spoken_dashboard(nickname_input):
 
         with gr.Row():
             score_chart = gr.Plot(label="📊 Score Comparison")
-            trend_chart = gr.Plot(label="📈 Tone Progress")
+            trend_chart = gr.Plot(label="📈 Progress Over Time")
 
         milestone_box = gr.Markdown(visible=False)
 
-        def tutor_feedback(audio_file, language, goal, focus_areas, nickname):
+        def tutor_feedback(audio_file, language, goal, focus_areas, trend_category, nickname):
             if not audio_file:
-                return "", "No audio received.", None, "", [], render_empty_chart("📊 Score Comparison"), render_empty_chart("📈 Tone Progress"), gr.update(visible=False)
+                return "", "No audio received.", None, "", [], render_empty_chart("📊 Score Comparison"), render_empty_chart("📈 Progress Over Time"), gr.update(visible=False)
 
             if hasattr(nickname, "value"):
                 nickname = nickname.value
@@ -140,19 +145,19 @@ def spoken_dashboard(nickname_input):
             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(full_feedback, previous_feedback)) if previous_feedback else render_empty_chart("📊 Score Comparison")
-            dates, tone_scores = build_trend_data(sessions, "Tone")
-            trend_plot = render_trend_chart(dates, tone_scores, "Tone") if tone_scores else render_empty_chart("📈 Tone Progress")
+            dates, trend_scores = build_trend_data(sessions, trend_category)
+            trend_plot = render_trend_chart(dates, trend_scores, trend_category) if trend_scores else render_empty_chart(f"📈 {trend_category} Progress")
 
             return transcript, feedback_clean, mp3_path, transcript, session_table, score_plot, trend_plot, gr.update(visible=bool(milestone), value=milestone)
 
         audio_input.change(
             fn=tutor_feedback,
-            inputs=[audio_input, language_dropdown, goal_dropdown, focus_checkboxes, nickname_input],
+            inputs=[audio_input, language_dropdown, goal_dropdown, focus_checkboxes, trend_category_dropdown, nickname_input],
             outputs=[transcript_box, feedback_box, audio_output, hidden_transcript, history_table, score_chart, trend_chart, milestone_box],
             show_progress="minimal"
         )
 
-        try_again.click(fn=lambda: ("", "", None, "", "", render_empty_chart("📊 Score Comparison"), render_empty_chart("📈 Tone Progress"), gr.update(visible=False)),
+        try_again.click(fn=lambda: ("", "", None, "", "", render_empty_chart("📊 Score Comparison"), render_empty_chart("📈 Progress Over Time"), gr.update(visible=False)),
                         inputs=None,
                         outputs=[transcript_box, feedback_box, audio_output, hidden_transcript, example_box, score_chart, trend_chart, milestone_box])