-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
53 lines (39 loc) · 1.32 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
from flask import Flask, request, jsonify
from atlas import execute_vector_search, update_plot
from sagemaker import create_embedding
app = Flask(__name__)
@app.route("/movies/<title>", methods=["PUT"])
def update_movie(title: str):
try:
request_json = request.get_json()
plot = request_json["plot"]
updated_movie = update_plot(title, plot)
if updated_movie:
return jsonify(
{
"message": "Movie updated successfully",
"updated_movie": updated_movie,
}
)
else:
return jsonify({"error": f"Movie with title {title} not found"}), 404
except Exception as e:
return jsonify({"error": str(e)}), 500
@app.route("/movies/search", methods=["POST"])
def search_movies():
try:
request_json = request.get_json()
query = request_json["query"]
embedding = create_embedding(query)
results = execute_vector_search(embedding)
jsonified_results = jsonify(
{
"message": "Movies searched successfully",
"results": results,
}
)
return jsonified_results
except Exception as e:
return jsonify({"error": str(e)}), 500
if __name__ == "__main__":
app.run(debug=True)