Create router.py

router.py

@@ -0,0 +1,41 @@
+from llm_clients import call_llama, call_gemini
+from search_tool import search_web
+from rag_engine import rag_response
+
+def is_reasoning(prompt):
+    keywords = ["solve", "explain deeply", "why", "logic", "proof"]
+    return any(k in prompt.lower() for k in keywords)
+
+def is_live_data(prompt):
+    keywords = ["today", "latest", "current", "news", "price", "weather"]
+    return any(k in prompt.lower() for k in keywords)
+
+def route_request(prompt, image=None):
+
+    # RAG first
+    rag = rag_response(prompt)
+    if rag:
+        return rag
+
+    # Image case
+    if image:
+        return call_gemini(prompt)
+
+    # Reasoning
+    if is_reasoning(prompt):
+        return call_gemini(prompt)
+
+    # Live data
+    if is_live_data(prompt):
+        search_data = search_web(prompt)
+        llama_answer = call_llama(prompt + "\n\nWeb Data:\n" + search_data)
+        gemini_answer = call_gemini(prompt + "\n\nWeb Data:\n" + search_data)
+
+        # Simple judge (can improve later)
+        if len(gemini_answer) > len(llama_answer):
+            return gemini_answer
+        return llama_answer
+
+    # Default
+    return call_llama(prompt)
+