Upload app.py with huggingface_hub

app.py

@@ -119,6 +119,29 @@ def test_download():
         return {"status": "error", "error": str(e), "type": type(e).__name__}
 
 
+@app.get("/test-query")
+def test_query():
+    """Test querying a parquet file."""
+    try:
+        path = get_parquet_path("courts")
+        conn = duckdb.connect(":memory:")
+        conn.execute(f"CREATE VIEW data AS SELECT * FROM read_parquet('{path}')")
+        result = conn.execute("SELECT COUNT(*) as cnt FROM data").fetchdf()
+        count = int(result['cnt'].iloc[0])
+
+        # Get one row
+        row = conn.execute("SELECT * FROM data LIMIT 1").fetchdf()
+        conn.close()
+
+        # Convert to dict
+        row_dict = row.to_dict(orient="records")[0] if len(row) > 0 else {}
+
+        return {"status": "ok", "count": count, "sample_row_keys": list(row_dict.keys())}
+    except Exception as e:
+        import traceback
+        return {"status": "error", "error": str(e), "type": type(e).__name__, "traceback": traceback.format_exc()}
+
+
 @app.get("/rows/{config}")
 def get_rows(
     config: str,