Update main.py

main.py

@@ -164,21 +164,38 @@ class VisualizationRequest(BaseModel):
     title: Optional[str] = None
     x_label: Optional[str] = None
     y_label: Optional[str] = None
-    style: str = "seaborn"
+    style: str = "seaborn-v0_8"  # Updated default
     filters: Optional[dict] = None
 
 class NaturalLanguageRequest(BaseModel):
     prompt: str
-    style: str = "seaborn"
+    style: str = "seaborn-v0_8"
 
 def validate_matplotlib_style(style: str) -> str:
     """Validate and return a valid matplotlib style"""
     available_styles = plt.style.available
+    # Map legacy style names to current ones
+    style_mapping = {
+        'seaborn': 'seaborn-v0_8',
+        'seaborn-white': 'seaborn-v0_8-white',
+        'seaborn-dark': 'seaborn-v0_8-dark',
+        # Add other legacy mappings if needed
+    }
+    
+    # Check if it's a legacy name we can map
+    if style in style_mapping:
+        return style_mapping[style]
+    
+    # Check if it's a valid current style
     if style in available_styles:
         return style
+    
     logger.warning(f"Invalid style '{style}'. Available styles: {available_styles}")
-    return "seaborn"  # Default fallback
+    return "seaborn-v0_8"  # Default fallback to current seaborn style
 
+
+
+    
 def generate_visualization_code(df: pd.DataFrame, request: VisualizationRequest) -> str:
     """Generate Python code for visualization based on request parameters"""
     # Validate style
@@ -307,7 +324,7 @@ def interpret_natural_language(prompt: str, df_columns: list) -> VisualizationRe
         y_column=y_col,
         hue_column=hue_col,
         title="Generated from: " + prompt[:50] + ("..." if len(prompt) > 50 else ""),
-        style="seaborn"
+        style="seaborn-v0_8"  # Updated default
     )
 
 @app.post("/summarize")
@@ -416,7 +433,7 @@ async def visualize_with_code(
     title: Optional[str] = Form(None),
     x_label: Optional[str] = Form(None),
     y_label: Optional[str] = Form(None),
-    style: str = Form("seaborn"),
+    style: str = Form("seaborn-v0_8"),  # Updated default
     filters: Optional[str] = Form(None)
 ):
     try:
@@ -467,7 +484,7 @@ async def visualize_with_natural_language(
     request: Request,
     file: UploadFile = File(...),
     prompt: str = Form(...),
-    style: str = Form("seaborn")
+    style: str = Form("seaborn-v0_8") 
 ):
     try:
         # Validate file