import gradio as gr
from gradio_client import Client

access_token = "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"

# Function to call the Gradio API
def call_gradio_api(user_input):
    
    client = Client("https://traversaal-internal-rag-faiss-gpt3-5.hf.space/", hf_token=access_token)
    result = client.predict(user_input, api_name="/predict")
    return result[0], result[1], result[2]
    
# Interface for the Gradio app
iface = gr.Interface(
    fn=call_gradio_api, 
    inputs="text",
    outputs = [gr.outputs.Textbox(label="Augmented Prompt"),
        gr.outputs.Textbox(label="Response"),
        gr.outputs.Textbox(label="Sources")],
    layout="vertical",
    title="CCL Playground",
    description="Enter a query to get response using RAG"
)

# Launch the Gradio app
iface.launch()