import gradio as gr
import requests

from g4f import Provider, models
from langchain.llms.base import LLM

from langchain_g4f import G4FLLM


url = "https://app.embedchain.ai/api/v1/pipelines/f14b3df8-db63-456c-8a7f-4323b4467271/context/"



def greet(name):
    payload = {
  "query": f"{name}",
  "count": 5
    }
    headers = {
  'Authorization': 'Token ec-pbVFWamfNAciPwb18ZwaQkKKUCCBnafko9ydl3Y5',
}

    
    response = requests.request("POST", url, headers=headers, json=payload)

    print(name)
    c = response.text
    llm = LLM = G4FLLM(
        model=models.gpt_35_turbo,
        provider=Provider.Liaobots,
    )

    res = llm(f"""
    contexte:{c}.  
    
    Question:{name}""")
    print(res) 
    return res



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