saphir / app /main.py
pytzen's picture
feature: usage documentation
e3a71c4
raw
history blame
1.92 kB
from fastapi import FastAPI, Request
from fastapi.responses import HTMLResponse
from fastapi.staticfiles import StaticFiles
from transformers import pipeline
# Set the title of your FastAPI application
app = FastAPI(title="Saphir AI Chatbot")
app.mount('/static', StaticFiles(directory='app/static'), name='static')
# Initialize your model
question_answerer = pipeline(task='question-answering',
model='distilbert-base-cased-distilled-squad')
@app.get('/', response_class=HTMLResponse)
def read_root():
"""
Root Endpoint
Returns the main HTML page for the chatbot interface.
"""
with open('app/index.html', 'r') as html_file:
return html_file.read()
@app.post('/chatbot')
async def ask_question(question: str, context: str):
"""
Chatbot Endpoint
Takes a question and context, returns the answer from the chatbot.
Parameters:
- question: A string containing the question to be answered.
- context: A string containing the context in which the question
should be answered.
Returns:
A JSON object with the 'answer' field containing the response from
the chatbot.
Example:
1. Go to the Swagger UI.
2. Scroll down to the /chatbot POST method section.
3. Click on the Try it out button.
4. In the question field, enter: "What is the capital of France?"
5. In the context field, enter: "France is a country in Western
Europe. It has several beautiful cities, including Paris, which
is its capital. France is known for its wine, sophisticated
cuisine, and landmarks like the Eiffel Tower."
6. Click the Execute button to submit the request.
7. Scroll down to see the server response, where you will find the
answer.
"""
result = question_answerer(question=question, context=context)
return {'answer': result['answer']}