Update app.py

app.py

@@ -1,8 +1,64 @@
+import os
+from pathlib import Path
+import tempfile
 import gradio as gr
+from moviepy.editor import VideoFileClip, AudioFileClip
+allowed_medias = [".mp3", ".wav", ".ogg"]
+def update(file,dclass):
+    if dclass == "":
+        raise gr.Error("请输入舞蹈类型.")
+    if dappear == "":
+        raise gr.Error("请输入人物外观.")
+    file_path = Path(file.name)
+    info = {}
+    info["size"] = os.path.getsize(file_path)
+    info["name"] = file_path.name
+    print(file_path.name)
+    file_extension = file_path.suffix
+    info["type"] = "audio"
+    audio = AudioFileClip(file.name)
+    info["duration"] = audio.duration
+    info["audio_channels"] = audio.nchannels
+    filename = file_path.name
+    ######
+    audio.close()
+    #info为音频信息
+    videopath="myCoolSong.mp4"
+    #返回视频
+    if info["size"] > 100000000:
+        raise gr.Error(
+            "Please make sure all files are less than 100MB in size."
+        )
+    return videopath
 
-
-def greet(name):
-    return "Hello " + name + "!!"
-
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-print(iface.launch(share=False))
+with gr.Blocks() as demo:
+    gr.Markdown(
+        """
+            # <span style="margin-right: 0.3rem;">🏞</span>音频自动生成舞蹈
+            上传音频,系统会自动生成对应的舞蹈
+        """,
+        elem_id="header",
+    )
+    with gr.Row():
+        with gr.Column():
+            user_file = gr.File(
+                file_count="single", label="音频文件", keep_filename=True,
+                file_types=allowed_medias
+            )
+            dclass=gr.Dropdown(["流行", "民族", "混合"], label="舞蹈类型", info="")
+            dappear=gr.Dropdown(["舞蹈服", "休闲服", "混合"], label="人物外观", info="")
+            number = gr.Slider(minimum=-0, maximum=20, value=1, step=1,
+                                  interactive=True, label="人物数量")
+            btn = gr.Button("Run", label="Run")
+        with gr.Column():
+            generated_video = gr.Video(
+                interactive=False, label="舞蹈视频", include_audio=True
+            )
+            # generated_command = gr.Markdown()
+        btn.click(
+            fn=update,
+            inputs=[user_file, dclass],
+            outputs=[generated_video]
+        )
+if __name__ == "__main__":
+    demo.launch()