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()