Update main.py

main.py

@@ -5,11 +5,19 @@ 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 datetime import datetime, timedelta
+from fastapi_cache import FastAPICache
+from fastapi_cache.backends.inmemory import InMemoryBackend
 
 # Load environment variables from .env file
 #load_dotenv("keys.env")
 
 app = FastAPI()
+
+@app.on_event("startup")
+async def startup():
+    FastAPICache.init(InMemoryBackend(), prefix="fastapi-cache")
+    
 TOGETHER_API_KEY = os.getenv('TOGETHER_API_KEY')
 BRAVE_API_KEY = os.getenv('BRAVE_API_KEY')
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
@@ -54,7 +62,11 @@ 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.post("/generate_report")
+@cache(expire=3600)
 async def generate_report(request: Request, query: QueryModel):
     query_str = query.topic
     description = query.description