Upload app.py with huggingface_hub

app.py

@@ -60,8 +60,6 @@ def get_parquet_path(config: str, shard: int = 0) -> str:
 def query_parquet(config: str, sql: str, params: dict = None) -> list:
     """Execute SQL query on parquet file."""
     path = get_parquet_path(config)
-    print(f"Querying parquet: {path}")
-    print(f"SQL: {sql}")
     try:
         conn = duckdb.connect(":memory:")
         conn.execute(f"CREATE VIEW data AS SELECT * FROM read_parquet('{path}')")
@@ -72,12 +70,31 @@ def query_parquet(config: str, sql: str, params: dict = None) -> list:
             result = conn.execute(sql).fetchdf()
 
         conn.close()
-        # Convert NaN to None for JSON serialization
-        result = result.where(pd.notnull(result), None)
-        return result.to_dict(orient="records")
+
+        # Convert to JSON-safe format
+        import json
+        import numpy as np
+
+        def clean_value(v):
+            if v is None:
+                return None
+            if isinstance(v, float) and (np.isnan(v) or np.isinf(v)):
+                return None
+            if isinstance(v, (np.integer, np.int64)):
+                return int(v)
+            if isinstance(v, (np.floating, np.float64)):
+                return float(v)
+            return v
+
+        records = []
+        for _, row in result.iterrows():
+            record = {k: clean_value(v) for k, v in row.items()}
+            records.append(record)
+
+        return records
     except Exception as e:
-        print(f"Query error: {e}")
-        raise HTTPException(status_code=500, detail=f"Query error: {str(e)}")
+        import traceback
+        raise HTTPException(status_code=500, detail=f"Query error: {str(e)}. Traceback: {traceback.format_exc()}")
 
 
 @app.get("/")