Update app.py

app.py

@@ -3,6 +3,8 @@ from openai import OpenAI
 from ddgs import DDGS
 import os
 import base64
+from PIL import Image
+import io
 
 # --- 1. CONFIGURATION ---
 
@@ -19,10 +21,15 @@ nebius_client = OpenAI(
     base_url=NEBIUS_BASE_URL,
 )
 
+# Convert image bytes → PIL Image
+def bytes_to_pil(image_bytes: bytes):
+    return Image.open(io.BytesIO(image_bytes))
+
+
 # --- 2. DEFINE LOCAL TOOLS ---
 
 def tool_generate_text(prompt: str, system_message: str = "You are a helpful assistant.") -> str:
-    """Tool: Generates text using the LLM."""
+    """Text generation tool."""
     print(f"📝 Text Tool calling API: {prompt[:40]}...")
     try:
         response = nebius_client.chat.completions.create(
@@ -41,7 +48,7 @@ def tool_generate_text(prompt: str, system_message: str = "You are a helpful ass
 
 
 def tool_generate_image(prompt: str):
-    """Tool: Generates an image and returns raw bytes instead of URL."""
+    """Image tool that returns PIL Image (not bytes)."""
     print(f"🎨 Image Tool calling API: {prompt[:40]}...")
     try:
         response = nebius_client.images.generate(
@@ -49,12 +56,14 @@ def tool_generate_image(prompt: str):
             prompt=prompt,
             n=1,
             size="1024x1024",
-            response_format="b64_json",   # prevents 404
+            response_format="b64_json",  
         )
 
         b64_data = response.data[0].b64_json
         image_bytes = base64.b64decode(b64_data)
-        return image_bytes
+
+        # FIX: return PIL Image so Gradio can render it
+        return bytes_to_pil(image_bytes)
 
     except Exception as e:
         print(f"❌ Image Tool Error: {e}")
@@ -62,7 +71,7 @@ def tool_generate_image(prompt: str):
 
 
 def tool_web_search(query: str) -> str:
-    """Tool: Searches the web using DuckDuckGo."""
+    """Web search using DuckDuckGo."""
     print(f"🔍 Search Tool searching for: {query}...")
     try:
         results = DDGS().text(query, max_results=4)
@@ -86,14 +95,14 @@ def tool_web_search(query: str) -> str:
 # --- 3. AGENT LOGIC ---
 
 def run_autonomous_agent(company_name, business_type, style, progress=gr.Progress()):
-    """The autonomous branding agent."""
+    """Autonomous branding agent."""
     
     # PHASE 1: PLANNING
     progress(0.1, desc="Step 1: Planning & Color 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 color names/hex codes."
+        f"with a {style} style. Return ONLY a comma-separated list of hex codes."
     )
 
     colors_text = tool_generate_text(
@@ -109,24 +118,24 @@ def run_autonomous_agent(company_name, business_type, style, progress=gr.Progres
 
     logo_prompt = (
         f"A professional, minimalist logo for a {business_type} named '{company_name}'. "
-        f"The design must use ONLY these colors: {colors_text}. Vector flat style, centered, high quality, no text."
+        f"Use ONLY these colors: {colors_text}. Vector flat style, centered, no text."
     )
 
     logo_img = tool_generate_image(logo_prompt)
     log += "🖼️ Logo Generated.\n"
 
-    # PHASE 3: BRAND ASSET
-    progress(0.5, desc="Step 3: Generating Brand Asset...")
+    # PHASE 3: COLOR PALETTE ASSET
+    progress(0.5, desc="Step 3: Generating Color Palette Guide...")
 
     asset_prompt = (
-        f"A professional brand color palette guide. Five clean, circular color swatches arranged in a row, "
-        f"showing exactly these colors: {colors_text}. Minimalist white background."
+        f"A professional color palette guide with 5 circular swatches, arranged in a row. "
+        f"Use these colors: {colors_text}. Minimalist white background."
     )
 
     asset_img = tool_generate_image(asset_prompt)
     log += "🖼️ Brand Asset Generated.\n"
 
-    # PHASE 4: SEARCH
+    # PHASE 4: WEB SEARCH
     progress(0.7, desc="Step 4: Researching Market Trends...")
 
     search_query = f"marketing trends for {business_type} 2025"
@@ -140,8 +149,7 @@ def run_autonomous_agent(company_name, business_type, style, progress=gr.Progres
 
     copy_prompt = f"""
     Write 3 engaging Twitter/X posts for a new {business_type} called '{company_name}'.
-    Tone: {style}
-
+    Tone: {style}.
     Use these market insights if helpful:
     {search_data}
     """
@@ -171,12 +179,14 @@ def chat_response(message, history):
             model=TEXT_MODEL,
             max_tokens=300,
         )
+
         bot_reply = response.choices[0].message.content
 
         history.append({"role": "user", "content": message})
         history.append({"role": "assistant", "content": bot_reply})
 
         return history, ""
+
     except Exception as e:
         history.append({"role": "assistant", "content": f"Error: {e}"})
         return history, ""