Update spoken_module.py

spoken_module.py

@@ -1,17 +1,47 @@
 import gradio as gr
 import uuid
+import os
 import matplotlib.pyplot as plt
 from gtts import gTTS
 from datetime import datetime
-from sqlmodel import Session
+from openai import OpenAI
 from app_utils import (
-    model, LANG_CODES, client, save_to_db, fetch_user_sessions,
-    convert_to_wav, transcribe_audio, generate_feedback,
+    LANG_CODES, save_to_db, fetch_user_sessions,
+    convert_to_wav, transcribe_audio,
     parse_scores_from_feedback, generate_progress_summary,
     build_score_comparison_data, render_score_chart,
     build_trend_data, render_trend_chart
 )
 
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+
+def generate_feedback(transcript, language, goal="general improvement", focus_areas=None, previous_transcript=None):
+    focus_str = ", ".join(focus_areas) if focus_areas else "Clarity, Structure, Fluency, Content Relevance, and Tone"
+    history_section = f"\n\nFor reference, their previous transcript was:\n{previous_transcript}" if previous_transcript else ""
+
+    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}
+""".strip()
+
+    response = client.chat.completions.create(
+        model="gpt-4",
+        messages=[
+            {"role": "system", "content": f"You are a warm and constructive communication coach responding in {language}."},
+            {"role": "user", "content": prompt}
+        ],
+        temperature=0.7
+    )
+    return response.choices[0].message.content
+
 def generate_example_response(transcript, language):
     prompt = f"""Rewrite this speech to make it more polished, fluent, and confident.
 Keep the meaning and tone the same, but improve clarity and structure.
@@ -28,53 +58,41 @@ Transcript:
     )
     return response.choices[0].message.content
 
-
 def spoken_dashboard(nickname):
     with gr.Column() as spoken_panel:
         gr.Markdown("""
         <div id="header" style="text-align: center;">
             <img src="file/images/chatter_owl.png" width="120">
             <h2>🦉 Meet <strong>Chatter the Owl</strong></h2>
-            <p>Choose a language and goal, and speak! I’ll help you grow as a communicator.</p>
+            <p>Choose your goal, speak, and get better every time.</p>
         </div>
         """)
 
-        language_dropdown = gr.Dropdown(label="🌍 Select Your Language", choices=list(LANG_CODES.keys()), value="English")
-
-        goal_dropdown = gr.Dropdown(
-            label="🎯 Communication Goal",
-            choices=[
-                "Interview preparation",
-                "Public speaking",
-                "Class presentation",
-                "General improvement"
-            ],
-            value="General improvement"
-        )
-
-        focus_checkboxes = gr.CheckboxGroup(
-            label="🧠 Areas to Focus On",
-            choices=["Clarity", "Structure", "Fluency", "Tone", "Content Relevance"],
-            value=["Clarity", "Structure", "Fluency", "Tone", "Content Relevance"]
-        )
+        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():
             audio_input = gr.Audio(type="filepath", label="🎙 Speak or Upload Audio")
+            audio_output = gr.Audio(label="🔊 Feedback Audio", type="filepath")
 
-        transcript_box = gr.Textbox(label="📖 What You Said", interactive=False)
-        feedback_box = gr.Textbox(label="💡 Chatter’s Feedback", interactive=False)
-        audio_output = gr.Audio(label="🔊 Chatter Speaks", 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():
             try_again = gr.Button("🔁 Try Again")
-            show_example = gr.Button("🎯 Show Me an Example")
-
-        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)"])
+            show_example = gr.Button("🎯 Show Example")
 
-        comparison_plot = gr.Plot(label="📊 Current vs Previous Scores")
-        trend_plot = gr.Plot(label="📈 Tone Progress Over Time")
+        example_box = gr.Textbox(label="🗣 Suggested Improvement", visible=True)
+        history_table = gr.Dataframe(headers=["🕒 Timestamp", "🌐 Language", "📝 Transcript", "📋 Feedback"])
+        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):
@@ -83,79 +101,49 @@ def spoken_dashboard(nickname):
 
             wav_path = convert_to_wav(audio_file)
             transcript = transcribe_audio(wav_path)
-            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
-
-            # Get new feedback
-            feedback_text = generate_feedback(
-                transcript=transcript,
-                language=language,
-                goal=goal,
-                focus_areas=focus_areas,
-                previous_transcript=previous_transcript
-            )
-
-            # Progress summary
+            sessions = fetch_user_sessions(nickname)
+            previous_transcript = sessions[-1].transcript if sessions else None
+            previous_feedback = sessions[-1].feedback if sessions else None
+
+            feedback_text = generate_feedback(transcript, language, goal, focus_areas, previous_transcript)
+
             if previous_feedback:
-                progress_summary = generate_progress_summary(feedback_text, previous_feedback)
-                feedback_text += progress_summary
+                feedback_text += generate_progress_summary(feedback_text, previous_feedback)
 
-            # Milestone message
-            session_count = len(previous_sessions) + 1
-            milestone_message = ""
+            session_count = len(sessions) + 1
+            milestone_msg = ""
             if session_count in [3, 5, 10]:
-                milestone_message = f"🎉 Congrats! You’ve completed **{session_count} sessions**!"
-                feedback_text += f"\n\n{milestone_message}"
+                milestone_msg = f"🎉 You’ve completed {session_count} sessions!"
+                feedback_text += f"\n\n{milestone_msg}"
 
-            # 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 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]
 
-            # 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")
+            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
 
-            return (
-                transcript, feedback_text, mp3_path,
-                transcript, session_table,
-                score_chart, trend_chart,
-                gr.update(visible=bool(milestone_message), value=milestone_message)
-            )
+            return transcript, feedback_text, mp3_path, transcript, session_table, chart_plot, trend_chart, gr.update(visible=bool(milestone_msg), value=milestone_msg)
 
         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, comparison_plot, trend_plot, milestone_box
-            ],
+            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, "", "", None, None, 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, 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