import gradio as gr
from Lex import *

'''
lex = Lexica(query="man woman fire snow").images()
'''

inputs = gr.Textbox(label = 'Enter prompt to search Lexica.art')
#gr.Slider(label='Number of images ', minimum = 4, maximum = 20, step = 1, value = 4)]

outputs = gr.Gallery(lable='Output gallery').style(grid=3,height=768)
#gr.Dataframe(label='prompts for corresponding images')]

def lexica(prompt):
    lex = Lexica(query=prompt).images()
    return lex

interface = gr.Interface(lexica,
    inputs=inputs,
    outputs=outputs,
    examples =[
                'Chinese Traditional Culture',
                'trending digital art',
                'beautiful home',
                'interior design of living room'
                ]
                  ,
     title = " 🔍 🖌️🎨 Lexica Art - A Search Engine for Generative Art Prompts and Works ",
     description = "🔍🖌️ 🎨 lexica huggingface space  , Find inspiration and discover new generative artworks with Lexica Art \n\n a search engine built  by  by @[Sharif shameem](https://twitter.com/sharifshameem) . Explore a vast collection of prompts and corresponding artworks, and let your imagination take over as you create your own masterpieces.",
    cache_examples = False,
                        )

interface.launch("0.0.0.0")