fix image_app gens table creation/insert

Author: mjaniszewski

image_app_session_credits/main.py

@@ -17,8 +17,8 @@
 
 # gens database for storing generated image details
 tables = database('data/gens.db').t
-gens = tables.gens
-if not gens in tables:
+gens = tables.t.gens
+if not gens in tables.t:
     gens.create(prompt=str, session_id=str, id=int, folder=str, pk='id')
 Generation = gens.dataclass()
 
@@ -104,9 +104,8 @@ def post(prompt: str, session):
     folder = f"data/gens/{str(uuid.uuid4())}"
     os.makedirs(folder, exist_ok=True)
     g = gens.insert(
-        Generation(prompt=prompt,
-                    folder=folder,
-                    session_id=session['session_id']))
+        Generation(prompt=prompt, folder=folder, session_id=session['session_id'], id=f"{gens.count}")
+    )
     generate_and_save(g.prompt, g.id, g.folder)
 
     return generation_preview(g, session), clear_input

image_app_session_credits/session.py

@@ -9,8 +9,8 @@
 
 # gens database for storing generated image details
 tables = database('gens.db').t
-gens = tables.gens
-if not gens in tables:
+gens = tables.t.gens
+if not gens in tables.t:
     gens.create(prompt=str, session_id=str, id=int, folder=str, pk='id')
 Generation = gens.dataclass()
 
@@ -61,7 +61,9 @@ async def static(fname:str, ext:str): return FileResponse(f'{fname}.{ext}')
 async def post(prompt:str, session):
     folder = f"gens/{str(uuid.uuid4())}"
     os.makedirs(folder, exist_ok=True)
-    g = gens.insert(Generation(prompt=prompt, folder=folder, session_id=session['session_id']))
+    g = gens.insert(
+        Generation(prompt=prompt, folder=folder, session_id=session['session_id'], id=f"{gens.count}")
+    )
     generate_and_save(g.prompt, g.id, g.folder)
     clear_input =  Input(id="new-prompt", name="prompt", placeholder="Enter a prompt", hx_swap_oob='true')
     return generation_preview(g), clear_input