stvnnnnnn commited on
Commit
0524a3d
verified
1 Parent(s): 260e798

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -3
app.py CHANGED
@@ -110,9 +110,17 @@ def nl2sql(q: NLQuery):
110
  if not text:
111
  raise ValueError("Consulta vac铆a.")
112
 
113
- # Traducci贸n ES->EN si detectamos caracteres no ASCII
114
- is_ascii = all(ord(c) < 128 for c in text)
115
- query_en = text if is_ascii else GoogleTranslator(source="auto", target="en").translate(text)
 
 
 
 
 
 
 
 
116
 
117
  df = get_table(SPLIT, INDEX, MAX_ROWS)
118
  enc = tok(table=df, query=query_en, return_tensors="pt", truncation=True)
 
110
  if not text:
111
  raise ValueError("Consulta vac铆a.")
112
 
113
+ lower = text.lower().strip()
114
+ looks_like_sql = lower.startswith(("select", "with", "insert", "update", "delete", "create", "drop", "alter"))
115
+
116
+ query_en = text
117
+ if not looks_like_sql:
118
+ try:
119
+ translated = GoogleTranslator(source="auto", target="en").translate(text)
120
+ if translated: # si viene None o vac铆o, mantenemos original
121
+ query_en = translated
122
+ except Exception:
123
+ query_en = text
124
 
125
  df = get_table(SPLIT, INDEX, MAX_ROWS)
126
  enc = tok(table=df, query=query_en, return_tensors="pt", truncation=True)