Update app.py

app.py

@@ -7,9 +7,7 @@ import os
 
 # Get your secret key from Hugging Face Space Secrets
 NEBIUS_API_KEY = os.environ.get("NEBIUS_API_KEY")
-
-# --- UPDATE THESE VALUES ---
-NEBIUS_BASE_URL = "https://api.tokenfactory.nebius.com/v1/" # Use the URL from your Nebius dashboard" # Use the URL from your Nebius dashboard
+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
 
@@ -65,23 +63,22 @@ def call_image_gen_tool(prompt: str) -> str:
 
 # --- 3. DEFINE THE AUTONOMOUS AGENT LOGIC ---
 
-# *** UPDATED: Added 'business_type' as an input ***
 def run_brand_agent_plan(company_name, business_type, style, progress=gr.Progress()):
     
     # 1. PLANNING
     # The agent creates its 3-step plan autonomously.
     progress(0, desc="Agent is creating a plan...")
     
-    # *** UPDATED: Prompts now use the 'business_type' variable ***
+    # *** UPDATED: Plan now includes the color palette prompt ***
     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 high-quality, photorealistic 'brand asset' image capturing a {style} vibe for a {business_type}. This could be a product shot, an interior, or an abstract lifestyle image. No visible text or branding elements."},
+        {"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}."}
     ]
     
     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__}\n"
+    log_message += f"- Step 2: {plan[1]['action'].__name__} (Color Palette)\n"
     log_message += f"- Step 3: {plan[2]['action'].__name__}\n"
     
     results = {}
@@ -93,10 +90,10 @@ def run_brand_agent_plan(company_name, business_type, style, progress=gr.Progres
     results["logo_url"] = plan[0]["action"](plan[0]["prompt"])
     log_message += f"Step 1 Complete. Logo URL acquired.\n"
 
-    # Step 2: Generate Brand Asset
-    progress(0.5, desc="Executing Step 2: Calling Nebius Image Tool (Brand Asset)...")
-    results["asset_url"] = plan[1]["action"](plan[1]["prompt"])
-    log_message += f"Step 2 Complete. Brand Asset 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 3: Generate Social Copy
     progress(0.8, desc="Executing Step 3: Calling Nebius Text Tool (Copy)...")
@@ -109,10 +106,9 @@ def run_brand_agent_plan(company_name, business_type, style, progress=gr.Progres
     
     # 3. RESULT
     # The agent returns all 3 results to the UI.
-    return results["logo_url"], results["asset_url"], results["copy_text"], log_message
+    return results["logo_url"], results["palette_url"], results["copy_text"], log_message
 
-# --- 4. NEW CHATBOT FUNCTION ---
-# (This function is already general, no changes needed)
+# --- 4. CHATBOT FUNCTION ---
 def chat_with_brand_agent(message, history):
     system_prompt_chat = "You are a friendly and creative branding expert. A user is asking for ideas for their new brand. Give them suggestions, ask clarifying questions, and help them brainstorm."
     
@@ -146,7 +142,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         """
     )
     
-    # *** UPDATED: Added 'business_type_input' and changed default values ***
     with gr.Row():
         company_input = gr.Textbox(label="Company Name", value="Aura")
         business_type_input = gr.Textbox(label="Business Type / Industry", value="Skincare Brand")
@@ -159,7 +154,8 @@ 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)
-                asset_output = gr.Image(label="Generated Brand Asset (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)
         
         # --- TAB 2: AGENT LOG ---
@@ -169,17 +165,16 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         # --- TAB 3: CHATBOT ---
         with gr.TabItem("Brand Chatbot"):
             chatbot = gr.Chatbot(label="Brand Brainstormer")
-            # *** UPDATED: Changed placeholder text to be generic ***
             chat_msg = gr.Textbox(label="Ask for ideas...", placeholder="e.g., 'What are some good names for my new skincare brand?'")
             chat_btn = gr.Button("Send")
 
     # --- Wire up the UI ---
     
-    # *** UPDATED: Added 'business_type_input' to the inputs list ***
+    # *** 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],
-        outputs=[logo_output, asset_output, copy_output, log_output]
+        outputs=[logo_output, color_palette_output, copy_output, log_output]
     )
     
     # Click handlers for the new chatbot