feat: Update server to handle file process

astrolabe-expeditions · Feb 2, 2024 · 9b9f76e · 9b9f76e
1 parent 26610d7
commit 9b9f76e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/server.py b/src/server.py
@@ -12,6 +12,8 @@
 import pymongo
 import io
 
+from process import process_file
+
 DEFAULT_PORT = "8080"
 
 app = Flask(__name__)
@@ -26,8 +28,10 @@ def resource_not_found(e):
 @app.route("/", methods=HTTP_METHODS)
 def root():
     print(request.get_data(), flush=True)
-    response = make_response("Hello from container")
-    return response
+    data = json.loads(b'{"file_id":"65bbdcd4b4b804782a5a35e5"}')
+    print(f"Processing file n°{data['file_id']}")
+    result = process_file(data['file_id'])
+    return result
 
 
 if __name__ == "__main__":