Update app.py

app.py

@@ -56,8 +56,8 @@ def extract_image(pos_prompts, neg_prompts, img, threshold):
 
     return output_image, final_mask
 
-# Define Gradio interface
-iface = gr.Interface(
+# Gradio UI
+iface_ui = gr.Interface(
     fn=extract_image,
     inputs=[
         gr.Textbox(
@@ -77,5 +77,17 @@ iface = gr.Interface(
     ],
 )
 
-# Launch Gradio API with an explicit endpoint
-iface.launch(share=True, debug=True, host="0.0.0.0", port=7860, open_browser=False)
+# Launch Gradio UI
+iface_ui.launch()
+
+# Non-UI Version
+def run_non_ui(image_path, pos_prompts, neg_prompts, threshold):
+    img = Image.open(image_path)
+    output_image, output_mask = extract_image(pos_prompts, neg_prompts, img, threshold)
+
+    # Save or use the output_image and output_mask as needed
+    output_image.show()  # For demonstration purposes, opens the image with the default image viewer
+    output_mask.show()  # For demonstration purposes, opens the mask with the default image viewer
+
+# Example of using non-UI version
+# run_non_ui("path/to/your/image.jpg", "positive prompt", "negative prompt", 0.5)