SoumyaJ commited on
Commit
b81311c
·
verified ·
1 Parent(s): 9f64fdf

changes to execute as API

Browse files
Files changed (1) hide show
  1. app.py +24 -25
app.py CHANGED
@@ -1,25 +1,24 @@
1
- from fastapi import FastAPI
2
- from fastapi.middleware.cors import CORSMiddleware
3
- from programmeRecommendation import recommendProgrammes
4
- import uvicorn
5
-
6
- app = FastAPI()
7
- origins = ["*"]
8
-
9
- #handle CORS issue
10
- app.add_middleware(
11
- CORSMiddleware,
12
- allow_origins=origins,
13
- allow_credentials=True,
14
- allow_methods=["*"],
15
- allow_headers=["*"]
16
- )
17
-
18
- @app.post("/cmsai/recommendmovies")
19
- def recommendMovies(scheduleDate : str, genre: list[str], referenceProgrammeTitle : str):
20
- moviesInRecommendationList = recommendProgrammes(scheduleDate, genre, referenceProgrammeTitle)
21
- return moviesInRecommendationList
22
-
23
- if __name__ == '__main__':
24
- recommendMovies('2024-12-01', ["Drama","Fantasy"], 'Halloween')
25
- #uvicorn.run(app= app)
 
1
+ from fastapi import FastAPI
2
+ from fastapi.middleware.cors import CORSMiddleware
3
+ from programmeRecommendation import recommendProgrammes
4
+ import uvicorn
5
+
6
+ app = FastAPI()
7
+ origins = ["*"]
8
+
9
+ #handle CORS issue
10
+ app.add_middleware(
11
+ CORSMiddleware,
12
+ allow_origins=origins,
13
+ allow_credentials=True,
14
+ allow_methods=["*"],
15
+ allow_headers=["*"]
16
+ )
17
+
18
+ @app.post("/cmsai/recommendmovies")
19
+ def recommendMovies(scheduleDate : str, genre: list[str], referenceProgrammeTitle : str):
20
+ moviesInRecommendationList = recommendProgrammes(scheduleDate, genre, referenceProgrammeTitle)
21
+ return moviesInRecommendationList
22
+
23
+ if __name__ == '__main__':
24
+ uvicorn.run(app= app)