import gradio as gr
import torch
from diffusers import StableDiffusionPipeline
from PIL import image_diffsuion
import os




def image_diffsuion(promt, sample, steps, scale, seed):
    
    return images

def gligen_grounded_inpainting():
    return img

def Yolov8_SAM():
    return None
       
def TripoRS_Stable_Zero123_Wonder3D():
    return None
   
def convert_3d_to_2d():
    return  img
         
def image_fusion()：
    return img

with gr.Blocks() as main:
    with gr.Row():
        with gr.Column():
          gr.Markdown("""
          ### 请上传两张同一房间，不同角度的照片
          """)
          with gr.Group():  
            photo_input1 = gr.Image(label = "photo1", show_label = True, interactive = True)
            photo_input2 = gr.Image(label = "photo2", show_label = True, interactive = True)
          radio = gr.Radio(["复古风格","现代风格","田园风格","波西米亚","极简主义"],label = "选择你想要的房间风格")
          generate_btn = gr.Button("生成！")
        with gr.Column():
          gr.Markdown("""
          ### 请在下面的图片中框选出希望家具出现的位置，并在文本指令中输入对家具的描述，用分号分隔
          """)
          with gr.Group(): 
            image_output1 = gr.ImageEditor(label = "generated_image")
            gr.Textbox(label = "文本指令",info = "用分号（;）分隔")
            gen_btn = gr.Button("完成了")
        with gr.Column():
            gr.Markdown("""
            ### 生成结果，可下载和分享
            """)
            result1 = gr.Image(label = "generated_result1",interactive = False)
            result2 = gr.Image(label = "generated_result2",interactive = False) 

main.launch()