Update app.py

app.py

@@ -2,12 +2,18 @@ import streamlit as st
 from gradio_client import Client
 import time
 import concurrent.futures
+import os
+
+# Get token from environment variable
+HF_TOKEN = os.getenv('ArtToken')
+if not HF_TOKEN:
+    raise ValueError("Please set the 'ArtToken' environment variable with your Hugging Face token")
 
 class ModelGenerator:
     @staticmethod
     def generate_midjourney(prompt):
         try:
-            client = Client("mukaist/Midjourney")
+            client = Client("mukaist/Midjourney", hf_token=HF_TOKEN)
             result = client.predict(
                 prompt=prompt,
                 negative_prompt="(deformed iris, deformed pupils, semi-realistic, cgi, 3d, render, sketch, cartoon, drawing, anime:1.4), text, close up, cropped, out of frame, worst quality, low quality, jpeg artifacts, ugly, duplicate, morbid, mutilated, extra fingers, mutated hands, poorly drawn hands, poorly drawn face, mutation, deformed, blurry, dehydrated, bad anatomy, bad proportions, extra limbs, cloned face, disfigured, gross proportions, malformed limbs, missing arms, missing legs, extra arms, extra legs, fused fingers, too many fingers, long neck",
@@ -27,7 +33,7 @@ class ModelGenerator:
     @staticmethod
     def generate_stable_cascade(prompt):
         try:
-            client = Client("multimodalart/stable-cascade")
+            client = Client("multimodalart/stable-cascade", hf_token=HF_TOKEN)
             result = client.predict(
                 prompt=prompt,
                 negative_prompt=prompt,
@@ -48,7 +54,7 @@ class ModelGenerator:
     @staticmethod
     def generate_stable_diffusion_3(prompt):
         try:
-            client = Client("stabilityai/stable-diffusion-3-medium")
+            client = Client("stabilityai/stable-diffusion-3-medium", hf_token=HF_TOKEN)
             result = client.predict(
                 prompt=prompt,
                 negative_prompt=prompt,
@@ -67,7 +73,7 @@ class ModelGenerator:
     @staticmethod
     def generate_stable_diffusion_35(prompt):
         try:
-            client = Client("stabilityai/stable-diffusion-3.5-large")
+            client = Client("stabilityai/stable-diffusion-3.5-large", hf_token=HF_TOKEN)
             result = client.predict(
                 prompt=prompt,
                 negative_prompt=prompt,
@@ -86,7 +92,7 @@ class ModelGenerator:
     @staticmethod
     def generate_playground_v2_5(prompt):
         try:
-            client = Client("https://playgroundai-playground-v2-5.hf.space/--replicas/ji5gy/")
+            client = Client("https://playgroundai-playground-v2-5.hf.space/--replicas/ji5gy/", hf_token=HF_TOKEN)
             result = client.predict(
                 prompt,
                 prompt,  # negative prompt
@@ -127,6 +133,10 @@ def generate_images(prompt, selected_models):
     return results
 
 def handle_prompt_click(prompt_text, key):
+    if not HF_TOKEN:
+        st.error("Environment variable 'ArtToken' is not set!")
+        return
+        
     st.session_state[f'selected_prompt_{key}'] = prompt_text
     
     selected_models = st.session_state.get('selected_models', [])
@@ -144,6 +154,15 @@ def main():
     st.title("🎨 Multi-Model Art Generator")
 
     with st.sidebar:
+        st.header("Configuration")
+        
+        # Show token status
+        if HF_TOKEN:
+            st.success("✓ ArtToken loaded from environment")
+        else:
+            st.error("⚠ ArtToken not found in environment")
+        
+        st.markdown("---")
         st.header("Model Selection")
         st.session_state['selected_models'] = st.multiselect(
             "Choose AI Models",