Update app.py

app.py

@@ -532,7 +532,39 @@ async def chat(message: str = Form(...)):
     
     lang_mode = detect_input_language(message)
     system_prompt = get_system_prompt(lang_mode, USERNAME)
-    
+    @app.post("/chat")
+async def chat(message: str = Form(...)):
+    # ✅ LANGUAGE DETECTION
+    from language_detector import detect_input_language, get_system_prompt, generate_basic_code
+    from memory_db import tidb_memory  # ✅ TiDB IMPORT
+
+    lang_mode = detect_input_language(message)
+    system_prompt = get_system_prompt(lang_mode, USERNAME)
+
+    # --- CODE DETECTION BLOCK ---
+    msg_lower = message.lower()
+    CODE_KEYWORDS = [
+        "code", "script", "program", "function",
+        "mount", "google drive", "colab",
+        "python", "build", "create", "generate"
+    ]
+
+    if any(k in msg_lower for k in CODE_KEYWORDS):
+        code_response = generate_basic_code(message)
+        try:
+            tidb_memory.save_chat(message, code_response, lang_mode)
+        except Exception as e:
+            print(f"⚠️ TiDB save error (code): {e}")
+        return {"response": code_response}
+
+    # ✅ GET HISTORY FROM TiDB (instead of JSON)
+    recent_chats = []
+    try:
+        recent_chats = tidb_memory.get_recent_chats(limit=10)
+    except Exception as e:
+        print(f"⚠️ TiDB history fetch error: {e}")
+        # Fallback to empty history if TiDB fails
+
     # ✅ GET HISTORY FROM TiDB (instead of JSON)
     recent_chats = []
     try: