add stats handle

app/api.py

@@ -1,10 +1,12 @@
 import config.config as config
-from app.src.src import pipeline_sentiment
+from app.src.src import pipeline_sentiment, pipeline_stats
 
 from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import pipeline
 import uvicorn
+import pandas as pd
+import os
 
 sentiment_model = pipeline(model=config.sentiment_model)
 app = FastAPI()
@@ -22,6 +24,12 @@ def get_comments(url_video: YouTubeUrl):
     data.to_csv(f"{config.DATA_FILE}", index=False)
     return {'message': 'Success'}
 
+@app.get('/stats')
+def get_stats_sent():
+    if f"{config.NAME_DATA}" in os.listdir(f"{config.PATH_DATA}"):
+        data = pd.read_csv(f"{config.DATA_FILE}")
+        return pipeline_stats(data)
+
 
 if __name__ == '__main__':
     uvicorn.run(app, host='127.0.0.1', port=80)