Update app.py

app.py

@@ -1,8 +1,9 @@
 import numpy as np
 import pandas as pd
 from dateutil import parser
-from flask import Flask, render_template
-from flask_cors import cross_origin, CORS
+from quart_cors import cors
+from quart import Quart
+from quart import render_template
 from db_operations.db_operations import DBOperations
 import logging
 import traceback 
@@ -15,8 +16,8 @@ from word_cloud import get_frequent_words_html
 from config import NEWS_RETENTION_SECONDS, UK_EDITION_URL
 
 
-app = Flask(__name__)
-CORS(app)
+app = Quart(__name__)
+app = cors(app, allow_origin="*")
 redis_client = redis.Redis(host='localhost', port=6379, decode_responses=True)
 logging.basicConfig(format='%(asctime)s %(message)s')
 logging.warning(f'Is Redis available?: {redis_client.ping()}') 
@@ -106,12 +107,12 @@ def elapsed_time_str(mins):
 
 
 
-def fetch_from_db(fetch_flag):
+async def fetch_from_db(fetch_flag):
     try:
         logging.warning(f'[session_id: {session_id}] fetch_flag: {fetch_flag}')
         if fetch_flag == 1:
-            final_df = db.read_news_from_db()
-            freq_tokens = get_frequent_words_html(final_df)
+            final_df = await db.read_news_from_db()
+            freq_tokens = await get_frequent_words_html(final_df)
             logging.warning(f'[session_id: {session_id}] Fetched From DB')
 
             final_df['_id'] = final_df['_id'].astype('str')
@@ -132,8 +133,7 @@ def fetch_from_db(fetch_flag):
 
 
 @app.route("/")
-@cross_origin()
-def index():
+async def index():
     """
     Entry point
     """
@@ -143,7 +143,7 @@ def index():
         src_str = ''
         status_code = 200
         logging.warning(f'[session_id: {session_id}] Entering the application')
-        final_df, freq_tokens = fetch_from_db(is_db_fetch_reqd())
+        final_df, freq_tokens = await fetch_from_db(is_db_fetch_reqd())
         if len(final_df) > 1:
             
             final_df["parsed_date"] = [correct_date(date_) for date_ in final_df['parsed_date']]
@@ -249,9 +249,9 @@ def index():
     result_str += '</form></div>'
     logging.warning(f'[session_id: {session_id}] Successfully rendered template')
     gc.collect()
-    return render_template("index.html", body=result_str), status_code
+    return await render_template("index.html", body=result_str), status_code
 
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, workers=3, threads=3) # workers=(2*ncores) + 1, threads= (2 to 4*ncores) + 1
+    app.run(host="0.0.0.0", port=7860, workers=5, threads=5) # workers=(2*ncores) + 1, threads= (2 to 4*ncores) + 1