startup

App/Generate/generatorRoutes.py

@@ -18,13 +18,6 @@ async def update_scene(model_scene):
 async def main(request: GeneratorRequest):
     topic = request.prompt
     renderr = RenderVideo()
-    try:
-        await models._create_all(database_url)
-    except:
-        pass
-    finally:
-        if not database.is_connected:
-            await database.connect()
     message = chatbot(Prompt.format(topic=topic))
 
     generated_story = Story.from_dict(message["scenes"])

App/app.py

@@ -2,7 +2,7 @@ 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
+from .Generate.generatorRoutes import generator_router, database, database_url, models
 
 app = FastAPI()
 manager = WorkerClient()
@@ -10,11 +10,19 @@ manager = WorkerClient()
 
 @app.on_event("startup")
 async def startup_event():
+    try:
+        await models._create_all(database_url)
+    except:
+        pass
+    finally:
+        if not database.is_connected:
+            await database.connect()
+
     await bot.start()
-    #if SERVER_STATE.MASTER:
+    # if SERVER_STATE.MASTER:
 
-    #response = await manager.register_worker()
-    #if not response:
+    # response = await manager.register_worker()
+    # if not response:
     #    print("Error registering worker")