Fix crash when starting the app an existing db.

CHANGES.txt

@@ -1,6 +1,12 @@
 Changelog
 =========
 
+0.5.2 - 2024-09-06
+------------------
+
+- Fix crash when starting the app an existing db.
+  [ggozad]
+
 0.5.1 - 2024-09-06
 ------------------
 

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "oterm"
-version = "0.5.1"
+version = "0.5.2"
 description = "A text-based terminal client for Ollama."
 authors = [{ name = "Yiorgis Gozadinos", email = "ggozadinos@gmail.com" }]
 license = { text = "MIT" }

src/oterm/app/oterm.py

@@ -1,3 +1,4 @@
+import asyncio
 import json
 from typing import Iterable
 
@@ -145,7 +146,12 @@ async def on_mount(self) -> None:
         store = await Store.get_store()
         self.dark = appConfig.get("theme") == "dark"
         saved_chats = await store.get_chats()
+        await self.push_screen(SplashScreen())
+        await asyncio.sleep(1.0)
+        self.pop_screen()
+
         if not saved_chats:
+            print("No saved chats, creating a new one.")
             self.action_new_chat()
         else:
             tabs = self.query_one(TabbedContent)
@@ -163,7 +169,6 @@ async def on_mount(self) -> None:
                 )
                 pane = TabPane(name, container, id=f"chat-{id}")
                 tabs.add_pane(pane)
-        await self.push_screen(SplashScreen())
 
     @on(TabbedContent.TabActivated)
     async def on_tab_activated(self, event: TabbedContent.TabActivated) -> None:

src/oterm/app/splash.py

@@ -1,5 +1,3 @@
-import asyncio
-
 from textual.app import ComposeResult
 from textual.containers import Container
 from textual.screen import ModalScreen
@@ -39,12 +37,6 @@
 
 
 class SplashScreen(ModalScreen):
-    async def remove_splash(self) -> None:
-        await asyncio.sleep(0.5)
-        self.app.pop_screen()
-
-    async def on_mount(self) -> None:
-        asyncio.create_task(self.remove_splash())
 
     def compose(self) -> ComposeResult:
         yield Header()