Update app.py

app.py

@@ -61,6 +61,33 @@ def extract_image(pos_prompts, neg_prompts, img, threshold):
     output_image.paste(img, mask=final_mask)
     return output_image, final_mask
 
+
+@app.route('/api', methods=['POST'])
+def api():
+    data = request.form
+    img_url = data['input_image']
+    positive_prompts = data['positive_prompts']
+    negative_prompts = data['negative_prompts']
+    threshold = float(data['input_slider_T'])
+
+    # Download image from URL
+    response = requests.get(img_url)
+    img = Image.open(BytesIO(response.content))
+
+    # Process image
+    masks = get_masks(positive_prompts, negative_prompts, img, threshold)
+    final_mask = np.any(np.stack(masks), axis=0)
+
+    # Convert mask to image
+    final_mask = Image.fromarray(final_mask.astype(np.uint8) * 255, "L")
+
+    # Convert the final image to bytes
+    img_bytes = BytesIO()
+    final_mask.save(img_bytes, format='PNG')
+    img_bytes.seek(0)
+
+    return send_file(img_bytes, mimetype='image/png')
+    
 # Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# CLIPSeg: Image Segmentation Using Text and Image Prompts")