Update app.py

app.py

@@ -10,9 +10,10 @@ import subprocess
 import shutil
 from datetime import datetime
 from sqlmodel import SQLModel, Field, create_engine, Session, select
-
-# === Clean safe app-generated temp files ===
+from typing import Optional
 import glob
+
+# === Clean temp audio files ===
 for pattern in ["/tmp/*.wav", "/tmp/*.mp3"]:
     for filepath in glob.glob(pattern):
         try:
@@ -35,9 +36,10 @@ LANG_CODES = {
     "Arabic": "ar", "Chinese": "zh", "Portuguese": "pt", "Japanese": "ja", "Korean": "ko"
 }
 
-# === SQLModel Setup ===
+# === SQLModel with nickname support ===
 class SessionEntry(SQLModel, table=True):
-    id: int = Field(default=None, primary_key=True)
+    id: Optional[int] = Field(default=None, primary_key=True)
+    user: str
     timestamp: str
     transcript: str
     feedback: str
@@ -46,9 +48,10 @@ class SessionEntry(SQLModel, table=True):
 engine = create_engine(f"sqlite:///{db_path}")
 SQLModel.metadata.create_all(engine)
 
-def save_to_db(transcript, feedback, language):
+def save_to_db(user, transcript, feedback, language):
     session = Session(engine)
     entry = SessionEntry(
+        user=user,
         timestamp=datetime.now().strftime("%Y-%m-%d %H:%M"),
         transcript=transcript,
         feedback=feedback,
@@ -58,9 +61,9 @@ def save_to_db(transcript, feedback, language):
     session.commit()
     session.close()
 
-def fetch_all_sessions():
+def fetch_user_sessions(user):
     session = Session(engine)
-    statement = select(SessionEntry)
+    statement = select(SessionEntry).where(SessionEntry.user == user)
     results = session.exec(statement).all()
     session.close()
     return results
@@ -121,7 +124,7 @@ Transcript:
     )
     return response.choices[0].message.content
 
-def tutor_feedback(audio_file, language):
+def tutor_feedback(audio_file, language, nickname):
     if not audio_file or not os.path.exists(audio_file):
         return "", "No audio received.", None, "", []
 
@@ -134,8 +137,8 @@ def tutor_feedback(audio_file, language):
     mp3_path = f"/tmp/{uuid.uuid4()}.mp3"
     tts.save(mp3_path)
 
-    save_to_db(transcript, feedback_text, language)
-    sessions = fetch_all_sessions()
+    save_to_db(nickname, transcript, feedback_text, language)
+    sessions = fetch_user_sessions(nickname)
 
     return transcript, feedback_text, mp3_path, transcript, [[s.timestamp, s.language, s.transcript[:40], s.feedback[:40]] for s in sessions]
 
@@ -145,10 +148,11 @@ with gr.Blocks(css="light_mode_chatter_owl.css") as app:
     <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 your language, speak into the mic, and I’ll give you structured feedback to help you grow as a communicator!</p>
+        <p>Enter your name, choose a language, and speak! I’ll help you grow as a communicator.</p>
     </div>
     """)
 
+    nickname_box = gr.Textbox(label="👤 Your Nickname", placeholder="Enter your name...")
     language_dropdown = gr.Dropdown(label="🌍 Select Your Language", choices=list(LANG_CODES.keys()), value="English")
 
     with gr.Row():
@@ -168,7 +172,7 @@ with gr.Blocks(css="light_mode_chatter_owl.css") as app:
 
     # Interactions
     audio_input.change(fn=tutor_feedback,
-                       inputs=[audio_input, language_dropdown],
+                       inputs=[audio_input, language_dropdown, nickname_box],
                        outputs=[transcript_box, feedback_box, audio_output, hidden_transcript, history_table])
 
     try_again.click(fn=lambda: ("", "", None, "", "", []),