Init5

api_get.py

@@ -6,6 +6,7 @@ import time
 import base64
 import urllib
 import urllib3
+import gradio as gr
 # 加密函数
 def hmac_sha256_encrypt(message, key):
     message = message.encode('utf-8')
@@ -44,11 +45,37 @@ def  pre_request():
 #   // 设置本次请求的签名签名串
   global reqSignature
   reqSignature = encodeSign;
-  print(reqTimestamp, reqSignature, appkey)
+  # print(reqTimestamp, reqSignature, appkey)
 
+def req_audioMake(text):
+  pre_request()
+  url = "https://gw.tvs.qq.com/v2/ivh/videomaker/broadcastservice/tts?timestamp="+reqTimestamp+"&signature="+reqSignature+"&appkey="+appkey
+
+  payload = json.dumps({
+    "Header": {},
+    "Payload": {
+      "TimbreKey": "",
+      "VirtualmanKey": "36b88108f6de6da255534eaf4363c7d7",
+      "InputSsml": text,
+      "Speed": 1
+    }
+  })
+  headers = {
+    'Content-Type': 'application/json'
+  }
+
+  response = requests.request("POST", url, headers=headers, data=payload)
+  
+  res = json.loads(response.text)
+  if res["Header"]["Code"] != 0:
+    print(response.text)
+    return 
 
+  taskID = (res["Payload"]["TaskId"])
+  print(taskID)
+  return taskID
 
-def get_taskID(text = "你好"):
+def req_videoMake(text = "你好"):
     pre_request()
     taskID = ""
 
@@ -74,19 +101,21 @@ def get_taskID(text = "你好"):
     headers = {
     'Content-Type': 'application/json'
     }
-    print("1111111111")
+
     response = requests.request("POST", url, headers=headers, data=payload)
-    time.sleep(1)
-    print("11111111112222222222")
 
-    # print(json.loads(response.text)["Payload"]["TaskId"])
-    taskID = (json.loads(response.text)["Payload"]["TaskId"])
+    res = json.loads(response.text)
+    if res["Header"]["Code"] != 0:
+      print(response.text)
+      return 
 
+    taskID = (res["Payload"]["TaskId"])
+    print(taskID)
     return taskID
 
 
 
-def get_video(taskID):
+def get_media(taskID):
 
   pre_request()
   url = "https://gw.tvs.qq.com/v2/ivh/videomaker/broadcastservice/getprogress?timestamp="+reqTimestamp+"&signature="+reqSignature+"&appkey="+appkey
@@ -103,22 +132,59 @@ def get_video(taskID):
 
   response = requests.request("POST", url, headers=headers, data=payload)
 
-  print(response.text)
+  # print(response.text)
   return response.text
 
 
 def generate_video(text = "你好"):
-    taskID = get_taskID(text)
-    print(taskID)
-
-    status = json.loads(get_video(taskID))["Payload"]["Status"]
-    while status != "SUCCESS":
-        time.sleep(5)
-        status = json.loads(get_video(taskID))["Payload"]["Status"]
+    taskID = req_videoMake(text)
+    # print(taskID)
+
+    sleepSecond = 5
+    time.sleep(sleepSecond)
+    videoRep = json.loads(get_media(taskID))
+ 
+    status = videoRep["Payload"]["Status"]
+
+    while status == "COMMIT" or status == "MAKING":
+        time.sleep(1)
+        videoRep = json.loads(get_media(taskID))
+        status = videoRep["Payload"]["Status"]
         print(status)
 
-    mediaUrl = json.loads(get_video(taskID))["Payload"]["MediaUrl"]
+    if status == "FAIL":
+        print(videoRep)
+        return
+ 
+    mediaUrl = videoRep["Payload"]["MediaUrl"]
 
     return mediaUrl
 
 
+def generate_audio(text):
+    taskID = req_audioMake(text)
+    # print(taskID)
+
+    sleepSecond = 5
+    time.sleep(sleepSecond)
+    videoRep = json.loads(get_media(taskID))
+ 
+    status = videoRep["Payload"]["Status"]
+
+    while status == "COMMIT" or status == "MAKING":
+        time.sleep(1)
+        videoRep = json.loads(get_media(taskID))
+        status = videoRep["Payload"]["Status"]
+        print(status)
+
+    if status == "FAIL":
+        print(videoRep)
+        return
+ 
+    mediaUrl = videoRep["Payload"]["MediaUrl"]
+
+    return mediaUrl
+
+  
+  
+  

app.py

@@ -7,15 +7,12 @@ from modules.ui_components import FormRow, FormGroup, ToolButton, FormHTML
 # c = os.path.join(os.path.dirname(__file__), "files/b.mp4")  # Video
 clear_prompt_symbol = '\U0001f5d1\ufe0f'  # 🗑️
 def generate_video(text = "<break time=\"5000\"/>，大家奖fsdfdf，房间号是吗？<break time=\"5000\"/>"):
+    # gr.Warning("fjskdf=======")
     return api_get.generate_video( text = text)
 
-
-
-# demo = gr.Interface(fn=generate_video, 
-#                 inputs=gr.Textbox(placeholder = "输入文案", label = "文字转视频"), 
-#                 outputs=gr.Video(type="mp4", label = "视频"),
-#                  title="文字转视频", 
-#                  description="输入文案，生成视频")
+def generate_audio(text = "<break time=\"5000\"/>，大家奖fsdfdf，房间号是吗？<break time=\"5000\"/>"):
+    # gr.Warning("fjskdf=======")
+    return api_get.generate_audio( text = text)
 
 
 with gr.Blocks() as demo:
@@ -24,26 +21,35 @@ with gr.Blocks() as demo:
         id_part = "text_to_video"
         with gr.Row():
             with gr.Column():
-                text_input = gr.Textbox(placeholder = "输入文案", label = "文字")
+                text2video_input = gr.Textbox(placeholder = "输入文案", label = "文字")
                 with gr.Row():
-                    text_button2 = gr.Button("插入停顿",  variant='secondary')
-                    text_button3 = gr.Button("插入数量",  variant='secondary')
-                    text_button3 = gr.Button("插入动作",  variant='secondary')
-                    cancel_button = gr.ClearButton(variant='stop')
-                    text_button = gr.Button("生成",  variant='primary')
+                    text2video_button2 = gr.Button("插入停顿",  variant='secondary')
+                    text2video_button3 = gr.Button("插入数量",  variant='secondary')
+                    cancel_button1 = gr.ClearButton(variant='stop')
+                    text2video_button = gr.Button("生成",  variant='primary')
 
             video_output = gr.Video(type="mp4", label = "视频")
-            text_button.click(generate_video, inputs=text_input, outputs=video_output)   
-            cancel_button.click(lambda x: "",  outputs=text_input)   
-            text_button2.click(lambda x: x+"<break time=\"5000\"/>,", inputs = text_input,  outputs=text_input)
+            text2video_button.click(generate_video, inputs=text2video_input, outputs=video_output)   
+            cancel_button1.click(lambda x: "",  outputs=text2video_input)   
+            text2video_button2.click(lambda x: x+"<break time=\"5000\"/>,", inputs = text2video_input,  outputs=text2video_input)
+            text2video_button3.click(lambda x: x+"<say-as interpret-as=\"cardinal\">123456</say-as>,", inputs = text2video_input,  outputs=text2video_input)
     with gr.Tab("文生音频"):
-        with gr.Column():
-            text_input1 = gr.Textbox(placeholder = "输入文案", label = "文字")
-            with gr.Row():
-                cancel_button1 = gr.ClearButton()
-                text_button1 = gr.Button("生成",  variant='primary')
-            video_outpu1t = gr.Video(type="mp4", label = "视频")
+        id_part = "text_to_audio"
+        with gr.Row():
+            with gr.Column():
+                text2audio_input = gr.Textbox(placeholder = "输入文案", label = "文字")
+                with gr.Row():
+                    text2audio_button2 = gr.Button("插入停顿",  variant='secondary')
+                    text2audio_button3 = gr.Button("插入数量",  variant='secondary')
+                    cancel_button2 = gr.ClearButton(variant='stop')
+                    text2audio_button = gr.Button("生成",  variant='primary')
+
+            audio_output = gr.Video(type="mp3", label = "音频")
+            text2audio_button.click(generate_audio, inputs=text2audio_input, outputs=audio_output)   
+            cancel_button2.click(lambda x: "",  outputs=text2audio_input)   
+            text2audio_button2.click(lambda x: x+"<break time=\"5000\"/>,", inputs = text2audio_input,  outputs=text2audio_input)
+            text2audio_button3.click(lambda x: x+"<say-as interpret-as=\"cardinal\">123456</say-as>,", inputs = text2audio_input,  outputs=text2audio_input)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_port = 8080)