Update main.py

main.py

@@ -5,6 +5,9 @@ from typing import List, Dict, Any
 from helper_functions_api import md_to_html, search_brave, fetch_and_extract_content, limit_tokens, together_response, insert_data
 import os
 from dotenv import load_dotenv, find_dotenv
+from fastapi_cache import FastAPICache
+from fastapi_cache.decorator import cache
+from fastapi_cache.backends.inmemory import InMemoryBackend
 
 # Load environment variables from .env file
 #load_dotenv("keys.env")
@@ -54,7 +57,12 @@ class QueryModel(BaseModel):
     output_format: str = Query(default="Tabular Report", description="Output format for the report", enum=["Chat", "Full Text Report", "Tabular Report", "Tables only"])
     data_format: str = Query(default="Structured data", description="Type of data to extract from the internet", enum=["No presets", "Structured data", "Quantitative data"])
 
+@app.on_event("startup")
+async def startup():
+    FastAPICache.init(InMemoryBackend(), prefix="fastapi-cache")
+
 @app.post("/generate_report")
+@cache(expire=60 * 60 * 24 * 7)  # Cache for 7 days
 async def generate_report(request: Request, query: QueryModel):
     query_str = query.topic
     description = query.description