Update src/chimera_core.py

src/chimera_core.py

@@ -2,7 +2,7 @@ import os
 import random
 from google import genai
 from groq import Groq
-from ddgs import DDGS  # Changed from duckduckgo_search
+from ddgs import DDGS
 from huggingface_hub import InferenceClient
 from PIL import Image
 
@@ -40,13 +40,15 @@ class Chimera:
     # --- MODULE 2: IGM (Image Generation) ---
     def _generate_image(self, prompt):
         print(f"   ↳ 🎨 Generating image for: {prompt}")
-        if not self.hf_client: return None, "❌ HF_TOKEN missing."
+        if not self.hf_client: 
+            return None, "❌ HF_TOKEN missing."
         try:
             image = self.hf_client.text_to_image(prompt)
             # Save to a temporary path so Gradio can display it
             path = f"/tmp/gen_{random.randint(0,9999)}.png"
             image.save(path)
-            return path, f"Generated image saved!\n\n![Generated Image](file={path})"
+            # Return the path - Gradio will handle displaying it
+            return path, None
         except Exception as e:
             return None, f"Generation Error: {e}"
 
@@ -66,17 +68,17 @@ class Chimera:
     def process_request(self, message, history, manual_role, image_input=None):
         # 1. PRIORITY: VIM (If image is uploaded)
         if image_input:
-            return self._analyze_image(message or "Describe this.", image_input), "VIM"
+            return self._analyze_image(message or "Describe this image in detail.", image_input), "VIM"
 
         # 2. DETECT INTENT (Router)
         role = manual_role
         if role == "Auto":
             msg_lower = message.lower()
-            if any(x in msg_lower for x in ["generate", "draw", "create image", "paint", "make an image"]):
+            if any(x in msg_lower for x in ["generate", "draw", "create image", "paint", "make an image", "picture of"]):
                 role = "IGM"
-            elif any(x in msg_lower for x in ["search", "news", "price", "latest", "find"]):
+            elif any(x in msg_lower for x in ["search", "news", "price", "latest", "find", "look up"]):
                 role = "NET"
-            elif any(x in msg_lower for x in ["code", "python", "script", "function", "debug"]):
+            elif any(x in msg_lower for x in ["code", "python", "script", "function", "debug", "program"]):
                 role = "ASM"
             else:
                 role = "CHAT"
@@ -95,19 +97,22 @@ class Chimera:
         # 3. EXECUTE MODULES
         if role == "IGM" or role == "IGM (Generate Image)":
             # Image Gen Mode
-            path, markdown = self._generate_image(message)
-            return markdown, "IGM"
+            image_path, error_msg = self._generate_image(message)
+            if error_msg:
+                return error_msg, "IGM"
+            # Return both text and image path
+            return ("✨ Image generated successfully!", image_path), "IGM"
 
         elif role == "NET" or role == "NET (Search)":
             # Web Search Mode
             search_data = self._web_search(message)
             # Synthesize answer with Gemini
-            prompt = f"User Question: {message}\n\nSearch Results:\n{search_data}\n\nAnswer the user based on these results."
+            prompt = f"User Question: {message}\n\nSearch Results:\n{search_data}\n\nProvide a clear, concise answer based on these results."
             response = self.gemini_client.models.generate_content(model="gemini-2.5-flash", contents=prompt)
             return response.text, "NET"
 
         elif role == "ASM" or role == "ASM (Code)":
-            # Coding Mode (Qwen/Llama) - Simplified for stability
+            # Coding Mode (Qwen/Llama)
             if self.groq_client:
                 try:
                     res = self.groq_client.chat.completions.create(