Create app.py

app.py

@@ -0,0 +1,244 @@
+import gradio as gr
+import json
+import torch
+from sentence_transformers import SentenceTransformer, util
+import os
+import urllib.request
+from tools.song_download import SongDownloadHelper
+
+song_download_helper = SongDownloadHelper()
+song_mid_list, song_emb_list = list(), list()
+song_info_map = {}
+MAX_REC_CNT = 10
+
+
+def load_song_info(singer_mid_list: list):
+    cnt = 0
+    with open('source/music_sub', 'r', encoding='utf-8') as f:
+        for line in f:
+            obj = json.loads(line)
+            inter = [i for i in list(obj['singer_mid']) if i in singer_mid_list]
+            if len(inter) > 0:
+                cnt += 1
+                song_info_map[obj['song_mid']] = obj
+    print('loaded song info,cnt = ', cnt)
+
+
+load_song_info(['000CK5xN3yZDJt',  # 许嵩
+                '002J4UUk29y8BY',  # 薛之谦
+                '003Nz2So3XXYek',  # 陈奕迅
+                '001BLpXF2DyJe2',  # 林俊杰
+                '0025NhlN2yWrP4',  # 周杰伦
+                '000Sp0Bz4JXH0o',  # 五月天
+                '004COQ9L4X13uj',  # 吴青峰
+                '001f0VyZ1hmWZ1',  # 林宥嘉
+                '000Q4W691sMvLG',  # 苏打绿
+                '001pWERg3vFgg8',  # 孙燕姿
+                ])
+
+
+def load_song_emb():
+    global song_mid_list, song_emb_list
+    cnt = 0
+    with open('source/song_emb', 'r', encoding='utf-8') as f:
+        for line in f:
+            obj = json.loads(line)
+            if obj['song_mid'] in song_info_map.keys():
+                cnt += 1
+                song_mid_list.append(obj['song_mid'])
+                song_emb_list.append(obj['text_emb'])
+    print('loaded song emb,cnt = ', cnt)
+
+
+load_song_emb()
+embedder = SentenceTransformer('model', device='cuda')
+song_emb_list = torch.tensor(song_emb_list)
+
+
+def download_mp3(song_mid: str):
+    file_dir = 'source/mp3/{}.mp3'.format(song_mid)
+    if not os.path.exists(file_dir):
+        song_download_helper.download(song_mid)
+    # vip音乐下载不了，文件为空，则不进行推荐
+    return os.path.getsize(file_dir)
+
+
+def get_singer_intro(singer_mid: str):
+    url_ = 'https://y.qq.com/n/ryqq/singer/{}'.format(singer_mid)
+    res_ = urllib.request.urlopen(url_)
+    html_ = str(res_.read().decode('utf-8'))
+    return '\n'.join(str(html_).split('shortdesc":"')[1].split('","')[0].split('\\n'))[0:64]
+
+
+def get_album_intro(song_mid: str):
+    try:
+        url_ = 'https://y.qq.com/n/ryqq/songDetail/{}'.format(song_mid)
+        print(url_)
+        res_ = urllib.request.urlopen(url_)
+        html_ = str(res_.read().decode('utf-8'))
+        print(html_)
+        album_id = str(html_).split('albumDetail/')[1].split('"')[0]
+        url_ = 'https://y.qq.com/n/ryqq/albumDetail/{}'.format(album_id)
+        print(url_)
+        res_ = urllib.request.urlopen(url_)
+        html_ = str(res_.read().decode('utf-8'))
+        print(html_)
+        return album_id, '\n'.join(str(html_).split('简介')[1].split('<p>')[1].split('</p>')[0].split('\\n'))[0:64]
+    except Exception as e:
+        return None, None
+
+
+def generate_playlist(prompt):
+    prompt_embedding = embedder.encode(prompt, convert_to_tensor=True)
+    hits = util.semantic_search(prompt_embedding, song_emb_list, top_k=100)
+    song_names = list()
+    for hit_info in hits[0]:
+        song_mid = song_mid_list[hit_info['corpus_id']]
+        song_info = song_info_map[song_mid]
+        if not download_mp3(song_mid):
+            continue
+        print(song_mid, hit_info['score'], song_info)
+        song_names.append('✨'.join(
+            [str(song_info['song_name']), '/'.join(list(song_info['singer_name'])), str(song_mid)]))
+        if len(song_names) >= MAX_REC_CNT:
+            break
+    return gr.Radio.update(label='点击选择一首歌吧', show_label=True, choices=song_names)
+
+
+def set_song_info(full_title):
+    song_name, singer_name, song_mid = str(full_title).split('✨')
+    print(song_name, singer_name, song_mid)
+    song_info = song_info_map[song_mid]
+    title_markdown_ = gr.Markdown.update(
+        value='✨**播放歌曲**✨ [歌曲详情]({})'.format(song_info['song_url']))
+    hot_comment = list(song_info['hot_comments'])[0]
+    comment_textbox_ = gr.Textbox.update(
+        value='来自用户 ✨{}✨：\n\n{}'.format(hot_comment['comment_name'], hot_comment['comment_text']))
+    lyrics_textbox_ = gr.Textbox.update(value='\n'.join(str(song_info['lyric']).split('\\n')))
+
+    return [title_markdown_, comment_textbox_, lyrics_textbox_]
+
+
+def set_singer_info(full_title):
+    song_name, singer_name, song_mid = str(full_title).split('✨')
+    print(song_name, singer_name, song_mid)
+    song_info = song_info_map[song_mid]
+    singer_mid = list(song_info['singer_mid'])[0]
+    singer_markdown_ = gr.Markdown.update(
+        value='{}：{}… [更多]({})'.format(list(song_info['singer_name'])[0],
+                                         get_singer_intro(singer_mid),
+                                         'https://y.qq.com/n/ryqq/singer/{}'.format(singer_mid)))
+    singer_image_ = gr.Image.update(value='https://y.qq.com/music/photo_new/T001R300x300M000{}.jpg'.format(singer_mid)
+                                    , visible=True, show_label=False)
+
+    return [singer_markdown_, singer_image_]
+
+
+def set_album_info(full_title):
+    song_name, singer_name, song_mid = str(full_title).split('✨')
+    album_id, album_tro = get_album_intro(song_mid)
+    song_info = song_info_map[song_mid]
+    if album_id:
+        album_markdown_ = gr.Markdown.update(
+            value='《{}》：{}… [更多]({})'.format(song_info['album_name'], album_tro,
+                                               'https://y.qq.com/n/ryqq/albumDetail/{}'.format(album_id)))
+        album_image_ = gr.Image.update(value='https://y.qq.com/music/photo_new/T002R300x300M000{}.jpg'.format(album_id),
+                                       visible=True, show_label=False)
+    else:
+        album_markdown_ = gr.Markdown.update(value='无专辑')
+        album_image_ = gr.Image.update(visible=False)
+    return [album_markdown_, album_image_]
+
+
+def set_mp3_audio(full_title):
+    song_name, singer_name, song_mid = str(full_title).split('✨')
+    mp3_audio_ = gr.Audio.update(label=song_name, value='source/mp3/{}.mp3'.format(song_mid), show_label=True)
+    return mp3_audio_
+
+
+def set_example_prompt(example):
+    return gr.TextArea.update(value=example[0])
+
+
+demo = gr.Blocks()
+
+with demo:
+    gr.Markdown(
+        """
+        # 播放列表生成器 📻 🎵
+        """)
+
+    with gr.Row():
+        with gr.Column():
+            song_prompt = gr.TextArea(
+                value="狂野而自由地奔跑",
+                label="输入你的此刻的心情（或选择示例）"
+            )
+            gr.Markdown(value='✨**推荐歌单**✨:')
+            song_option = gr.Radio(show_label=False, interactive=True, choices=None)
+            with gr.Row():
+                fetch_songs = gr.Button(value="点击为您推荐播放列表 🧑 🏽 ‍🎤").style(full_width=True)
+                example_prompts = gr.Dataset(
+                    label='选择示例',
+                    components=[song_prompt],
+                    samples=[
+                        ["我怀念过去"],
+                        ["狂野而自由地奔跑"],
+                        ["有时候觉得没人懂"],
+                    ]
+                )
+        with gr.Column():
+            title_markdown = gr.Markdown(value='✨**播放歌曲**✨')
+            mp3_audio = gr.Audio(show_label=False)
+            gr.Markdown(value='✨**热评**✨:')
+            comment_textbox = gr.Textbox(show_label=False, placeholder="选择一首歌曲以查看其歌热评")
+            gr.Markdown(value='✨**歌词**✨:')
+            lyrics_textbox = gr.Textbox(show_label=False, placeholder="选择一首歌曲以查看其歌词")
+        with gr.Column():
+            gr.Markdown(value='✨**歌手介绍**✨:')
+            with gr.Row():
+                singer_markdown = gr.Markdown()
+                singer_image = gr.Image(show_label=False, visible=True)
+            gr.Markdown(value='✨**专辑介绍**✨:')
+            with gr.Row():
+                album_markdown = gr.Markdown()
+                album_image = gr.Image(show_label=False, visible=True)
+
+    fetch_songs.click(
+        fn=generate_playlist,
+        inputs=[song_prompt],
+        outputs=song_option
+    )
+
+    example_prompts.click(
+        fn=set_example_prompt,
+        inputs=example_prompts,
+        outputs=example_prompts.components,
+    )
+
+    song_option.change(
+        fn=set_song_info,
+        inputs=song_option,
+        outputs=[title_markdown, comment_textbox, lyrics_textbox]
+    )
+
+    song_option.change(
+        fn=set_mp3_audio,
+        inputs=song_option,
+        outputs=mp3_audio
+    )
+
+    song_option.change(
+        fn=set_singer_info,
+        inputs=song_option,
+        outputs=[singer_markdown, singer_image]
+    )
+
+    song_option.change(
+        fn=set_album_info,
+        inputs=song_option,
+        outputs=[album_markdown,
+                 album_image]
+    )
+
+demo.launch()