scoreboard feature and bugfix on empty transcript table

app.py

@@ -8,6 +8,24 @@ from dotenv import load_dotenv
 import base64
 import datetime
 
+# Positive sentiments
+positive_sentiments = [
+    "approval", 
+    "realization", 
+    "joy", 
+    "caring", 
+    "relief", 
+    "desire", 
+    "admiration",
+    "optimism",
+    "love",
+    "excitement",
+    "curiosity",
+    "amusement",
+    "gratitude",
+    "pride"
+]
+
 # Load google cloud credentials
 load_dotenv()
 base64_credentials = os.environ.get('GOOGLE_APPLICATION_CREDENTIALS')
@@ -18,42 +36,50 @@ db = firestore.Client.from_service_account_info(credentials_json)
 # ===== Authentication =====
 
 def authenticate(new_username, new_pw): 
-    if new_username == '' or new_pw == '': return [None, None, gr.update(), gr.update()]
+    if new_username == '' or new_pw == '': return [None, None, 0, gr.update(), gr.update()]
     users_ref = db.collection('Users')
     doc_ref = users_ref.document(new_username)
     doc = doc_ref.get()
-
+    new_score = 0
     if doc.exists:
         # User exists in Firestore
         user_data = doc.to_dict()
-        
+        new_score = user_data['score']
         # Handle incorrect password
         if user_data['password'] != new_pw:
             raise gr.Error("Incorrect password")
-            return None
     else: 
-        doc_ref.set({"username": new_username, "password": new_pw})
+        doc_ref.set({"username": new_username, "password": new_pw, "score": new_score})
 
     gr.Info(f"Welcome, {new_username}!")
-
     show_welcome = gr.update(visible=True, value=f'<div style=\'height:190px; display:flex; justify-content:center; align-items:center;\'><h1 style=\'text-align:center\'>Hello {new_username}! 👋</h1></div>')
     hide_signin = gr.update(visible=False)
 
-    return [new_username, new_pw, show_welcome, hide_signin]
+    return [new_username, new_pw, new_score, show_welcome, hide_signin]    
 
 def get_user_transcripts(username):
     arr = []
     if username is None: return [gr.update(value=arr)]
-
     # Fetch user's records
     user_transcripts = db.collection(f'Users/{username}/Transcripts').stream()
-
     for trans in user_transcripts:
         trans_dict = trans.to_dict()
         arr.append([trans_dict['date'], trans_dict['transcription'], trans_dict['sentiment_output']])
-    
+    if (len(arr) == 0): 
+        arr = ['', '', '']
     return arr
 
+def get_user_score(username):
+    doc = db.document(f'Users/{username}').get()
+    if doc.exists:
+        # User exists in Firestore
+        user_data = doc.to_dict()
+        return [f"""
+                <p align="center">Earn points by making customers happy!</p>
+                <br/>
+                <h1 align="center" style=\'font-size:56px;\'>{user_data["score"]}</h1>
+                """]
+    return [f'<h1 align="center"></h1>']
 
 # ===== Loading Whisper =====
 
@@ -66,6 +92,12 @@ def analyze_sentiment(text):
     sentiment_results = {result['label']: result['score'] for result in results}
     return sentiment_results
 
+def is_positive(result): 
+    result = result.split(' ')[0]
+    if (result in positive_sentiments): 
+        return True
+    return False
+
 def get_sentiment_emoji(sentiment):
     # Define the emojis corresponding to each sentiment
     emoji_mapping = {
@@ -123,17 +155,23 @@ def inference(username, audio, sentiment_option):
     result = whisper.decode(model, mel, options)
 
     sentiment_results = analyze_sentiment(result.text)
-    sentiment_output = display_sentiment_results(sentiment_results, sentiment_option)
-
+    sentiment_output = display_sentiment_results(sentiment_results, sentiment_option)        
     if username:
         # save results in firestore
         ts = datetime.datetime.now()
         ts_formatted = ts.strftime("%d %b %Y, %H:%M")
-        doc_ref = db.document(f'Users/{username}/Transcripts/{ts_formatted}')
-        doc_ref.set({"date": ts_formatted, "transcription": result.text, "sentiment_output": sentiment_output})
+        ref = db.document(f'Users/{username}')
+        transcript_ref = db.document(f'Users/{username}/Transcripts/{ts_formatted}')
+        transcript_ref.set({"date": ts_formatted, "transcription": result.text, "sentiment_output": sentiment_output})
+        person_doc = ref.get()
+        user_data = person_doc.to_dict()
+        new_score = user_data['score']
+        if is_positive(sentiment_output):
+            new_score = new_score + 1
+            db.document(f'Users/{username}').update({"score": new_score})
         gr.Info("Transcription saved!")
 
-    return lang.upper(), result.text, sentiment_output
+    return lang.upper(), result.text, sentiment_output, new_score
 
 title = """<h1 align="center">☕ Lim Kopi Call Center Service 💬</h1>"""
 image_path = "coffee_logo.jpg"
@@ -185,6 +223,7 @@ with app:
     gr.HTML(title)
     authed_username = gr.State(value=None)
     authed_password = gr.State(value=None)
+    user_score = gr.State(value=0)
 
     # ===== UI =====
 
@@ -206,7 +245,7 @@ with app:
                 login_btn.click(
                     authenticate, 
                     inputs=[username_input, password_input], 
-                    outputs=[authed_username, authed_password, user_welcome, auth_block]
+                    outputs=[authed_username, authed_password, user_score, user_welcome, auth_block]
                 )
 
         with gr.Column(scale=1):
@@ -224,17 +263,25 @@ with app:
     with gr.Row():
         with gr.Column():
             gr.HTML("<br/>")
-            gr.HTML("""<h2 align="center">🎉 Results</h1>""")
+            gr.HTML("""<h1 align="center">🎉 Results</h1>""")
             with gr.Group():
                 lang_str = gr.Textbox(label="Language")
                 text = gr.Textbox(label="Transcription")
                 sentiment_output = gr.Textbox(label="Sentiment Analysis Results")
-                btn.click(inference, inputs=[authed_username, audio, sentiment_option], outputs=[lang_str, text, sentiment_output])
+                btn.click(inference, inputs=[authed_username, audio, sentiment_option], outputs=[lang_str, text, sentiment_output, user_score])
+    
+    with gr.Row(visible=True) as scoreboard:
+        with gr.Column():
+            gr.HTML("<br/>")
+            gr.HTML("""<h1 align="center">💯 Your Score</h1>""")
+            score_sheet = gr.HTML(visible=True, value=f'<p align="center">Log in to see your score and transcripts</p>')
+            user_welcome.change(get_user_score, inputs=[authed_username], outputs=[score_sheet])
+            sentiment_output.change(get_user_score, inputs=[authed_username], outputs=[score_sheet])
     
     with gr.Row(visible=True) as transcription_records:
         with gr.Column():
             gr.HTML("<br/>")
-            gr.HTML("""<h2 align="center"> 🪩 Your Transcription Records</h1>""")
+            gr.HTML("""<h1 align="center"> 🪩 Your Transcription Records</h1>""")
             transcription_df = gr.Dataframe(
                 headers=["Date", "Transcription", "Sentiment"],
                 datatype=["str", "str", "str"],
@@ -245,4 +292,4 @@ with app:
             sentiment_output.change(get_user_transcripts, inputs=[authed_username], outputs=[transcription_df[0]])
 
 app.queue()
-app.launch()
+app.launch()