diff --git a/backend/server.py b/backend/server.py index 683c1241..de26cd67 100644 --- a/backend/server.py +++ b/backend/server.py @@ -1,10 +1,11 @@ -from flask import Flask, request, jsonify +from flask import Flask, request, jsonify, g from flask_cors import CORS from pprint import pprint import nltk import subprocess import os import glob +import uuid from sklearn.metrics.pairwise import cosine_similarity from sklearn.feature_extraction.text import TfidfVectorizer @@ -31,6 +32,18 @@ CORS(app) print("Starting Flask App...") +@app.before_request +def assign_request_id(): + g.request_id = str(uuid.uuid4())[:8] + if request.path in ['/upload', '/getTranscript']: + app.logger.info(f"[{g.request_id}] Started {request.method} {request.path}") + +@app.after_request +def log_request_end(response): + if hasattr(g, 'request_id') and request.path in ['/upload', '/getTranscript']: + app.logger.info(f"[{g.request_id}] Completed {request.method} {request.path} with status {response.status_code}") + return response + SERVICE_ACCOUNT_FILE = './service_account_key.json' SCOPES = ['https://www.googleapis.com/auth/documents.readonly']