Update

app.py

@@ -1,5 +1,15 @@
-# app2.py
-from gradio_interface import demo  # Direct import
+# app.py
+import sys
+import os
+
+# Add the src folder to the Python path
+src_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "src"))
+if src_path not in sys.path:
+    sys.path.append(src_path)
+
+# Import gradio_interface
+from gradio_interface import demo
+
 
 if __name__ == "__main__":
     # Launch the Gradio app

colab.ipynb

@@ -7,14 +7,21 @@
    "outputs": [],
    "source": [
     "# colab.ipynb\n",
-    "# Import necessary libraries\n",
+    "import sys\n",
     "import os\n",
+    "\n",
+    "# Add the root folder to the Python path\n",
+    "root_path = os.path.abspath(\"\")\n",
+    "if root_path not in sys.path:\n",
+    "    sys.path.append(root_path)\n",
+    "\n",
+    "# Import necessary libraries\n",
     "import random\n",
     "import ipywidgets as widgets\n",
     "from huggingface_hub import InferenceClient\n",
     "from IPython.display import display, clear_output\n",
-    "from src.img_gen_logic_colab import generate_image, save_image\n",
-    "from config.config_colab import models, prompts, api_token\n",
+    "from img_gen_logic_colab import generate_image, save_image\n",
+    "from config_colab import models, prompts, api_token  # Import from config folder\n",
     "from PIL import Image\n",
     "from google.colab import userdata\n",
     "from datetime import datetime\n",

src/gradio_interface.py

@@ -1,6 +1,6 @@
 # gradio_interface.py (HuggingFace Spaces)
 import gradio as gr
-from img_gen_logic import generate_image  # Direct import
+from src.img_gen_logic import generate_image  # Direct import
 from config.config import prompts, models  # Direct import
 
 def generate(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):

src/img_gen_logic.py

@@ -3,7 +3,7 @@ import random
 from huggingface_hub import InferenceClient
 from PIL import Image
 from datetime import datetime
-from config import api_token, models, prompts  # Direct import
+from config.config import api_token, models, prompts  # Direct import
 
 
 def generate_image(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):