From 8c76140b38b239ad080d940da4a5c49973c624ac Mon Sep 17 00:00:00 2001
From: Paillat <me@paillat.dev>
Date: Thu, 22 Feb 2024 15:12:56 +0100
Subject: [PATCH] Refactor UI module and add launcher function

---
 main.py        | 5 ++---
 ui/__init__.py | 1 +
 ui/launcher.py | 4 ++++
 3 files changed, 7 insertions(+), 3 deletions(-)
 create mode 100644 ui/launcher.py

diff --git a/main.py b/main.py
index f32ba5e..bfeeccc 100644
--- a/main.py
+++ b/main.py
@@ -1,4 +1,4 @@
-from ui import GenerateUI
+from ui import launch
 from dotenv import load_dotenv
 from src.models import init_db
 
@@ -6,5 +6,4 @@ load_dotenv()
 
 if __name__ == "__main__":
     init_db()
-    ui_generator = GenerateUI()
-    ui_generator.launch_ui()
+    launch()
\ No newline at end of file
diff --git a/ui/__init__.py b/ui/__init__.py
index 1fa9572..f8a2755 100644
--- a/ui/__init__.py
+++ b/ui/__init__.py
@@ -1 +1,2 @@
 from .gradio_ui import GenerateUI
+from .launcher import launch
\ No newline at end of file
diff --git a/ui/launcher.py b/ui/launcher.py
new file mode 100644
index 0000000..af7778a
--- /dev/null
+++ b/ui/launcher.py
@@ -0,0 +1,4 @@
+from . import GenerateUI
+def launch():
+    ui_generator = GenerateUI()
+    ui_generator.launch_ui()