pi.tts

App/TTS/TTSRoutes.py

@@ -17,6 +17,7 @@ from App import pi
 from .utils.Descript import DescriptTTS
 import os
 import asyncio
+from .utils.Pi import PiAIClient
 
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import StreamingResponse, FileResponse
@@ -34,6 +35,7 @@ data = {
 
 descript_tts = DescriptTTS()
 heyGentts = HeygenAPI(**data)
+pi = PiAIClient(headless=True)
 
 
 @tts_router.post("/generate_tts")
@@ -97,6 +99,8 @@ async def search_id(req: StatusRequest):
 
 @tts_router.post("/pi_tts")
 async def pi_tts(req: PiTTSRequest):
+    if not pi.initialized:
+        await pi.setup()
     return await pi.say(text=req.text, voice=req.voice)
 
 

App/TTS/utils/Pi.py

@@ -34,6 +34,7 @@ class PiAIClient:
         self.browser = None
         self.context = None
         self.page = None
+        self.initialized = False
 
         # Define actions with their selectors and corresponding handler methods
         self.actions = [
@@ -79,6 +80,7 @@ class PiAIClient:
 
         # Mapping from sid to (Future, VoiceType)
         self.sid_futures = asyncio.Queue()
+        self.initialized = True
 
     def ensure_download_directory(self):
         """Ensure that the downloads directory exists."""

App/app.py

@@ -4,7 +4,7 @@ from fastapi.middleware.gzip import GZipMiddleware
 
 from .TTS.TTSRoutes import tts_router
 from .Embedding.EmbeddingRoutes import embeddigs_router
-from .TTS.utils.Pi import PiAIClient
+
 
 from fastapi.middleware.cors import CORSMiddleware
 
@@ -33,13 +33,11 @@ app.add_middleware(
     allow_headers=["*"],
 )
 app.add_middleware(GZipMiddleware, minimum_size=1000)
-pi = PiAIClient(headless=True)
 
 
 @app.on_event("startup")
 async def startup():
     FastAPICache.init(InMemoryBackend())
-    await pi.setup()
 
 
 @app.get("/")