Refactor UI module and add launcher function

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()
+    launch()
-    ui_generator.launch_ui()

ui/__init__.py

@@ -1 +1,2 @@
 from .gradio_ui import GenerateUI
+from .launcher import launch

ui/launcher.py

@@ -0,0 +1,4 @@
+from . import GenerateUI
+def launch():
+    ui_generator = GenerateUI()
+    ui_generator.launch_ui()