SoumyaJ's picture
Upload 5 files
d1c2c35 verified
from fastapi import FastAPI
from fastapi.responses import JSONResponse
from fastapi.middleware.cors import CORSMiddleware
import pandas as pd
from general_recommendation import get_recommendations
from recommendation_explainer import explain_recommendation
import uvicorn
import numpy as np
app = FastAPI()
df = pd.read_csv('recommended_plan.csv')
# CORS middleware to allow requests from any origin
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
@app.get("/api/v1/recommendations")
def get_recommendations():
try:
sampled_data = df.sample(10)
results = sampled_data.to_dict(orient="records")
return JSONResponse(content={"recommendations": results})
except Exception as e:
return JSONResponse(status_code=500, content={"error": str(e)})
@app.get("/api/v1/explainrecommendation/{userid}")
def get_explanations(userid):
try:
explanation = explain_recommendation(userid, df)
return JSONResponse(content={"explanation": explanation})
except Exception as e:
return JSONResponse(status_code=500, content={"error": str(e)})
if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=8000)