AumCoreAI commited on
Commit
66acb3d
·
verified ·
1 Parent(s): baa633d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -350,8 +350,12 @@ async def chat(message: str = Form(...)):
350
  system_prompt = get_system_prompt(lang_mode, USERNAME)
351
 
352
  msg_lower = message.lower()
353
- CODE_KEYWORDS = ["code","script","program","function","mount","google drive","colab","python","build","create","generate"]
354
- if any(k in msg_lower for k in CODE_KEYWORDS):
 
 
 
 
355
  code_response = generate_basic_code(message)
356
  try: tidb_memory.save_chat(message, code_response, lang_mode)
357
  except Exception as e: print(f"⚠️ TiDB save error (code): {e}")
 
350
  system_prompt = get_system_prompt(lang_mode, USERNAME)
351
 
352
  msg_lower = message.lower()
353
+
354
+ # 🚀 STRICT CODING ROUTER - AI coding expert mode
355
+ # Only specific coding requests go to generate_basic_code()
356
+ # LLM ko free-style coding nahi karne dena
357
+ STRICT_CODE_KEYWORDS = ["python code","write code","generate code","create script","program for","function for","mount google drive","colab notebook","script for","coding task"]
358
+ if any(k in msg_lower for k in STRICT_CODE_KEYWORDS):
359
  code_response = generate_basic_code(message)
360
  try: tidb_memory.save_chat(message, code_response, lang_mode)
361
  except Exception as e: print(f"⚠️ TiDB save error (code): {e}")