Update app.py

app.py

@@ -19,7 +19,7 @@ def generate_apdl_script(file, press_force, material_json):
         model = cq.importers.importStep(step_file_path)
 
         # Export as STL for meshing
-        stl_file = step_file_path.replace(".step", ".stl")
+        stl_file = step_file_path.replace(".stp", ".stl")
         exporters.export(model, stl_file)
 
         # Generate APDL script
@@ -40,7 +40,7 @@ SOLVE
 /EXIT, SAVE
 """
         # Save APDL script to file
-        apdl_file_path = step_file_path.replace(".step", ".inp")
+        apdl_file_path = step_file_path.replace(".stp", ".inp")
         with open(apdl_file_path, "w") as apdl_file:
             apdl_file.write(apdl_script)
 
@@ -74,6 +74,11 @@ def main():
             submit_button = gr.Button("Submit")
 
         # Events
+        def handle_download(apdl_script, apdl_file_path):
+            if apdl_file_path and os.path.exists(apdl_file_path):
+                return apdl_file_path
+            return None
+
         generate_button.click(
             fn=generate_apdl_script,
             inputs=[step_input, force_input, material_input],