Upload app.py with huggingface_hub

app.py

@@ -3,6 +3,8 @@ from contextlib import asynccontextmanager
 from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from typing import Any
+import tempfile
+from pathlib import Path
 
 from models import (
     GenerateRequest, GenerateResponse,
@@ -25,7 +27,8 @@ data_designer = None
 async def lifespan(app: FastAPI):
     global data_designer
     from data_designer.interface import DataDesigner
-    data_designer = DataDesigner()
+    # Use temp directory for artifacts
+    data_designer = DataDesigner(artifact_path=tempfile.gettempdir())
     yield
 
 
@@ -161,17 +164,20 @@ async def health():
 @app.post("/generate", response_model=GenerateResponse)
 async def generate(request: GenerateRequest):
     """
-    Generate synthetic data.
+    Generate synthetic data using DataDesigner.create().
     """
     try:
         config_builder = build_config(request)
 
-        result = data_designer.generate(
+        # Use create() method - returns DatasetCreationResults
+        result = data_designer.create(
             config_builder=config_builder,
             num_records=request.num_records,
+            dataset_name="api-dataset"
         )
 
-        df = result.to_pandas()
+        # Get DataFrame from results
+        df = result.load_dataset()
         data = df.to_dict(orient="records")
 
         return GenerateResponse(
@@ -181,31 +187,40 @@ async def generate(request: GenerateRequest):
         )
 
     except Exception as e:
+        import traceback
         return GenerateResponse(
             success=False,
-            error=str(e)
+            error=f"{str(e)}\n{traceback.format_exc()}"
         )
 
 
 @app.post("/preview", response_model=PreviewResponse)
 async def preview(request: PreviewRequest):
     """
-    Preview a single record without full generation.
+    Preview a single record using DataDesigner.preview().
     """
     try:
         config_builder = build_config(request)
 
-        preview_result = data_designer.preview(config_builder=config_builder)
+        # Use preview() method - returns PreviewResults
+        preview_result = data_designer.preview(
+            config_builder=config_builder,
+            num_records=1
+        )
+
+        # Get sample record
+        sample = preview_result.dataset.to_dict(orient="records")[0] if len(preview_result.dataset) > 0 else {}
 
         return PreviewResponse(
             success=True,
-            sample=preview_result.sample_record
+            sample=sample
         )
 
     except Exception as e:
+        import traceback
         return PreviewResponse(
             success=False,
-            error=str(e)
+            error=f"{str(e)}\n{traceback.format_exc()}"
         )