54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
"""
|
|
Web application to interact with the RAG system.
|
|
|
|
Author
|
|
------
|
|
Nicolas Rojas
|
|
"""
|
|
|
|
import requests
|
|
import gradio as gr
|
|
|
|
API_URL = "http://localhost:8000/query/"
|
|
|
|
|
|
def query_model(text: str) -> str:
|
|
"""Query the RAG system through its API.
|
|
|
|
Parameters
|
|
----------
|
|
text : str
|
|
Question or query made by the user.
|
|
|
|
Returns
|
|
-------
|
|
str
|
|
Response given by the API as a single string, or error logs when found.
|
|
"""
|
|
try:
|
|
# Send a POST request to the API
|
|
response = requests.post(API_URL, json={"query": text}, timeout=30)
|
|
|
|
# Check whether the request was successful
|
|
if response.status_code != 200:
|
|
return f"Error: API returned status code {response.status_code}"
|
|
result = response.json()["response"]
|
|
result += "\nThis information was obtained from the following files:\n"
|
|
for source in response.json()["source_files"]:
|
|
result += f"- {source}\n"
|
|
return result
|
|
except requests.RequestException as e:
|
|
return f"Error: Could not connect to the API. {str(e)}"
|
|
|
|
|
|
# Create the Gradio interface
|
|
interface = gr.Interface(
|
|
fn=query_model,
|
|
inputs=gr.Textbox(lines=5, placeholder="Enter text here..."),
|
|
outputs="text",
|
|
title="Docuchat",
|
|
description="Enter your question, and I'll provide with an answer from the documentation.",
|
|
)
|
|
|
|
# Launch the app
|
|
interface.launch()
|