Update app.py

app.py

@@ -7,6 +7,9 @@ import os
 
 # Get your secret key from Hugging Face Space Secrets
 NEBIUS_API_KEY = os.environ.get("NEBIUS_API_KEY")
+
+# --- UPDATE THESE VALUES ---
+# Make sure to include "https://" and the "/v1/" path
 NEBIUS_BASE_URL = "https://api.tokenfactory.nebius.com/v1/" # Use the URL from your Nebius dashboard
 TEXT_MODEL = "meta-llama/Llama-3.3-70B-Instruct"           # Use a text model from your dashboard
 IMAGE_MODEL = "black-forest-labs/flux-schnell"             # Use an image model from your dashboard
@@ -66,40 +69,51 @@ def call_image_gen_tool(prompt: str) -> str:
 def run_brand_agent_plan(company_name, business_type, style, progress=gr.Progress()):
     
     # 1. PLANNING
-    # The agent creates its 3-step plan autonomously.
+    # The agent creates its 4-step plan autonomously.
     progress(0, desc="Agent is creating a plan...")
     
-    # *** UPDATED: Plan now includes the color palette prompt ***
+    # *** NEW 4-STEP PLAN ***
     plan = [
-        {"step": 1, "action": call_image_gen_tool, "prompt": f"A {style} logo for a {business_type} named '{company_name}'. Clean, high-resolution, white background."},
-        {"step": 2, "action": call_image_gen_tool, "prompt": f"A refined {style} color palette presentation for a {company_name} {business_type}. Show 5 distinct, harmonious colors in a modern, clean layout. Photorealistic, minimalist, with subtle textures. No text."},
-        {"step": 3, "action": call_text_gen_tool, "prompt": f"Write 3 short, catchy social media posts (for X/Twitter) announcing a new {business_type} called '{company_name}'. The vibe is {style}."}
+        {"step": 1, "action": "generate_color_text", "prompt": f"Generate a 5-color palette for a {business_type} with a {style} vibe. List the colors as a simple, comma-separated string, e.g., 'Deep espresso brown, warm cream, rustic bronze, slate grey, soft olive green'."},
+        {"step": 2, "action": "generate_logo_from_colors"}, # Will be defined in execution
+        {"step": 3, "action": "generate_palette_from_colors"}, # Will be defined in execution
+        {"step": 4, "action": "generate_social_copy", "prompt": f"Write 3 short, catchy social media posts (for X/Twitter) announcing a new {business_type} called '{company_name}'. The vibe is {style}."}
     ]
     
     log_message = "Agent Plan Created:\n"
-    log_message += f"- Step 1: {plan[0]['action'].__name__}\n"
-    log_message += f"- Step 2: {plan[1]['action'].__name__} (Color Palette)\n"
-    log_message += f"- Step 3: {plan[2]['action'].__name__}\n"
+    log_message += f"- Step 1: Generate color palette (as text)\n"
+    log_message += f"- Step 2: Generate logo from colors\n"
+    log_message += f"- Step 3: Generate palette image from colors\n"
+    log_message += f"- Step 4: Generate social media copy\n"
     
     results = {}
     
     # 2. EXECUTION
     
-    # Step 1: Generate Logo
-    progress(0.2, desc="Executing Step 1: Calling Nebius Image Tool (Logo)...")
-    results["logo_url"] = plan[0]["action"](plan[0]["prompt"])
-    log_message += f"Step 1 Complete. Logo URL acquired.\n"
-
-    # Step 2: Generate Color Palette
-    progress(0.5, desc="Executing Step 2: Calling Nebius Image Tool (Color Palette)...")
-    results["palette_url"] = plan[1]["action"](plan[1]["prompt"])
-    log_message += f"Step 2 Complete. Color Palette URL acquired.\n"
+    # Step 1: Generate Color Palette (as Text)
+    progress(0.2, desc="Executing Step 1: Calling Nebius Text Tool (for Colors)...")
+    system_prompt_colors = "You are a concise branding expert. Only output a single, comma-separated list of 5 colors. Do not add any other text, labels, or explanations."
+    color_palette_text = call_text_gen_tool(plan[0]["prompt"], system_prompt_colors)
+    results["color_text"] = color_palette_text.strip() # Store the colors
+    log_message += f"Step 1 Complete. Colors decided: {results['color_text']}\n"
+    
+    # Step 2: Generate Logo (using colors from Step 1)
+    progress(0.4, desc="Executing Step 2: Calling Nebius Image Tool (Logo)...")
+    logo_prompt = f"A {style} logo for a {business_type} named '{company_name}'. Use *only* the following colors: {results['color_text']}. The logo should be an abstract icon or geometric mark. NO TEXT."
+    results["logo_url"] = call_image_gen_tool(logo_prompt)
+    log_message += f"Step 2 Complete. Logo URL acquired.\n"
+
+    # Step 3: Generate Color Palette Image (using colors from Step 1)
+    progress(0.7, desc="Executing Step 3: Calling Nebius Image Tool (Color Palette)...")
+    palette_prompt = f"A refined color palette presentation. Show 5 distinct, harmonious colors in a modern, clean layout. The colors are: {results['color_text']}. No text."
+    results["palette_url"] = call_image_gen_tool(palette_prompt)
+    log_message += f"Step 3 Complete. Color Palette URL acquired.\n"
     
-    # Step 3: Generate Social Copy
-    progress(0.8, desc="Executing Step 3: Calling Nebius Text Tool (Copy)...")
+    # Step 4: Generate Social Copy
+    progress(0.9, desc="Executing Step 4: Calling Nebius Text Tool (Copy)...")
     system_prompt_kit = "You are a helpful branding assistant. Be creative and concise."
-    results["copy_text"] = plan[2]["action"](plan[2]["prompt"], system_prompt_kit)
-    log_message += "Step 3 Complete. Copy generated.\n"
+    results["copy_text"] = call_text_gen_tool(plan[3]["prompt"], system_prompt_kit)
+    log_message += "Step 4 Complete. Copy generated.\n"
     
     progress(1.0, desc="Agent Execution Complete!")
     log_message += "Plan executed successfully."
@@ -154,7 +168,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         with gr.TabItem("Brand Kit Results"):
             with gr.Row():
                 logo_output = gr.Image(label="Generated Logo (from Nebius)", height=400)
-                # *** UPDATED: Renamed component to 'color_palette_output' ***
                 color_palette_output = gr.Image(label="Generated Color Palette (from Nebius)", height=400)
             copy_output = gr.Textbox(label="Generated Social Media Copy (from Nebius)", lines=8, interactive=True)
         
@@ -170,7 +183,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     # --- Wire up the UI ---
     
-    # *** UPDATED: Changed one of the outputs to 'color_palette_output' ***
     submit_btn.click(
         fn=run_brand_agent_plan,
         inputs=[company_input, business_type_input, style_input],