Update app.py

app.py

@@ -12,7 +12,7 @@ NEBIUS_API_KEY = os.environ.get("NEBIUS_API_KEY")
 # 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
+IMAGE_MODEL = "black-forest-labs/flux-dev"             # Use an image model from your dashboard
 
 # Check if the API key is set
 if not NEBIUS_API_KEY:
@@ -119,7 +119,7 @@ def run_brand_agent_plan(company_name, business_type, style, progress=gr.Progres
     # 3. RESULT
     return results["logo_url"], results["palette_url"], results["copy_text"], log_message
 
-# --- 4. CHATBOT FUNCTION (*** THIS IS THE FIX ***) ---
+# --- 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."
     
@@ -138,10 +138,8 @@ def chat_with_brand_agent(message, history):
         )
         bot_message = response.choices[0].message.content
         
-        # *** FIX ***: Append the new (user, bot) pair to the history
         history.append((message, bot_message))
         
-        # *** FIX ***: Return the updated history and clear the textbox
         return history, ""
     
     except Exception as e: