Update src/chimera_core.py

src/chimera_core.py

@@ -45,7 +45,7 @@ class Chimera:
             # 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](/file={path})"
+            return path, f"Generated image saved!\n\n![Generated Image](file={path})"
         except Exception as e:
             return None, f"Generation Error: {e}"
 
@@ -65,54 +65,63 @@ 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 (Vision)"
+            return self._analyze_image(message or "Describe this.", 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"]):
+            if any(x in msg_lower for x in ["generate", "draw", "create image", "paint", "make an image"]):
                 role = "IGM"
-            elif any(x in msg_lower for x in ["search", "news", "price", "latest"]):
+            elif any(x in msg_lower for x in ["search", "news", "price", "latest", "find"]):
                 role = "NET"
-            elif any(x in msg_lower for x in ["code", "python", "script"]):
+            elif any(x in msg_lower for x in ["code", "python", "script", "function", "debug"]):
                 role = "ASM"
             else:
                 role = "CHAT"
+        else:
+            # Map dropdown values to internal codes
+            role_map = {
+                "ASM (Code)": "ASM",
+                "IGM (Generate Image)": "IGM",
+                "NET (Search)": "NET",
+                "VIM (Vision)": "VIM"
+            }
+            role = role_map.get(manual_role, role)
 
         print(f"👉 Routing to: [{role}]")
 
         # 3. EXECUTE MODULES
-        if role == "IGM":
+        if role == "IGM" or role == "IGM (Generate Image)":
             # Image Gen Mode
             path, markdown = self._generate_image(message)
-            return markdown, "IGM (Flux)"
+            return markdown, "IGM"
 
-        elif role == "NET":
+        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."
             response = self.gemini_client.models.generate_content(model="gemini-2.5-flash", contents=prompt)
-            return response.text, "NET (Web)"
+            return response.text, "NET"
 
-        elif role == "ASM":
+        elif role == "ASM" or role == "ASM (Code)":
             # Coding Mode (Qwen/Llama) - Simplified for stability
             if self.groq_client:
                 try:
                     res = self.groq_client.chat.completions.create(
-                        model="qwen-2.5-32b", # Or qwen/qwen3-32b
+                        model="qwen2.5-coder-32b-instruct",
                         messages=[{"role":"user", "content": message}]
                     )
-                    return res.choices[0].message.content, "ASM (Qwen)"
-                except:
-                     pass # Fallback to Gemini
+                    return res.choices[0].message.content, "ASM"
+                except Exception as e:
+                    print(f"Groq Error: {e}, falling back to Gemini")
             
         # Default Fallback (Gemini)
         try:
             res = self.gemini_client.models.generate_content(
                 model="gemini-2.5-flash", contents=message
             )
-            return res.text, f"{role} (Gemini)"
-        except:
-            return "System Error.", "ERR"
+            return res.text, f"{role}"
+        except Exception as e:
+            return f"System Error: {e}", "ERR"