Update app.py

app.py

@@ -26,6 +26,7 @@ def bytes_to_pil(image_bytes: bytes):
 
 # --- 2. MCP TOOL FUNCTIONS ---
 def tool_web_search(query: str):
+    """Performs a market research search using DuckDuckGo."""
     print(f"🔍 Search Tool: {query}...")
     try:
         results = DDGS().text(query, max_results=4)
@@ -37,6 +38,7 @@ def tool_web_search(query: str):
         return f"Search Error: {str(e)}"
 
 def tool_generate_image(prompt: str):
+    """Generates an image and returns the local file path."""
     print(f"🎨 Image Tool: {prompt[:40]}...")
     try:
         response = nebius_client.images.generate(
@@ -50,6 +52,7 @@ def tool_generate_image(prompt: str):
         b64_data = response.data[0].b64_json
         image_bytes = base64.b64decode(b64_data)
         img = Image.open(io.BytesIO(image_bytes))
+
         filename = f"gen_{abs(hash(prompt))}.png"
         img.save(filename)
         return filename
@@ -100,14 +103,17 @@ def tool_summarize_market_insights(text: str):
 
 # --- 3. AUTONOMOUS BRAND AGENT ---
 def run_autonomous_agent(company_name, business_type, style, progress=gr.Progress()):
+    # PHASE 1: COLOR PALETTE
     progress(0.1, desc="Step 1: Planning & Palette...")
     color_prompt = (
         f"Create a distinct, 5-color palette for a {business_type} named '{company_name}' "
         f"with a {style} style. Return ONLY a comma-separated list of hex codes."
     )
     colors_text = tool_generate_text(color_prompt, "You are a design expert. Output only the requested list.").strip().strip('.')
-    log = f"✅ Brand Plan Started for {company_name}\n🎨 Palette: {colors_text}\n"
+    log = f"✅ Brand Plan Started for {company_name}\n"
+    log += f"🎨 Palette: {colors_text}\n"
 
+    # PHASE 2: LOGO
     progress(0.3, desc="Step 2: Generating Logo...")
     logo_prompt = (
         f"A professional minimalist logo for a {business_type} named '{company_name}'. "
@@ -116,6 +122,7 @@ def run_autonomous_agent(company_name, business_type, style, progress=gr.Progres
     logo_path = tool_generate_image(logo_prompt)
     log += "🖼️ Logo Generated.\n"
 
+    # PHASE 3: COLOR PALETTE ASSET
     progress(0.5, desc="Step 3: Color Palette Asset...")
     asset_prompt = (
         f"A professional color palette guide with 5 circular swatches arranged in a row. "
@@ -124,15 +131,18 @@ def run_autonomous_agent(company_name, business_type, style, progress=gr.Progres
     asset_path = tool_generate_image(asset_prompt)
     log += "🖼️ Palette Asset Generated.\n"
 
+    # PHASE 4: RAW MARKET RESEARCH
     progress(0.65, desc="Step 4: Raw Market Research...")
     search_query = f"marketing trends for {business_type} 2025"
     raw_research = tool_web_search(search_query)
     log += f"🔍 Market Data Retrieved: {search_query}\n"
 
+    # PHASE 4.5: SUMMARIZE MARKET INSIGHTS
     progress(0.8, desc="Step 4.5: Summarizing Insights...")
     summarized = tool_summarize_market_insights(raw_research)
     log += f"🧠 Market Insights Summary: {summarized[:150]}...\n"
 
+    # PHASE 5: SOCIAL MEDIA COPY
     progress(0.95, desc="Step 5: Writing Social Content...")
     copy_prompt = f"""
     Write 3 engaging Twitter/X posts for a new {business_type} called '{company_name}'.
@@ -141,7 +151,8 @@ def run_autonomous_agent(company_name, business_type, style, progress=gr.Progres
     {summarized}
     """
     social_copy = tool_generate_text(copy_prompt, "You are a professional social media strategist.", max_tokens=1500)
-    log += "✍️ Social Copy Generated.\n✅ Autonomy Cycle Complete."
+    log += "✍️ Social Copy Generated.\n"
+    log += "✅ Autonomy Cycle Complete."
 
     return logo_path, asset_path, social_copy, log, summarized
 
@@ -167,8 +178,7 @@ def chat_response(message, history):
         history.append({"role": "assistant", "content": f"Error: {e}"})
         return history, ""
 
-# --- 5. GRADIO UI (Compatible with v6) ---
-custom_css = """#main-header {text-align: center; margin-bottom: 2rem;}"""
+# --- 5. GRADIO UI ---
 theme = gr.themes.Soft(
     primary_hue="indigo",
     secondary_hue="blue",
@@ -181,13 +191,17 @@ theme = gr.themes.Soft(
     block_shadow="0 4px 6px rgba(0,0,0,0.1)",
 )
 
-demo = gr.Blocks(css=custom_css, title="AutoBrand MCP Studio")
-demo.theme = theme  # set theme AFTER creating Blocks
+custom_css = "#main-header {text-align: center; margin-bottom: 2rem;}"
+
+# Create Blocks (Gradio 6 compatible)
+demo = gr.Blocks()
+demo.theme = theme
+demo.css = custom_css
 
 with demo:
     # HEADER
     with gr.Column(elem_id="main-header"):
-        gr.Markdown("# 🖼️ AutoBrand Studio")
+        gr.Markdown("# 🖼️ AutoBrand MCP Studio")
         gr.Markdown("Autonomous Agent powered by AI Models, MCP Tools & Web Search. Connect Claude Desktop to use these tools!")
 
     with gr.Row():
@@ -204,17 +218,20 @@ with demo:
         # RIGHT PANEL
         with gr.Column(scale=2):
             with gr.Tabs():
+                # BRAND IDENTITY TAB
                 with gr.TabItem("🎨 Brand Identity"):
                     with gr.Row():
                         out_logo = gr.Image(label="Logo Concept", height=350, type="filepath")
                         out_asset = gr.Image(label="Color Palette", height=350, type="filepath")
                     out_copy = gr.TextArea(label="Generated Social Content", lines=15, show_copy_button=True, interactive=False)
 
+                # BRAND CONSULTANT CHATBOT TAB
                 with gr.TabItem("💬 Brand Consultant"):
                     chatbot = gr.Chatbot(label="AI Consultant", height=650, type="messages")
                     chat_input = gr.Textbox(show_label=False, placeholder="Ask something...")
                     chat_btn = gr.Button("Send")
 
+                # MARKET INSIGHT SUMMARY TAB
                 with gr.TabItem("📊 Market Insight Summary"):
                     summary_box = gr.TextArea(label="Summarized Market Insights", lines=20, interactive=False, show_copy_button=True)
 
@@ -226,7 +243,7 @@ with demo:
     generate_btn.click(
         run_agent_and_extract_summary,
         [company_input, type_input, style_input],
-        [out_logo, out_asset, out_copy, out_log, summary_box]
+        [out_logo, out_asset, out_copy, out_log, summary_box],
     )
 
     chat_btn.click(chat_response, [chat_input, chatbot], [chatbot, chat_input])