Update app.py

app.py

@@ -1,17 +1,7 @@
 import streamlit as st
 from datasets import load_dataset
-from groq import Groq
-import os
-
-# Initialize Groq API client
-client = Groq(api_key="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")
-
-# Load dataset
-@st.cache_data
-def load_cadbench_dataset():
-    return load_dataset("FreedomIntelligence/CADBench")
-
-dataset = load_cadbench_dataset()
+from pymesh import Mesh  # For 3D mesh generation
+import pyvista as pv  # For 3D mesh visualization
 
 # Page Title
 st.title("Quick CAD Model Generator")
@@ -20,31 +10,27 @@ st.title("Quick CAD Model Generator")
 st.sidebar.title("Input Options")
 input_option = st.sidebar.radio("Select an Option", ["Upload CAD File for DFM", "Select Predefined Template"])
 
-# Case 1: Upload CAD File for DFM Analysis
-if input_option == "Upload CAD File for DFM":
+# Case 1: Upload CAD File for DFM Analysis (placeholder)
+if input_option == "Upload CAD File for DFM Analysis":
     st.header("Upload a CAD File for DFM Analysis")
     uploaded_file = st.file_uploader(
-        "Upload your CAD file (.stl, .step, .iges, .dwg, .dxf, .scad)", 
+        "Upload your CAD file (.stl, .step, .iges, .dwg, .dxf, .scad)",
         type=["stl", "step", "iges", "dwg", "dxf", "scad"]
     )
 
-    if uploaded_file and st.button("Run DFM Analysis"):
-        st.write("**Performing Design for Manufacturing (DFM) Analysis...**")
-        file_extension = os.path.splitext(uploaded_file.name)[-1].lower()
-        st.write(f"Uploaded file format: **{file_extension}**")
-        
-        # Placeholder for DFM processing
-        if file_extension in [".stl", ".step", ".iges", ".dwg", ".dxf", ".scad"]:
-            st.success("File accepted for DFM analysis. Processing...")
-        else:
-            st.error("Unsupported file format for DFM analysis.")
+    if uploaded_file and st.button("Run DFM Analysis (Placeholder)"):
+        st.write("**DFM analysis functionality is currently under development.**")
+        st.write(f"Uploaded file format: **{os.path.splitext(uploaded_file.name)[-1].lower()}**")
 
 # Case 2: Select Predefined Template
 elif input_option == "Select Predefined Template":
     st.header("Select Predefined Template")
 
+    # Load dataset
+    dataset = load_dataset("FreedomIntelligence/CADBench")
+
     # Use 'name' column instead of 'template_name' as per dataset structure
-    template_names = dataset["train"]["name"]
+    template_names = dataset["train"]["name"].tolist()
     selected_template = st.selectbox("Choose a Template", template_names)
 
     if selected_template:
@@ -59,7 +45,7 @@ elif input_option == "Select Predefined Template":
         st.write("**Instruction:**", template_instruction)
         st.write("**Criteria:**", template_criteria)
 
-        # Input parameters for the template
+        # Input parameters for the template (extract parameters from criteria)
         st.subheader("Provide Required Parameters")
         user_inputs = {}
         for criterion in template_criteria.split(";"):
@@ -69,6 +55,26 @@ elif input_option == "Select Predefined Template":
 
         if st.button("Generate CAD Model"):
             st.write("**Generating CAD Model based on provided parameters...**")
-            # Placeholder for CAD generation logic
-            st.info(f"Parameters provided: {user_inputs}")
-            st.success("CAD Model generated. Download feature coming soon!")
+
+            # Example using PyMesh for generating a simple cube
+            if selected_template == "Cube":  # Replace with your template logic
+                if "width" in user_inputs and "height" in user_inputs and "depth" in user_inputs:
+                    width = float(user_inputs["width"])
+                    height = float(user_inputs["height"])
+                    depth = float(user_inputs["depth"])
+                    cube_vertices = [
+                        [0, 0, 0], [width, 0, 0], [width, height, 0], [0, height, 0],
+                        [0, 0, depth], [width, 0, depth], [width, height, depth], [0, height, depth]
+                    ]
+                    cube_faces = [
+                        [0, 1, 2, 3], [4, 5, 6, 7],
+                        [0, 4, 5, 1], [1, 5, 6, 2],
+                        [2, 6, 7, 3], [0, 3, 7, 4]
+                    ]
+                    cube_mesh = Mesh(vertices=cube_vertices, faces=cube_faces)
+
+                    # **Visualization using PyVista**
+                    py_mesh = pv.wrap(cube_mesh)  # Convert PyMesh to PyVista mesh
+                    plotter = pv.Plotter()
+                    plotter.add_mesh(py_mesh)
+                    plotter.show(jupyter_backend='static')