Spaces:
Sleeping
Sleeping
File size: 5,643 Bytes
3a47468 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 |
import gradio as gr
import matplotlib.pyplot as plt
import numpy as np
import logging
# Configure logging
logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
logger = logging.getLogger()
# Function to generate the die design visualization (with more shapes)
def generate_die_ANSYS(length, width, thickness, die_shape):
try:
# Log inputs
logger.info(f"Generating die design: Length={length}, Width={width}, Thickness={thickness}, Shape={die_shape}")
# Validate input dimensions
if length <= 0 or width <= 0 or thickness <= 0:
logger.error("Dimensions must be greater than zero.")
return "Error: Dimensions must be greater than zero.", None
# Simulate a realistic die design with multiple shapes
fig, ax = plt.subplots(figsize=(8, 8))
if die_shape == "Rectangle":
# Visualize a rectangular die
ax.add_patch(plt.Rectangle((0, 0), length, width, linewidth=2, edgecolor='blue', facecolor='cyan', label='Rectangle Die'))
ax.set_xlim(0, length + 20)
ax.set_ylim(0, width + 20)
ax.set_title(f"Progressive Die Design - Rectangle {length}x{width}")
logger.info(f"Created rectangular die with dimensions {length} x {width}.")
elif die_shape == "Circle":
# Visualize a circular die
circle = plt.Circle((length / 2, width / 2), radius=min(length, width) / 2, color='orange', ec='black', lw=2, label='Circular Die')
ax.add_patch(circle)
ax.set_xlim(0, length + 20)
ax.set_ylim(0, width + 20)
ax.set_title(f"Progressive Die Design - Circle {length}x{width}")
logger.info(f"Created circular die with diameter {length} and width {width}.")
elif die_shape == "Ellipse":
# Visualize an elliptical die
ellipse = plt.Ellipse((length / 2, width / 2), width, length, color='green', ec='black', lw=2, label='Elliptical Die')
ax.add_patch(ellipse)
ax.set_xlim(0, length + 20)
ax.set_ylim(0, width + 20)
ax.set_title(f"Progressive Die Design - Ellipse {length}x{width}")
logger.info(f"Created elliptical die with dimensions {length} x {width}.")
elif die_shape == "Hexagon":
# Visualize a hexagonal die
angle = np.linspace(0, 2 * np.pi, 7)
x = length / 2 * np.cos(angle) + length / 2
y = width / 2 * np.sin(angle) + width / 2
ax.fill(x, y, color='purple', edgecolor='black', lw=2, label='Hexagonal Die')
ax.set_xlim(0, length + 20)
ax.set_ylim(0, width + 20)
ax.set_title(f"Progressive Die Design - Hexagon {length}x{width}")
logger.info(f"Created hexagonal die with side length {length}.")
elif die_shape == "Polygon":
# Visualize a polygonal die (random polygon with 5 sides)
num_sides = 5
angle = np.linspace(0, 2 * np.pi, num_sides + 1)
radius = min(length, width) / 2
x = radius * np.cos(angle) + length / 2
y = radius * np.sin(angle) + width / 2
ax.fill(x, y, color='red', edgecolor='black', lw=2, label='Polygonal Die')
ax.set_xlim(0, length + 20)
ax.set_ylim(0, width + 20)
ax.set_title(f"Progressive Die Design - Polygon {length}x{width}")
logger.info(f"Created polygonal die with {num_sides} sides.")
ax.set_xlabel("X Dimension (mm)")
ax.set_ylabel("Y Dimension (mm)")
ax.legend(loc="upper right")
# Log the file saving step
filename = f"generated_die_{die_shape}_{length}x{width}.bmp"
plt.savefig(filename, format='bmp') # Save the plot as a BMP file
logger.info(f"Die design saved as {filename}.")
plt.close(fig) # Close the plot to prevent displaying twice
# Return success message and file path
return f"Die design saved successfully as {filename}. You can download the file.", filename
except Exception as e:
logger.error(f"Error generating die design: {str(e)}")
return f"Error generating die design: {str(e)}", None
# Gradio interface functions
def progressive_die_interface(length, width, thickness, die_shape):
# Generate die design and return file path and plot
die_message, filename = generate_die_ANSYS(length, width, thickness, die_shape)
# Log the completion of the task
logger.info(f"Die design process completed for shape {die_shape} with filename {filename}.")
return die_message, filename
# Create the Gradio Interface
with gr.Blocks() as app:
gr.Markdown("## Progressive Die Design Simulator")
gr.Markdown("This tool simulates the creation of progressive die designs. Select the die shape and enter dimensions to generate the die design.")
with gr.Row():
length = gr.Number(label="Length (mm)", value=100)
width = gr.Number(label="Width (mm)", value=50)
thickness = gr.Number(label="Thickness (mm)", value=10)
die_shape = gr.Dropdown(label="Die Shape", choices=["Rectangle", "Circle", "Ellipse", "Hexagon", "Polygon"], value="Rectangle")
die_button = gr.Button("Generate Die Design")
die_output = gr.Textbox(label="Die Design Message", interactive=False)
die_download = gr.File(label="Download Die Design Image")
die_button.click(progressive_die_interface, inputs=[length, width, thickness, die_shape], outputs=[die_output, die_download])
# Launch the app
app.launch()
|