Update spoken_module.py

spoken_module.py

@@ -3,7 +3,6 @@ import uuid
 import os
 import matplotlib.pyplot as plt
 from gtts import gTTS
-from datetime import datetime
 from openai import OpenAI
 from app_utils import (
     LANG_CODES, save_to_db, fetch_user_sessions,
@@ -21,12 +20,16 @@ def generate_feedback(transcript, language, goal="general improvement", focus_ar
 
     prompt = f"""
 You are a supportive communication coach helping a learner whose goal is: **{goal}**.
+
 Evaluate the user's current speech based on the following areas:
 {focus_str}
+
 Give a score out of 10 and a short explanation for each area.
+
 Then provide:
 - A summary of strengths and improvement areas.
 - One motivational line to end with.
+
 Transcript:
 {transcript}
 {history_section}
@@ -58,68 +61,71 @@ Transcript:
     )
     return response.choices[0].message.content
 
+def render_empty_chart(title):
+    fig, ax = plt.subplots()
+    ax.set_title(title)
+    ax.text(0.5, 0.5, "No data yet", ha='center', va='center', fontsize=12)
+    ax.axis('off')
+    return fig
+
 def spoken_dashboard(nickname_input):
     with gr.Column() as spoken_panel:
         gr.Markdown("""
         <div id="header" style="text-align: center;">
+            <img src="images/chatternest_logo.png" width="100">
             <h2>🦉 Meet <strong>Chatter the Owl</strong></h2>
-            <p>Choose your goal, speak, and get better every time.</p>
+            <p>Speak and improve your communication with personalized feedback and progress tracking.</p>
         </div>
         """)
 
-        with gr.Row(elem_classes=["top-control-row"]):
+        with gr.Row():
             language_dropdown = gr.Dropdown(label="🌍 Language", choices=list(LANG_CODES.keys()), value="English")
-            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"])
-
-        with gr.Row(elem_classes=["audio-row"]):
+            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"])
+
+        with gr.Row():
             audio_input = gr.Audio(type="filepath", label="🎙 Speak or Upload Audio")
-            audio_output = gr.Audio(label="🔊 Feedback Audio", type="filepath")
+            audio_output = gr.Audio(label="🔊 Chatter's Response", type="filepath")
 
         transcript_box = gr.Textbox(label="📖 Transcript", interactive=False)
         feedback_box = gr.Textbox(label="💡 Feedback", interactive=False)
         hidden_transcript = gr.Textbox(visible=False)
 
-        with gr.Row(elem_classes=["text-columns"]):
+        with gr.Row():
             try_again = gr.Button("🔁 Try Again")
-            show_example = gr.Button("🎯 Show Example")
+            show_example = gr.Button("🎯 Show Me an Example")
 
         example_box = gr.Textbox(label="🗣 Suggested Improvement", visible=True)
         history_table = gr.Dataframe(headers=["🕒 Timestamp", "🌐 Language", "📄 Transcript", "💬 Feedback"])
         score_chart = gr.Plot(label="📊 Score Comparison")
         trend_chart = gr.Plot(label="📈 Tone Progress")
         milestone_box = gr.Markdown(visible=False)
-        comparison_plot = gr.Plot(label="📊 Score Comparison")
-        trend_plot = gr.Plot(label="📈 Tone Trend")
-        milestone_box = gr.Markdown(visible=False)
 
         def tutor_feedback(audio_file, language, goal, focus_areas, nickname):
             if not audio_file:
-                return "", "No audio received.", None, "", [], None, None, gr.update(visible=False)
+                return "", "No audio received.", None, "", [], render_empty_chart("📊 Score Comparison"), render_empty_chart("📈 Tone Progress"), gr.update(visible=False)
+
+            if hasattr(nickname, "value"):
+                nickname = nickname.value  # Handle gr.Textbox or gr.State
 
             wav_path = convert_to_wav(audio_file)
             transcript = transcribe_audio(wav_path)
-            sessions = fetch_user_sessions(nickname)
-            previous_transcript = sessions[-1].transcript if sessions else None
-            previous_feedback = sessions[-1].feedback if sessions else None
+            previous_sessions = fetch_user_sessions(nickname)
+            previous_transcript = previous_sessions[-1].transcript if previous_sessions else None
+            previous_feedback = previous_sessions[-1].feedback if previous_sessions else None
 
             feedback_text = generate_feedback(transcript, language, goal, focus_areas, previous_transcript)
 
             if previous_feedback:
                 feedback_text += generate_progress_summary(feedback_text, previous_feedback)
 
-            session_count = len(sessions) + 1
-            milestone_msg = ""
+            milestone = ""
+            session_count = len(previous_sessions) + 1
             if session_count in [3, 5, 10]:
-                milestone_msg = f"🎉 You’ve completed {session_count} sessions!"
-                feedback_text += f"\n\n{milestone_msg}"
+                milestone = f"🎉 Congrats on completing **{session_count} sessions**!"
+                feedback_text += f"\n\n{milestone}"
 
-            lang_code = LANG_CODES.get(language, "en")
-            tts = gTTS(feedback_text, lang=lang_code)
+            tts = gTTS(feedback_text, lang=LANG_CODES.get(language, "en"))
             mp3_path = f"/tmp/{uuid.uuid4()}.mp3"
             tts.save(mp3_path)
 
@@ -127,33 +133,25 @@ def spoken_dashboard(nickname_input):
             sessions = fetch_user_sessions(nickname)
             session_table = [[s.timestamp, s.language, s.transcript[:40], s.feedback[:40]] for s in sessions]
 
-            chart_plot = render_score_chart(build_score_comparison_data(feedback_text, previous_feedback)) if previous_feedback else None
-            dates, tone_scores = build_trend_data(sessions, category="Tone")
-            trend_chart = render_trend_chart(dates, tone_scores, "Tone") if tone_scores else None
+            score_plot = render_score_chart(build_score_comparison_data(feedback_text, 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")
 
-            return transcript, feedback_text, mp3_path, transcript, session_table, chart_plot, trend_chart, gr.update(visible=bool(milestone_msg), value=milestone_msg)
+            return transcript, feedback_text, 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  # ✅ This is the actual textbox, not a State object
-            ],
-            outputs=[
-                transcript_box, feedback_box, audio_output,
-                hidden_transcript, history_table,
-                score_chart, trend_chart, milestone_box
-            ]
+            inputs=[audio_input, language_dropdown, goal_dropdown, focus_checkboxes, 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, "", "", None, None, gr.update(visible=False)),
+        try_again.click(fn=lambda: ("", "", None, "", "", render_empty_chart("📊 Score Comparison"), render_empty_chart("📈 Tone Progress"), gr.update(visible=False)),
                         inputs=None,
-                        outputs=[transcript_box, feedback_box, audio_output, hidden_transcript, example_box, comparison_plot, trend_plot, milestone_box])
+                        outputs=[transcript_box, feedback_box, audio_output, hidden_transcript, example_box, score_chart, trend_chart, milestone_box])
 
         show_example.click(fn=generate_example_response,
                            inputs=[hidden_transcript, language_dropdown],
                            outputs=example_box)
 
-    return spoken_panel
+    return spoken_panel