#Script added by SPDraptor
import spaces
from typing import Optional
import subprocess
subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
import torch
print("cuda present = ",torch.cuda.is_available())
import os
import sys
import gradio as gr
from PIL import Image

from modules import masking_module



DESCRIPTION = "Welcome to Raptor APIs"

css = """
  #output {
    height: 500px; 
    overflow: auto; 
    border: 1px solid #ccc; 
  }
"""

with gr.Blocks(css=css) as demo:
    gr.Markdown(DESCRIPTION)
    with gr.Tab(label="OBJ_mask"):
        with gr.Row():
            with gr.Column():
                image = gr.Image(label="Input main Picture")
                image_object = gr.Textbox(label="object name")
                mask_btn = gr.Button(value="createMask")
                
            with gr.Column():
                output_mask = gr.Image(label="mask")
                mask_btn.click(masking_module.masking_process,inputs=[image,image_object],outputs=output_mask,api_name="masking_step")

demo.launch()