| |
|
| | import uvicorn |
| | import pandas as pd |
| | from fastapi import FastAPI |
| |
|
| | description = """ |
| | # Climate news generator |
| | |
| | Generate a random news about climate change from [readerbench/fakenews-climate-fr](https://huggingface.co/datasets/readerbench/fakenews-climate-fr) dataset. |
| | |
| | Use for testing data pipelines. |
| | |
| | Where you can: |
| | * `/get_news` : Get a random news |
| | |
| | Check out documentation for more information on each endpoint. |
| | """ |
| |
|
| | tags_metadata = [ |
| | { |
| | "name": "Generator", |
| | "description": "Endpoints that generate news about climate change", |
| | }, |
| | ] |
| |
|
| | data = pd.read_csv("hf://datasets/readerbench/fakenews-climate-fr/fake-fr.csv") |
| |
|
| | app = FastAPI( |
| | title="API for Climate news generator", |
| | description=description, |
| | version="1.0", |
| | contact={ |
| | "name": "Olivier-52", |
| | "url": "https://huggingface.co/Olivier-52", |
| | }, |
| | openapi_tags=tags_metadata,) |
| |
|
| | @app.get("/") |
| | def index(): |
| | """Return a message to the user. |
| | |
| | This endpoint does not take any parameters and returns a message |
| | to the user. It is used to test the API. |
| | |
| | Returns: |
| | str: A message to the user. |
| | """ |
| | return "Hello world! Go to /docs to try the API." |
| |
|
| | @app.get("/get_news", tags=["Generator"]) |
| | def get_news(): |
| | """Return a random news about climate change.""" |
| | if data.empty: |
| | return {"error": "No data available"} |
| | return data.sample(1).to_dict(orient="records") |
| |
|
| | if __name__ == "__main__": |
| | import uvicorn |
| | uvicorn.run(app, host="localhost", port=8000) |
