app.py

from fastapi import FastAPI
import uvicorn
import sys
import os
from fastapi.templating import Jinja2Templates
from starlette.responses import RedirectResponse
from fastapi.responses import Response 
from textSummarizer.pipeline.prediction import PredictionPipeline


text: str = 'What is text summarization'

app = FastAPI()

@app.get('/', tags = ['authentication'])
async def index():
    return RedirectResponse(url='/docs')


@app.get('/train')
async def training():
    try:
        os.system('Python main.py')
        return Response('Training successful')
    
    except Exception as e:
        return Response(f'Error Occured! {e}')
    

@app.post('/predict')
async def predict_route(text)
    try:

        obj=PredictionPipeline()
        text = obj.predict(text)
        return text
    except Exception as e:
        raise e