app.py

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.Textbox(placeholder="Augmented prompt"),
        gr.Textbox(placeholder="Response"),
        gr.Textbox(placeholder="Sources")],
    layout="vertical",
    title="CCL Playground",
    description="Enter a query to get response using RAG"
)

# Launch the Gradio app
iface.launch()