Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 import cadquery as cq
 from cadquery import exporters
@@ -5,9 +6,9 @@ import trimesh
 import pyvista as pv
 from io import BytesIO
 import json
-import os
 import atexit
 import glob
+import os
 
 # Cleanup function to remove temporary files
 def cleanup_temp_files():
@@ -128,15 +129,9 @@ def main():
             inputs=[step_input, force_input, material_input],
             outputs=[script_output, download_button]
         )
-
-    app.launch(server_port=7861, debug=True)
-
-import os
-import gradio as gr
+    return app
 
 if __name__ == "__main__":
-    # Use environment variable or dynamic port assignment
+    app = main()
     port = int(os.environ.get("GRADIO_SERVER_PORT", 7861))  # Default to 7861
     app.launch(server_port=port, debug=True)
-
-