import streamlit as st from components.visualizations import plot_emotion_distribution from utils import model_inference from components.audio_player import play_audio from components.debug_tools import DebugTools import json import os import time # ✅ 初始化 session_state 以存储分析结果和反馈状态 if "emotion_result" not in st.session_state: st.session_state.emotion_result = None if "feedback_submitted" not in st.session_state: st.session_state.feedback_submitted = False if "user_feedback" not in st.session_state: st.session_state.user_feedback = "" def show_history(): """📜 显示历史记录（最近 5 条）""" history_file = "history/history.json" if os.path.exists(history_file): with open(history_file, 'r') as f: history = json.load(f) if history: st.subheader("📜 History") with st.expander("Click to view past records", expanded=False): for record in history[-5:]: # 只显示最近 5 条 st.write(f"🎧 **Audio File**: {record['audio_file']}") st.write(f"📝 **Transcript**: {record['transcript']}") st.write(f"🎭 **Emotions**: {record['emotions']}") st.write("---") else: st.write("🚫 No history records found.") else: st.write("🚫 No history file.") def show(): # ✅ 初始化 session_state 以存储分析结果和反馈状态 if "emotion_result" not in st.session_state: st.session_state.emotion_result = None if "feedback_submitted" not in st.session_state: st.session_state.feedback_submitted = False if "user_feedback" not in st.session_state: st.session_state.user_feedback = "" st.markdown( "

😉 Emotion Analyzer 🎭

", unsafe_allow_html=True ) # 📜 显示历史记录 show_history() # 🛠️ 初始化调试工具 debug = DebugTools() debug.show_debug_info() # 🎤 让用户上传音频和文本输入 st.markdown("---") col1, col2 = st.columns(2) with col1: audio_file = st.file_uploader("Upload your audio file", type=['wav', 'mp3']) # 🎧 如果音频存在，则显示音频信息 if audio_file is not None: debug.show_audio_info(audio_file) st.audio(audio_file, format="audio/mp3") with col2: text_input = st.text_input("Enter your text input") transcript = text_input # 默认情况下，文本输入即为转写结果 st.write("📄 **Audio transcript:**", transcript) # 🎯 **如果用户删除了音频，则自动重置情绪分析结果** if audio_file is None and "emotion_result" in st.session_state and st.session_state.emotion_result is not None: st.warning("🚫 Audio file removed. Resetting analysis results.") st.session_state.emotion_result = None # # 🎧 如果音频存在，则显示音频信息 # if audio_file is not None: # debug.show_audio_info(audio_file) # st.audio(audio_file, format="audio/mp3") if audio_file is not None and text_input: if st.button("Analyse Your Emotion! 😊"): st.session_state.feedback_submitted = False # 重置反馈状态 # 📊 显示进度条 progress_bar = st.progress(0) for percent in range(0, 101, 20): progress_bar.progress(percent) time.sleep(0.2) # 提高加载体验 try: transcript = text_input # 🔍 执行情绪分析 with st.spinner("Analysing emotion..."): for percent in range(30, 101, 20): progress_bar.progress(percent) time.sleep(0.2) # 避免过快闪过 emotions = model_inference.predict_emotion(text_input, audio_file) if emotions: best_emotion = max(emotions, key=lambda x: float(emotions[x].strip('%'))) # 🎭 显示情绪分析结果 st.markdown("---") st.markdown("### 🎭 Emotion Analysis Results") st.success(f"🎭 **Predicted Emotion:** {best_emotion}") st.success(f"📊 **Emotion Probabilities:** {emotions}") # ✅ 存储分析结果，防止页面刷新导致结果丢失 st.session_state.emotion_result = { "best_emotion": best_emotion, "emotions": emotions, "transcript": transcript } except Exception as e: debug.log_error(e, context=f"Processing file: {audio_file.name}") # 🎭 只有当有分析结果时，才显示结果 if st.session_state.emotion_result: best_emotion = st.session_state.emotion_result["best_emotion"] emotions = st.session_state.emotion_result["emotions"] # 🎈 Streamlit 内置动画（仅用于 "Joy" 和 "Sad"） if best_emotion == "Joy": st.balloons() st.markdown("**Yay! You seem so happy! 🎉😊**") st.write("Keep smiling, it's contagious! 🌟") elif best_emotion == "Sad": st.snow() st.markdown("**Oh no, you seem a bit down... 😔💭**") st.write("I'm here if you need to talk or cheer up! 💬💖") else: emotion_messages = { "Angry": ["😡 **Whoa! Looks like you're really angry...**", "Take a deep breath, let's cool off together! 🧘‍♂️"], "Surprised": ["🤯 **Wow, you're really surprised!**", "That's a big reaction! Hope it's a good surprise! 🎉"], "Fearful": ["😨 **Looks like you're feeling scared...**", "It's okay to be scared. I'm here to help! 💪"], "Disgusted": ["🤢 **Yikes, you're feeling disgusted...**", "That doesn't sound pleasant. Let's turn it around! 🌈"], "Neutral": ["🤔 **Hmm, your emotions seem mixed...**", "Can you tell me more? Let's figure this out together! 💬"] } if best_emotion in emotion_messages: st.markdown(emotion_messages[best_emotion][0]) st.write(emotion_messages[best_emotion][1]) # 📊 显示情绪分布图 plot_emotion_distribution(emotions) # 💾 保存历史记录 if audio_file is not None: model_inference.save_history(audio_file, transcript, emotions) else: st.warning("⚠️ No audio file uploaded. History not saved.") # 📣 提供反馈功能 if st.session_state.emotion_result: st.subheader("📣 Give us your feedback!") feedback = st.radio("Do you agree with the prediction?", ["😀 Great!", "😐 Okay", "😡 Wrong"], index=1) user_feedback = st.text_area("Tell us more (optional):", st.session_state.user_feedback) if st.button("Submit Feedback"): st.session_state.user_feedback = user_feedback st.session_state.feedback_submitted = True if st.session_state.feedback_submitted: st.success("✅ Thank you for your feedback! It has been recorded.")