“Update”

app.py

@@ -3,6 +3,7 @@ import sys
 import os
 
 # Add the src folder to the Python path
+# Solves all problems w subfolders - option2
 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)

config/config.py

@@ -1,13 +1,9 @@
 # config.py
 import os
 from src.prompts import prompts  # Import prompts from prompts.py
+from config.models import models
 
 # Retrieve the Hugging Face token
-#try:
-    # Try to get the token from Colab secrets
-#    api_token = userdata.get("HF_CTB_TOKEN")
-#except ImportError:
-    # Fall back to environment variable (for local execution)
 api_token = os.getenv("HF_CTB_TOKEN")
 
 # Debugging: Check if the Hugging Face token is available
@@ -17,12 +13,6 @@ else:
     print("Hugging Face token loaded successfully.")
 
 
-# List of models with aliases
-models = [
-    {"alias": "FLUX.1-dev", "name": "black-forest-labs/FLUX.1-dev"},
-    {"alias": "Midjourney", "name": "strangerzonehf/Flux-Midjourney-Mix2-LoRA"}
-]
-
 # Debugging: Print prompt and model options
 print("Prompt Options:", [p["alias"] for p in prompts])
 print("Model Options:", [m["alias"] for m in models])

config/config_colab.py

@@ -1,6 +1,6 @@
 # config_colab.py
 from google.colab import userdata
-from src.prompts import prompts  # Import prompts from prompts.py
+from config.prompts import prompts  # Import prompts from prompts.py
 
 # Retrieve the Hugging Face token from Colab secrets
 api_token = userdata.get("HF_CTB_TOKEN")
@@ -13,12 +13,6 @@ else:
     print("=== Debug: Success ===")
     print("Hugging Face token loaded successfully.")
 
-# List of models with aliases
-models = [
-    {"alias": "FLUX.1-dev", "name": "black-forest-labs/FLUX.1-dev"},
-    {"alias": "Midjourney", "name": "strangerzonehf/Flux-Midjourney-Mix2-LoRA"}
-]
-
 # Debugging: Print prompt and model options
 print("=== Debug: Available Options ===")
 print("Prompt Options:", [p["alias"] for p in prompts])

config/models.py

@@ -0,0 +1,5 @@
+# List of models with aliases
+models = [
+    {"alias": "FLUX.1-dev", "name": "black-forest-labs/FLUX.1-dev"},
+    {"alias": "Midjourney", "name": "strangerzonehf/Flux-Midjourney-Mix2-LoRA"}
+]

src/gradio_interface.py

@@ -1,16 +1,7 @@
 # gradio_interface.py (HuggingFace Spaces)
 import gradio as gr
-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):
-    try:
-        # Generate the image
-        image_path, message = generate_image(prompt_alias, team, model_alias, custom_prompt, height, width, num_inference_steps, guidance_scale, seed)
-        return image_path, message
-    except Exception as e:
-        return None, f"An error occurred: {e}"
-
 # Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("# CtB AI Image Generator")

src/{img_gen_logic.py → img_gen.py}

@@ -1,10 +1,17 @@
-# img_gen_logic.py
+# img_gen.py
 import random
 from huggingface_hub import InferenceClient
-from PIL import Image
 from datetime import datetime
 from config.config import api_token, models, prompts  # 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):
+    try:
+        # Generate the image
+        image_path, message = generate_image(prompt_alias, team, model_alias, custom_prompt, height, width, num_inference_steps, guidance_scale, seed)
+        return image_path, message
+    except Exception as e:
+        return None, f"An error occurred: {e}"
+
 
 def generate_image(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):
     # Debugging: Check if the token is available