chatbot-PUC-Rio / app_details.py
lauraparra28's picture
Upload 10 files
2e1aa7a verified
raw
history blame
665 Bytes
import gradio as gr
import functions as fn
import json
data = fn.load_embeddings()
num_documents = data['num_documents']
num_segment_contents = data['num_segment_contents']
with open("gradio.json") as f:
config = json.load(f)
config['description'] = config['description'].format(num_documents=num_documents, num_segment_contents=num_segment_contents)
config['title'] += " - Interface de Respostas Detalhadas"
def on_submit(query, history):
response = fn.rag_response(query, data=data, detailed_response=True)
return gr.HTML(response.replace("\n", "<br>"))
demo = gr.ChatInterface(fn=on_submit, **config)
demo.launch()