Update app.py

app.py

@@ -52,3 +52,40 @@ if case == "Predefined Template":
                 cq.exporters.export(cad_model, "cargo_crane.stl")
                 st.download_button("Download STEP File", open("cargo_crane.step", "rb").read(), "cargo_crane.step")
                 st.download_button("Download STL File", open("cargo_crane.stl", "rb").read(), "cargo_crane.stl")
+
+
+if case == "DFM Analysis":
+    st.subheader("Case 1: DFM Analysis")
+    cad_file = st.file_uploader("Upload a CAD file (.stl, .step, .dwg)", type=["stl", "step", "dwg"])
+    if cad_file:
+        file_format = os.path.splitext(cad_file.name)[1][1:]
+        analysis_result = analyze_dfm(cad_file, file_format)
+        st.success(analysis_result)
+
+elif case == "Predefined Template Selection":
+    st.subheader("Case 2: Predefined Template Selection")
+    template_names = dataset["name"]
+    selected_template = st.selectbox("Select a template", template_names)
+
+    if selected_template:
+        st.write(f"Selected Template: {selected_template}")
+        template_data = dataset.filter(lambda x: x["name"] == selected_template).to_pandas()
+        criteria = template_data.iloc[0]["criteria"]  # Already a dict
+
+        responses = {}
+        for criterion, prompt in criteria.items():
+            prefilled_text = f"Enter {criterion} ({prompt}):"
+            response = st.text_input(prefilled_text, key=criterion)
+            if response:
+                responses[criterion] = response
+
+        if st.button("Generate CAD Model"):
+            try:
+                model = generate_cad_model(responses)
+                file_format = st.selectbox("Select file format for download", ["stl", "step", "dwg"])
+                file_path = export_cad(model, file_format)
+                st.success("CAD Model Generated Successfully!")
+                with open(file_path, "rb") as f:
+                    st.download_button("Download CAD File", data=f.read(), file_name=f"model.{file_format}")
+            except Exception as e:
+                st.error(f"Error generating CAD model: {e}")