Update app.py

app.py

@@ -54,7 +54,6 @@ def tool_generate_image(prompt: str):
         print(f"❌ Image Tool Error: {e}")
         return None
 
-# --- Text Generation Helper ---
 def tool_generate_text(prompt: str, system_message: str = "You are a helpful assistant.", max_tokens: int = 1024) -> str:
     try:
         response = nebius_client.chat.completions.create(
@@ -70,7 +69,6 @@ def tool_generate_text(prompt: str, system_message: str = "You are a helpful ass
     except Exception as e:
         return f"Error: {e}"
 
-# --- Market Insights Summarization Tool ---
 def tool_summarize_market_insights(text: str):
     print("🧠 Summarization Tool Running...")
     try:
@@ -162,10 +160,27 @@ def chat_response(message, history):
         history.append({"role": "assistant", "content": f"Error: {e}"})
         return history, ""
 
-# --- 5. GRADIO UI ---
-with gr.Blocks() as demo:
+# --- 5. GRADIO THEME & CSS ---
+theme = gr.themes.Soft(
+    primary_hue="indigo",
+    secondary_hue="blue",
+    neutral_hue="slate",
+).set(
+    button_primary_background_fill="linear-gradient(90deg, #6366f1 0%, #4338ca 100%)",
+    button_primary_background_fill_hover="linear-gradient(90deg, #4f46e5 0%, #3730a3 100%)",
+    button_primary_text_color="white",
+    block_title_text_weight="600",
+    block_shadow="0 4px 6px rgba(0,0,0,0.1)",
+)
+
+custom_css = """
+#main-header {text-align: center; margin-bottom: 2rem;}
+"""
+
+# --- 6. GRADIO UI ---
+with gr.Blocks(theme=theme, css=custom_css, title="AutoBrand MCP Studio") as demo:
     # HEADER
-    with gr.Column():
+    with gr.Column(elem_id="main-header"):
         gr.Markdown("# 🖼️ AutoBrand Studio")
         gr.Markdown("Autonomous Agent powered by AI Models, MCP Tools & Web Search. Connect Claude Desktop to use these tools!")