Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -110,9 +110,17 @@ def nl2sql(q: NLQuery):
|
|
| 110 |
if not text:
|
| 111 |
raise ValueError("Consulta vac铆a.")
|
| 112 |
|
| 113 |
-
|
| 114 |
-
|
| 115 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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)
|