import gradio as gr
import banana_dev as banana
import os
api_key = os.getenv("API_KEY")
model_key = os.getenv("MODEL_KEY")
def call_model(prompt):
    model_parameters = {
        "prompt": prompt,
        "max_length":1000

    }
    out = banana.run(api_key, model_key, model_parameters)
    if out.message!="success":
            return "error"
    
    return out["modelOutputs"][0]["output"]

iface = gr.Interface(fn=call_model, inputs="text", outputs="text")
iface.launch()