separate workers

App/Generate/database/Model.py

@@ -10,7 +10,7 @@ from typing import List
 from pydantic import BaseModel
 import json
 
-database_url = "sqlite+aiosqlite:///ok.db"
+database_url = f"sqlite+aiosqlite:///{str(uuid.uuid4())}.db"
 database = databases.Database(database_url)
 models = orm.ModelRegistry(database=database)
 
@@ -208,8 +208,9 @@ class Scene(orm.Model):
         print("Failed to generate narration after 3 attempts.")
 
     def calculate_durations(self):
-        wav_file = AudioSegment.from_file(self.narration_path, format="wav")
-        self.narration_duration = int(len(wav_file) / 1000)
+        file_format = self.narration_path.split(".")[-1]
+        audio_file = AudioSegment.from_file(self.narration_path, format=file_format)
+        self.narration_duration = int(len(audio_file) / 1000)
         self.image_duration = self.narration_duration / len(self.image_prompts)
 
     async def generate_images(self):

App/app.py

@@ -2,7 +2,12 @@ from fastapi import FastAPI, BackgroundTasks
 from .Editor.editorRoutes import videditor_router
 from App import bot
 from App.utilis import WorkerClient, SERVER_STATE
-from .Generate.generatorRoutes import generator_router, database, database_url, models
+from .Generate.generatorRoutes import (
+    generator_router,
+    database,
+    models,
+)
+import uuid
 
 app = FastAPI()
 manager = WorkerClient()
@@ -10,11 +15,17 @@ manager = WorkerClient()
 
 @app.on_event("startup")
 async def startup_event():
+    app.state.db = database
+    app.state.models = models
+
     try:
-        await models._create_all(database_url)
+        # print(type(database.url), database_url)
+        # await models.create_all()
+        await models._create_all(str(database.url))
     except:
-        pass
+        print("failed to create")
     finally:
+        print(database.is_connected)
         if not database.is_connected:
             await database.connect()
         await database.execute("pragma journal_mode=wal;")