Update app.py

app.py

@@ -110,23 +110,25 @@ def nl2sql(q: NLQuery):
         if not text:
             raise ValueError("Consulta vacía.")
 
+        # Detectar si parece SQL
         lower = text.lower().strip()
         looks_like_sql = lower.startswith(("select", "with", "insert", "update", "delete", "create", "drop", "alter"))
 
+        # Traducir a inglés si no es SQL
         query_en = text
         if not looks_like_sql:
             try:
                 translated = GoogleTranslator(source="auto", target="en").translate(text)
-                if translated:  # si viene None o vacío, mantenemos original
+                if translated:
                     query_en = translated
-        except Exception:
-            query_en = text
+            except Exception:
+                query_en = text  # fallback seguro
 
-        df  = get_table(SPLIT, INDEX, MAX_ROWS)
+        # Procesar con TAPEX
+        df = get_table_from_wikisql(SPLIT, INDEX, MAX_ROWS)
         enc = tok(table=df, query=query_en, return_tensors="pt", truncation=True)
         if torch.cuda.is_available():
             enc = {k: v.to("cuda") for k, v in enc.items()}
-
         out = model.generate(**enc, max_length=160, num_beams=1)
         sql = tok.batch_decode(out, skip_special_tokens=True)[0]
 
@@ -135,5 +137,6 @@ def nl2sql(q: NLQuery):
             "consulta_traducida": query_en,
             "sql_generado": sql
         }
+
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))