Update app.py

app.py

@@ -51,7 +51,6 @@ def _heuristic_sql(question: str, table: str, cols: list = None):
     if "count" in q or "how many" in q:
         return f'SELECT COUNT(*) AS total_rows FROM "{table}"'
 
-    # Extract number from question if present, default to 10
     match = re.search(r'\b(\d+)\b', q)
     limit = match.group(1) if match else "10"
 
@@ -193,4 +192,126 @@ async def upload_csv(file: UploadFile = File(...)):
         if table_name and table_name[0].isdigit():
             table_name = "t_" + table_name
 
-        table_n
+        table_name = table_name[:40]
+
+        with tempfile.NamedTemporaryFile(delete=False) as tf:
+            conn = sqlite3.connect(tf.name)
+            df.to_sql(table_name, conn, index=False, if_exists="replace")
+
+            schema = conn.execute(
+                "SELECT sql FROM sqlite_master WHERE type='table'"
+            ).fetchone()[0]
+
+            conn.close()
+
+            db_bytes = open(tf.name, "rb").read()
+
+        os.remove(tf.name)
+
+        _db_store[session_id] = {
+            "bytes": db_bytes,
+            "table": table_name,
+            "cols":  list(df.columns)
+        }
+
+        _schema_store[session_id] = schema
+
+        return {
+            "session_id": session_id,
+            "row_count":  len(df),
+            "columns":    list(df.columns),
+            "table_name": table_name
+        }
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+# ─────────────────────────────
+# QUERY ENGINE
+# ─────────────────────────────
+
+@app.post("/query")
+async def query(req: QueryRequest):
+
+    data = _db_store.get(req.session_id)
+
+    if not data and _db_store:
+        data = list(_db_store.values())[-1]
+
+    if not data:
+        raise HTTPException(status_code=404, detail="No dataset loaded")
+
+    table = data["table"]
+
+    sql = _heuristic_sql(req.question, table, data["cols"])
+
+    if not sql:
+        schema = (
+            _schema_store.get(req.session_id)
+            or list(_schema_store.values())[-1]
+        )
+        sql = _call_gemini(req.question, schema, data["cols"], table)
+
+    if not sql or "error" in sql.lower():
+        sql = f'SELECT * FROM "{table}" LIMIT 10'
+
+    results = execute_sql(sql, data["bytes"])
+
+    return {
+        "sql":     sql,
+        "results": results
+    }
+
+
+# ─────────────────────────────
+# HEALTH
+# ─────────────────────────────
+
+@app.get("/health")
+def health():
+    return {
+        "status":  "ok",
+        "model":   "gemini-2.5-flash",
+        "service": "AI Data Analyst"
+    }
+
+
+# ─────────────────────────────
+# TELEGRAM WEBHOOK
+# ─────────────────────────────
+
+@app.on_event("startup")
+async def on_startup():
+    try:
+        from bot import setup_webhook
+        setup_webhook()
+    except Exception as e:
+        print(f"⚠️ Webhook setup skipped: {e}")
+
+
+@app.post("/webhook/{token}")
+async def telegram_webhook(token: str, request: Request):
+    expected = os.getenv("BOT_TOKEN", "")
+    if token != expected:
+        raise HTTPException(status_code=403, detail="Invalid token")
+    try:
+        import telebot
+        from bot import bot
+        json_data = await request.json()
+        update = telebot.types.Update.de_json(json_data)
+        bot.process_new_updates([update])
+    except Exception as e:
+        print(f"❌ Webhook error: {e}")
+    return {"ok": True}
+
+
+# ─────────────────────────────
+# FRONTEND
+# ─────────────────────────────
+
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
+@app.get("/")
+def root():
+    return FileResponse("static/webapp.html")