From 4ad2ad468e6bd29055dc6a66f006bab6f9d7f4c7 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Thu, 27 Nov 2025 10:10:11 -0800
Subject: [PATCH] Add basic structure

---
 modules/ui_image_generation.py | 9 +++++++++
 server.py                      | 2 ++
 2 files changed, 11 insertions(+)
 create mode 100644 modules/ui_image_generation.py

diff --git a/modules/ui_image_generation.py b/modules/ui_image_generation.py
new file mode 100644
index 00000000..389bc1e2
--- /dev/null
+++ b/modules/ui_image_generation.py
@@ -0,0 +1,9 @@
+import gradio as gr
+
+
+def create_ui():
+    pass
+
+
+def create_event_handlers():
+    pass
diff --git a/server.py b/server.py
index c804c342..87bbdc4a 100644
--- a/server.py
+++ b/server.py
@@ -50,6 +50,7 @@ from modules import (
     ui_chat,
     ui_default,
     ui_file_saving,
+    ui_image_generation,
     ui_model_menu,
     ui_notebook,
     ui_parameters,
@@ -163,6 +164,7 @@ def create_interface():
         ui_chat.create_character_settings_ui()  # Character tab
         ui_model_menu.create_ui()  # Model tab
         if not shared.args.portable:
+            ui_image_generation.create_ui()  # Image generation tab
             training.create_ui()  # Training tab
         ui_session.create_ui()  # Session tab