From c7196af52345dc916e8e95bb14adbffc7a86ec43 Mon Sep 17 00:00:00 2001
From: Xclbr <xclbrmusic@gmail.com>
Date: Mon, 2 Sep 2024 18:10:07 +0530
Subject: [PATCH] Update __init__.py

---
 __init__.py | 23 +++++++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index deef1c1..39c9c26 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,3 +1,22 @@
-from .magic_photo_prompt import NODE_CLASS_MAPPINGS
+import os
+import sys
+import subprocess
 
-__all__ = ['NODE_CLASS_MAPPINGS']
+# Try to install the Google AI SDK if it's not already installed
+try:
+    import google.generativeai as genai
+except ImportError:
+    print("Google AI SDK not found. Attempting to install...")
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "google-generativeai"])
+    import google.generativeai as genai
+
+# Import the node classes from separate files
+from .gemini_prompt_expander import GeminiPromptExpander, NODE_CLASS_MAPPINGS as GEMINI_NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+from .magic_photo_prompt import NODE_CLASS_MAPPINGS as MAGIC_PHOTO_NODE_CLASS_MAPPINGS
+
+WEB_DIRECTORY = "./web"
+
+# Merge the NODE_CLASS_MAPPINGS dictionaries
+NODE_CLASS_MAPPINGS = {**GEMINI_NODE_CLASS_MAPPINGS, **MAGIC_PHOTO_NODE_CLASS_MAPPINGS}
+
+__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]