Update app.py

app.py

@@ -3,7 +3,8 @@ import streamlit as st
 from groq import Groq
 from pyvista import read
 from cadquery import exporters
-import requests
+from datasets import load_dataset
+import json
 
 # Initialize Groq API client
 client = Groq(api_key="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")
@@ -23,14 +24,14 @@ def perform_dfm_analysis(file_path):
     return result
 
 # Case 2: Template Selection and CAD Generation
-def get_template_list():
-    dataset_url = "https://huggingface.co/datasets/FreedomIntelligence/CADBench"
-    response = requests.get(f"{dataset_url}/resolve/main/metadata.json")
-    if response.status_code == 200:
-        templates = response.json().get("templates", [])
+def get_templates_from_dataset():
+    st.write("Loading templates from CADBench dataset...")
+    try:
+        dataset = load_dataset("FreedomIntelligence/CADBench", split="train")
+        templates = [entry["name"] for entry in dataset if "name" in entry]
         return templates
-    else:
-        st.error("Failed to fetch templates.")
+    except Exception as e:
+        st.error(f"Error loading dataset: {e}")
         return []
 
 def generate_cad_from_template(template_name, parameters):
@@ -57,14 +58,17 @@ if case == "Upload File for DFM":
                 st.error(f"Error in DFM Analysis: {e}")
 
 elif case == "Select Predefined Template":
-    templates = get_template_list()
+    templates = get_templates_from_dataset()
     if templates:
         template_name = st.selectbox("Choose a Template", templates)
         parameters = st.text_area("Enter Parameters (JSON format)", placeholder='{"length": 100, "width": 50}')
         if st.button("Generate CAD"):
             try:
-                generated_file = generate_cad_from_template(template_name, parameters)
+                params = json.loads(parameters) if parameters else {}
+                generated_file = generate_cad_from_template(template_name, params)
                 st.success("CAD Model Generated!")
                 st.download_button("Download CAD File", open(generated_file, "rb"), file_name=generated_file)
+            except json.JSONDecodeError:
+                st.error("Invalid JSON format for parameters.")
             except Exception as e:
                 st.error(f"Error in CAD Generation: {e}")