Update app.py

app.py

@@ -1,38 +1,20 @@
 import gradio as gr
 from transformers import pipeline
-import pandas as pd
 from datetime import datetime
 import sqlite3
 import os
 import random
 
 # -------------------------------------------------
-# 🎯 MODEL & DATABASE INITIALIZATION
+# 🎯 MODEL INITIALIZATION
 # -------------------------------------------------
 MODEL_NAME = "superb/hubert-large-superb-er"
 emotion_classifier = pipeline("audio-classification", model=MODEL_NAME)
 
+# (Optional) Simulated DB path
 os.makedirs("data", exist_ok=True)
 DB_PATH = "data/scm_emotion.db"
 
-def init_db():
-    conn = sqlite3.connect(DB_PATH)
-    cursor = conn.cursor()
-    cursor.execute("""
-    CREATE TABLE IF NOT EXISTS emotion_logs (
-        id INTEGER PRIMARY KEY AUTOINCREMENT,
-        datetime TEXT,
-        team TEXT,
-        purpose TEXT,
-        emotion TEXT,
-        score REAL
-    )
-    """)
-    conn.commit()
-    conn.close()
-
-init_db()
-
 # -------------------------------------------------
 # 😄 EMOTION MAP + COLORS
 # -------------------------------------------------
@@ -48,65 +30,37 @@ EMOTION_MAP = {
     "dis": ("Disgusted", "🤢", "#8bc34a"),
 }
 
-# -------------------------------------------------
-# 💾 LOGGING
-# -------------------------------------------------
-def log_to_db(datetime_str, team, purpose, emotion, score):
-    conn = sqlite3.connect(DB_PATH)
-    cursor = conn.cursor()
-    cursor.execute(
-        "INSERT INTO emotion_logs (datetime, team, purpose, emotion, score) VALUES (?, ?, ?, ?, ?)",
-        (datetime_str, team, purpose, emotion, score),
-    )
-    conn.commit()
-    conn.close()
-
 # -------------------------------------------------
 # 🧠 EMOTION ANALYSIS
 # -------------------------------------------------
-def analyze_emotion(audio, team, custom_team, purpose, custom_purpose):
+def analyze_emotion(audio, team, purpose):
     if audio is None:
-        return "⚠️ Please record or upload audio first.", None
+        return "<p style='color:red;'>⚠️ Please record or upload audio.</p>", None
 
-    # Determine final values
-    team_name = custom_team if team == "Other" else team
-    purpose_name = custom_purpose if purpose == "Other" else purpose
+    # If team/purpose not filled
+    if team == "Other":
+        team = "Custom/Unspecified"
+    if purpose == "Other":
+        purpose = "Custom/Unspecified"
 
-    if not team_name:
-        team_name = "Unspecified Team"
-    if not purpose_name:
-        purpose_name = "Unspecified Purpose"
-
-    # Handle file input
-    if isinstance(audio, tuple):
-        audio_file = audio[0]
-    else:
-        audio_file = audio
-
-    # Predict emotion
-    results = emotion_classifier(audio_file)
+    results = emotion_classifier(audio)
     results = sorted(results, key=lambda x: x['score'], reverse=True)
     top = results[0]
     label, emoji, color = EMOTION_MAP.get(top['label'], (top['label'], "🎭", "#607d8b"))
     score = round(top['score'] * 100, 2)
 
-    # Log data
-    log_to_db(datetime.now().strftime("%Y-%m-%d %H:%M:%S"), team_name, purpose_name, label, score)
-
-    # Dashboard UI
+    # Dashboard HTML
     dashboard_html = f"""
-    <div style="text-align:center; padding:15px;">
-        <h2>🏢 <b>{team_name}</b></h2>
-        <h4>🎯 Purpose: <b>{purpose_name}</b></h4>
+    <div style="text-align:center; padding:15px; position:sticky; top:10px;">
+        <h2>🏢 <b>{team}</b> | 🎯 <b>{purpose}</b></h2>
         <h3 style="color:{color}; font-size:26px;">{emoji} {label.upper()} — {score}%</h3>
         <p style="color:#666;">Detected Emotion Intensity (Confidence)</p>
-        <div style="width:80%; margin:auto; background:#eee; border-radius:20px;">
+        <div style="width:85%; margin:auto; background:#eee; border-radius:20px;">
             <div style="width:{score}%; background:{color}; height:20px; border-radius:20px;"></div>
         </div>
-    </div>
-    <br>
-    <h4>📊 Full Emotion Breakdown</h4>
-    <ul>
+        <br>
+        <h4>📊 Full Emotion Breakdown</h4>
+        <ul style="list-style:none; padding:0;">
     """
     for r in results:
         lbl, emo, clr = EMOTION_MAP.get(r['label'], (r['label'], "🎭", "#777"))
@@ -114,98 +68,96 @@ def analyze_emotion(audio, team, custom_team, purpose, custom_purpose):
         dashboard_html += f"<li>{emo} <b>{lbl}</b>: {scr}%</li>"
     dashboard_html += "</ul><hr>"
 
-    # AI Insight
+    # Insight
     insights = random.choice([
-        "🧩 The conversation tone was balanced — good team stability.",
-        "⚡ Slight emotional tension detected — consider a quick check-in.",
-        "💬 Positive tone overall — keep up the good engagement.",
-        "🚨 Possible stress indicators — leadership follow-up advised.",
-        "📈 Emotional fluctuation noted — monitor workload or dynamics."
+        "🧩 Team seems calm and balanced today. Great stability!",
+        "⚡ Slight emotional tension detected. Consider quick sync-up meetings.",
+        "💬 High positive tone — keep up the good energy!",
+        "🚨 Stress indicators detected. HR may follow up proactively.",
+        "📈 Emotion variation is rising — review workloads or deadlines."
     ])
 
     insight_html = f"""
-    <div style="background:#f5f5f5; padding:15px; border-radius:12px; margin-top:15px; border-left:6px solid #4a90e2;">
+    <div style="background:#f5f5f5; padding:15px; border-radius:12px;
+                margin-top:15px; border-left:6px solid #4a90e2;
+                box-shadow:0 2px 6px rgba(0,0,0,0.05);">
         <h4 style="color:#222; margin-bottom:6px;">🧠 AI Insight</h4>
         <p style="color:#333; font-size:16px; font-weight:500;">{insights}</p>
-        <p style="font-size:13px; color:#777;">Logged for Power BI visualization.</p>
+        <p style="font-size:13px; color:#777;">(Demo mode — not stored to any database)</p>
     </div>
     """
 
     return dashboard_html, insight_html
 
-
 # -------------------------------------------------
-# 🎨 GRADIO APP DESIGN
+# 🎨 GRADIO INTERFACE
 # -------------------------------------------------
-TEAM_OPTIONS = ["Test","SAP","Analytics","AI","Procurement", "Logistics", "Planning", "HR", "Operations", "IT", "Other"]
-PURPOSE_OPTIONS = ["Test","HR Meeting", "Team Stand-up", "One-on-One", "Customer Call", "Interview", "Project Review", "Other"]
+DEPARTMENTS = [
+    "Procurement", "Logistics", "Planning", "Inventory", "Distribution", "HR", "Other"
+]
+PURPOSES = [
+    "HR Meeting", "Team Stand-up", "One-on-One", "Customer Call", "Interview", "Other"
+]
+
+disclaimer_html = """
+<div style="display:flex; align-items:center; background:linear-gradient(135deg, #fff8e1, #fff3cd);
+            padding:15px; border-radius:12px; margin-top:20px; border-left:8px solid #ff9800;
+            box-shadow:0 2px 6px rgba(0,0,0,0.1);">
+    <div style="font-size:28px; margin-right:12px; color:#ff9800;">💡</div>
+    <div>
+        <h3 style="color:#d35400; margin:0 0 5px 0;">Demo Disclaimer</h3>
+        <p style="color:#333; font-size:15px; line-height:1.6; margin:0;">
+            This is a <b style="color:#e65100;">demonstration version</b> created for HR and leadership showcases.<br>
+            Your voice data is 
+            <b style="background-color:#fffbe6; color:#c0392b; padding:2px 6px; border-radius:4px;">
+            NOT being saved or shared
+            </b> anywhere.<br>
+            Database logging is <b style="color:#e65100;">simulated</b> — you can record or upload audio freely.<br>
+            ✅ Safe to explore and test in this environment.
+        </p>
+    </div>
+</div>
+"""
 
 with gr.Blocks(theme=gr.themes.Soft()) as app:
-    # Header
     gr.HTML("""
     <div style="text-align:center; padding:20px;">
-        <h1>🎧 Emotion Intelligence for Teams</h1>
+        <h1>🎧 SCM Emotion Intelligence Dashboard</h1>
         <p style="font-size:16px; color:#555;">
-            Detect and visualize emotional tone from team meetings or recorded conversations.<br>
-            Designed for managers and HR to track emotional health and engagement trends.
+            Analyze live or uploaded audio from meetings to understand emotional tone within teams.<br>
+            Built for HR & Managers to assess engagement and team well-being.
         </p>
     </div>
     """)
 
-    # Inputs
     with gr.Row():
-        with gr.Column(scale=1):
-            team_dropdown = gr.Dropdown(TEAM_OPTIONS, label="🏢 Select Team / Department")
-            custom_team = gr.Textbox(label="✏️ Enter Team (if 'Other')", placeholder="Enter custom team name here")
-
-            purpose_dropdown = gr.Dropdown(PURPOSE_OPTIONS, label="🎯 Purpose of the Recording")
-            custom_purpose = gr.Textbox(label="✏️ Enter Purpose (if 'Other')", placeholder="Enter custom meeting purpose here")
-
+        # Left Panel: Inputs
+        with gr.Column(scale=1, min_width=350):
             audio_input = gr.Audio(
                 sources=["microphone", "upload"],
                 type="filepath",
-                label="🎙️ Record or Upload Team Meeting Audio",
+                label="🎙️ Record or Upload Audio",
             )
+            team_input = gr.Dropdown(DEPARTMENTS, label="🏢 Select Team / Department")
+            purpose_input = gr.Dropdown(PURPOSES, label="🎯 Purpose of Audio")
             analyze_btn = gr.Button("🚀 Analyze Emotion", variant="primary")
+            gr.HTML(disclaimer_html)
 
-        with gr.Column(scale=2):
+        # Right Panel: Sticky Output
+        with gr.Column(scale=2, min_width=500):
             output_html = gr.HTML()
             insight_html = gr.HTML()
 
-    # Disclaimer
-    gr.HTML("""
-    <div style="background:linear-gradient(135deg, #fff8e1, #fff3cd); 
-                padding:15px; 
-                border-radius:12px; 
-                margin-top:20px; 
-                border-left:8px solid #ff9800;
-                box-shadow:0 2px 6px rgba(0,0,0,0.1);">
-        <h3 style="color:#d35400; margin-bottom:8px;">⚠️ Demo Disclaimer</h3>
-        <p style="color:#333; font-size:15px; line-height:1.6;">
-            This is a <b style="color:#e65100;">demonstration version</b> created for concept showcasing.<br>
-            Your voice data is 
-            <b style="background-color:#fffbe6; color:#c0392b; padding:2px 6px; border-radius:4px;">
-            NOT being saved or shared
-            </b> anywhere.<br><br>
-            All recordings are processed only in-memory and database logging is 
-            <b style="color:#e65100;">simulated</b> for visualization purposes.<br>
-            ✅ Feel free to explore, record, and test the system freely.
-        </p>
-    </div>
-    """)
-
-
-    # Button Action
     analyze_btn.click(
         fn=analyze_emotion,
-        inputs=[audio_input, team_dropdown, custom_team, purpose_dropdown, custom_purpose],
+        inputs=[audio_input, team_input, purpose_input],
         outputs=[output_html, insight_html],
     )
 
     gr.HTML("""
     <hr>
-    <p style="text-align:center; color:#999;">
-        💾 Results are locally logged in <code>scm_emotion.db</code> (for Power BI integration demo).
+    <p style="text-align:center; color:#999; font-size:14px;">
+        💾 (Demo Mode) Database integration coming soon for Power BI visualization.
     </p>
     """)