import gradio as gr
from utils_model import ModelFactory

def retrieve_embeddings(input_text_query):
    model_type = 'mock'
    model_factory_obj = ModelFactory()
    model = model_factory_obj.create_model(model_type)

    query_embeddings = model.retrieve_embeddings(input_text_query)

    return query_embeddings

def build():
    app = gr.Interface(fn=retrieve_embeddings, inputs="text", outputs="dataframe")

    return app

def run(app):
    app.launch(server_name='0.0.0.0')

app = build()
run(app)