import os
from embedchain import Pipeline as App
import gradio as gr

os.environ["GOOGLE_API_KEY"] = "AIzaSyBbruzn10nez-0a-_60TA9R9h6qumLD1Es"

app = App.from_config(config={
    "llm": {
        "provider": "google",
        "config": {
            "model": "gemini-pro",
            "temperature": 0.5,
            "max_tokens": 1000,
            "top_p": 1,
            "stream": False,
        },
    },
    "embedder": {
        "provider": "google",
        "config": {
            "model": "models/embedding-001",
            "task_type": "retrieval_document", 
        },
    },
    "chunker": {
        "chunk_size": 2000,
        "chunk_overlap": 0,
        "length_function": "len",
    },
})

app.add('http://www.droit-afrique.com/uploads/Gabon-Code-2019-penal.pdf', data_type='pdf_file')




def query(message, history):
    return app.chat(message)


demo = gr.ChatInterface(query)

demo.launch()