Update visualization.py

visualization.py

@@ -1,13 +1,8 @@
-import pyvista as pv
-import numpy as np
 import matplotlib.pyplot as plt
 
-# Start virtual framebuffer for headless environments
-pv.start_xvfb()
-
 def visualize_results(simulator, length, width, thickness, stress, deformation):
     """
-    Generates 2D and 3D visualizations for simulation results.
+    Generates 2D visualizations for simulation results.
 
     Parameters:
         simulator (str): Name of the simulator (e.g., 'Python-Based Solver').
@@ -18,26 +13,18 @@ def visualize_results(simulator, length, width, thickness, stress, deformation):
         deformation (float): Deformation value.
 
     Returns:
-        tuple: Paths to the 2D and 3D visualization images.
+        str: Path to the 2D visualization image.
     """
-    # Generate 2D visualization
-    fig, ax = plt.subplots()
+    # Generate 2D bar chart
+    fig, ax = plt.subplots(figsize=(6, 4))
     ax.bar(["Stress", "Deformation"], [stress, deformation], color=["red", "blue"])
-    ax.set_title(f"Results ({simulator})")
-    plt.savefig("results_2d.png")
-    plt.close(fig)
-
-    # Generate 3D visualization
-    # Create a simple rectangular mesh
-    mesh = pv.Box(bounds=(0, length, 0, width, 0, thickness))
+    ax.set_title(f"Simulation Results ({simulator})")
+    ax.set_ylabel("Magnitude")
+    ax.grid(True, linestyle="--", alpha=0.6)
     
-    # Assign a scalar array to match the number of cells
-    cell_data = np.full(mesh.n_cells, stress)  # Assign stress value to each cell
-    mesh.cell_data["Stress"] = cell_data
+    # Save the chart
+    output_path = "results_2d.png"
+    plt.savefig(output_path)
+    plt.close(fig)
 
-    # Plot 3D visualization
-    plotter = pv.Plotter(off_screen=True)
-    plotter.add_mesh(mesh, scalars="Stress", cmap="coolwarm")
-    plotter.show(screenshot="results_3d.png")
-    
-    return "results_2d.png", "results_3d.png"
+    return output_path, None