Update app.py

app.py

@@ -2,28 +2,43 @@ import gradio as gr
 import os,secrets,requests
 from basic_pitch.inference import predict_and_save
 from urllib.parse import quote
-os.system('chmod +x ./pget')
+import deta
+from deta import Deta  # Import Deta
+
+# Initialize
+deta = Deta()
 
+# This how to connect to or create a database.
+last= deta.Base("last")
+records=deta.Base("records")
+files=deta.Drive('music')
+
+os.system('chmod +x ./pget')
 
 
+def addrecord(orig,pitmp3,mid,gname,ghint):
+    if not gname or not ghint:
+        return 'E/缺少name或hint.'
+    names=gname.split('\n')
+    lastkey=last.get('last')
+    if not lastkey:
+        thiskey=0
+    else:
+        thiskey=lastkey['next']
+    #upload file src
+    files.put(name=orig,path=orig)
+    files.put(name=pitmp3,path=pitmp3)
+    files.put(name=mid,path=mid)
+    records.put({"name":names,"hint":ghint,"orig":orig,"pitmp3":pitmp3,"mid":mid},str(thiskey))
+    last.put({'next':thiskey+1},'last')
+    return 'OK'
+    
 
 
-def downloadaud(query):
+def downloadaud(video_id,gname,ghint):
     filename=secrets.token_hex()
     # Step 1: Search for videos with the given query
-    search_url = f"https://draw-8fj.begin.app/api/search/{quote(query)}"
-    search_response = requests.get(search_url).json()
-    print('1=>', search_response)
-    # Step 2: Find the first video with duration less than 10 minutes and extract its ID
-    video_id = None
-    #search_response = sorted(search_response, key=lambda x: x["views"], reverse=True)
-    for item in search_response:
-        duration = item.get("duration_raw")
-        if duration and len(duration.split(':'))< 3 and int(duration.split(':')[0])<10:
-            video_id = item.get("id", {}).get("videoId")
-            break
-    print('1-r',video_id)
-    # If no video with duration less than 10 minutes was found, return None
+
     if not video_id:
         return None
 
@@ -51,8 +66,9 @@ def downloadaud(query):
     os.system(command)
     os.system(f'ffmpeg -i {filename}.mp4 -q:a 0 -map a {filename}.mp3')
     os.system(f'mv {filename}.mp3 /tmp/{filename}.mp3')
-    res= predict(f'/tmp/{filename}.mp3')
-    return res
+    o,m,i= predict(f'/tmp/{filename}.mp3')
+    res=addrecord(o,m,i,gname,ghint)
+    return o,m,i,res
 
 
 def predict(ina):
@@ -67,13 +83,18 @@ def predict(ina):
 )
     ina='.'.join(ina.split('.')[:-1])
     
-    ina=ina+'_basic_pitch'
+    ina=ina+'_t_'
     os.system(f'ffmpeg -i "{ina}.wav" -f mp3 -acodec libmp3lame -y "{ina}.mp3"')
     
     return f'{fn}',f'{ina}.mp3',f'{ina}.mid'
-aud=gr.Textbox(label="yt query")
+
+
+aud=gr.Textbox(label="视频ID")
+g_name=gr.Textarea(label="游戏名称（谜底）,回车分名称,一行一个,最多20个")
+g_hint=gr.Textarea(label="开发厂商（提示）")
 oar=gr.Audio(label="下载的音频")
 oa=gr.Audio(label="预览midi")
 pf=gr.File(label="下载midi")
-demo = gr.Interface(fn=downloadaud, inputs=[aud], outputs=[oar,oa,pf])
+res=gr.Textbox(label="结果")
+demo = gr.Interface(fn=downloadaud, inputs=[aud,g_name,g_hint], outputs=[oar,oa,pf,res])
 demo.launch()