from fastapi import FastAPI, HTTPException,APIRouter,Request
from pydantic import BaseModel
from fastapi.responses import HTMLResponse
from fastapi.staticfiles import StaticFiles
from starlette.responses import FileResponse 
from fastapi.middleware.cors import CORSMiddleware
import requests
import json
from transformers import pipeline
app = FastAPI()
origins = ["*"]
app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
)
app.mount("/files/", StaticFiles(directory='../code'), name="index")
@app.get("/")
async def read_index():
    return FileResponse('app/html/index.html')
@app.post("/Event_Query/")
async def get_query(query,apikey):
    headers = {"X-API-Key": apikey}
    params = {"query": query}
    return requests.get(
        f"https://chat-api.you.com/smart?query={query}",
        params=params,
        headers=headers, 
    ).json()
@app.post("/Event_Summary/")
async def get_summary(query):
    summarizer = pipeline("summarization")
    summary = summarizer(query, max_length=10, min_length=10, do_sample=False)
    return (summary[0]['summary_text'])