Update app.py

app.py

@@ -9,7 +9,7 @@ from dotenv import load_dotenv
 load_dotenv()
 
 # API Configuration
-API_KEY = os.getenv('STABLECOG_API_KEY', '')
+API_KEY = os.getenv('StableCogKey', '')
 if not API_KEY:
     API_KEY = "StableCogKey"
 
@@ -441,14 +441,9 @@ def get_recommendation(remaining_credits, percentage_used):
     else:
         return "🚀 **Plenty of credits!** Ready for extensive image generation work and experimentation."
 
-# Create Gradio interface with Gradio 6 theme
-# Note: Gradio 6 has a simplified theme system
+# Create Gradio interface with simplified theme configuration for Gradio 6
+# In Gradio 6, theme is set differently or uses default
 with gr.Blocks(
-    theme=gr.themes.Soft(
-        primary_hue="purple",
-        secondary_hue="indigo",
-        font=[gr.themes.GoogleFont("Inter"), "ui-sans-serif", "system-ui", "sans-serif"]
-    ),
     title="StableCog Dashboard",
     css="""
     footer {display: none !important;}
@@ -456,6 +451,17 @@ with gr.Blocks(
     .tab-nav {background: rgba(255,255,255,0.1) !important; border-radius: 10px !important; padding: 5px !important;}
     .stat-box input {font-weight: bold !important; font-size: 18px !important;}
     .recommendation-box textarea {font-size: 16px !important; line-height: 1.5 !important;}
+    body {
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        min-height: 100vh;
+        padding: 20px;
+    }
+    .gradio-container {
+        background: rgba(255, 255, 255, 0.95);
+        border-radius: 20px;
+        padding: 20px;
+        box-shadow: 0 20px 60px rgba(0, 0, 0, 0.3);
+    }
     """
 ) as demo:
     gr.Markdown("""
@@ -483,14 +489,12 @@ with gr.Blocks(
                 with gr.Column():
                     credits_display = gr.Number(
                         label="Remaining Credits", 
-                        interactive=False,
-                        elem_classes="stat-box"
+                        interactive=False
                     )
                 with gr.Column():
                     usage_display = gr.Number(
                         label="Usage Percentage", 
-                        interactive=False,
-                        elem_classes="stat-box"
+                        interactive=False
                     )
             
             # Recommendation row
@@ -498,16 +502,14 @@ with gr.Blocks(
                 recommendation_box = gr.Textbox(
                     label="🎯 AI Recommendation", 
                     interactive=False,
-                    lines=3,
-                    elem_classes="recommendation-box"
+                    lines=3
                 )
             
             # Control row
             with gr.Row():
                 check_credits_btn = gr.Button(
                     "🔄 Check Credits", 
-                    variant="primary",
-                    scale=0
+                    variant="primary"
                 )
         
         with gr.Tab("🤖 Available Models", id="models"):
@@ -525,8 +527,7 @@ with gr.Blocks(
             with gr.Row():
                 check_models_btn = gr.Button(
                     "🔄 Refresh Models", 
-                    variant="primary",
-                    scale=0
+                    variant="primary"
                 )
     
     # Instructions and info
@@ -570,10 +571,6 @@ with gr.Blocks(
         outputs=[models_html_output, models_raw_output]
     )
     
-    # Update recommendation after credits load
-    def update_recommendation_after_load():
-        return get_recommendation(credits_display.value, usage_display.value)
-    
     # Connect buttons
     def update_all_credits():
         html, raw, credits, usage = update_display()