Update backend/open_webui/apps/audio/main.py

backend/open_webui/apps/audio/main.py

@@ -253,8 +253,8 @@ async def speech(request: Request, user=Depends(get_verified_user)):
             body = body.decode("utf-8")
             body = json.loads(body)
             body["model"] = app.state.config.TTS_MODEL
-            query=body["input"]
-            print(query)
+            #query=body["input"]
+            #print(query)
             body = json.dumps(body).encode("utf-8")
         except Exception:
             pass
@@ -264,12 +264,13 @@ async def speech(request: Request, user=Depends(get_verified_user)):
             client = Client("https://s5k.cn/api/v1/studio/iic/CosyVoice-300M-WAIC/gradio/")
             r= client.predict(
 		          _sound_radio="中文女",
-		          _synthetic_input_textbox={"query":query},
+		         # _synthetic_input_textbox={"query":query},
+                  _synthetic_input_textbox=body,
 		          _seed=0,
 		          api_name="/generate_audio"
 
             )
-            print(r)
+            #print(r)
             #r = requests.post(
                # url=f"{app.state.config.TTS_OPENAI_API_BASE_URL}/audio/speech",
                 #data=body,
@@ -277,12 +278,13 @@ async def speech(request: Request, user=Depends(get_verified_user)):
                # stream=True,
             #)
 
-           # r.raise_for_status()
+            r.raise_for_status()
 
             # Save the streaming content to a file
             with open(file_path, "wb") as f:
-               # for chunk in r.iter_content(chunk_size=8192):
-                    f.write(r.content)
+               for chunk in r.iter_content(chunk_size=8192):
+                    f.write(chunk)
+                   #f.write(r.content)
 
             with open(file_body_path, "w") as f:
                 json.dump(json.loads(body.decode("utf-8")), f)