Create app.py

app.py

@@ -0,0 +1,89 @@
+import os
+import random
+import zipfile
+from transformers import pipeline
+import gradio as gr
+
+# 使用 Hugging Face 的翻譯管道
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-zh-en")
+
+# 使用 Hugging Face 的英文情緒分類管道
+classifier = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base")
+
+# 定義情緒標籤的映射
+emotion_labels = ["joy", "surprise", "sadness", "anger", "anxiety", "fear"]
+
+base_dir = '/content/'
+
+for emotion_label in emotion_labels:
+    os.makedirs(f'{base_dir}/{emotion_label}', exist_ok=True)
+
+!wget --no-check-certificate \
+    https://github.com/AlanTFK/Cat/releases/download/NewAgeMusic/Music.zip \
+    -O /content/Music.zip
+
+local_zip = '/content/Music.zip'
+zip_ref = zipfile.ZipFile(local_zip, 'r')
+zip_ref.extractall('/content')
+zip_ref.close()
+
+def classify_emotions(text):
+    # 首先將中文文本翻譯成英文
+    translated_text = translator(text, max_length=512)[0]['translation_text']
+    print(f"翻譯後的英文文本：{translated_text}")
+
+    # 然後對翻譯後的英文文本進行情緒分類
+    results = classifier(translated_text, return_all_scores=True)
+
+    # 提取各個情緒的分數
+    emotion_scores = {emotion: 0.0 for emotion in emotion_labels}
+    for result in results:
+        for score in result:
+            emotion = score['label']
+            if emotion in emotion_labels:
+                emotion_scores[emotion] = score['score']
+
+    return emotion_scores
+
+def get_main_emotion(emotion_scores):
+    # 找出分數最高的情緒
+    main_emotion = max(emotion_scores, key=emotion_scores.get)
+    return main_emotion
+
+def select_random_song(emotion_folder):
+    # 列出文件夾中的所有文件
+    files = os.listdir(emotion_folder)
+    # 過濾出音頻文件（例如 .mp3 文件）
+    audio_files = [file for file in files if file.endswith('.mp3')]
+    if not audio_files:
+        raise FileNotFoundError(f"No audio files found in {emotion_folder}")
+
+    # 隨機選擇一個音頻文件
+    random_song = random.choice(audio_files)
+    return os.path.join(emotion_folder, random_song)
+
+def emotion_music_recommendation(text):
+    emotion_scores = classify_emotions(text)
+    main_emotion = get_main_emotion(emotion_scores)
+    emotion_folder = os.path.join(base_dir, main_emotion)
+
+    try:
+        selected_song = select_random_song(emotion_folder)
+        return selected_song, f"檢測到的情緒：{main_emotion}"
+    except FileNotFoundError as e:
+        return str(e), "error"
+
+def play_music_interface(text):
+    audio_file, emotion = emotion_music_recommendation(text)
+    return audio_file, emotion
+
+# 創建 Gradio 介面
+iface = gr.Interface(
+    fn=play_music_interface,
+    inputs="text",
+    outputs=["audio", "text"],
+    title="🎵 New Age AI 🎵",
+    description="今天還好嗎？",
+)
+
+iface.launch(inline=False)