dict cache

main.py

@@ -12,6 +12,7 @@ from youtube_transcript_api import YouTubeTranscriptApi
 from cachetools import TTLCache
 import asyncio
 import hashlib
+import time
 
 
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
@@ -218,22 +219,62 @@ def extract_transcript(youtube_link):
 
 
 
-cache = TTLCache(maxsize=100, ttl=400) 
+# cache = TTLCache(maxsize=100, ttl=400) 
    
+# @app.get("/summary")
+# async def test(url:str):
+#     try:
+#         cache_key = hashlib.sha256(url.encode()).hexdigest()
+#         cached_response = cache.get(cache_key)
+#         if cached_response:
+#             print("hii")
+#             return cached_response
+            
+#         status,content=scrweb(url)
+#         if status==200:
+#             response=genetate_gemini_content(prompt6,content)
+#             # print(response)
+#             cache[cache_key] = response
+#             return response
+#         else:
+#             return status
+
+#     except:
+#         raise HTTPException(status_code=500, detail="Internal Server Error")
+
+
+cache={}
+
+@app.on_event("startup")
+async def startup_event():
+    # Start a background task to clean up expired entries from the cache
+    asyncio.create_task(clean_up_cache())
+    
+async def clean_up_cache():
+    while True:
+        await asyncio.sleep(1000)  # Check cache every 60 seconds
+        
+        current_time = time.time()
+        expired_keys = [key for key, value in cache.items() if current_time - value["timestamp"] > 300]
+        for key in expired_keys:
+            del cache[key]
+            
 @app.get("/summary")
 async def test(url:str):
     try:
         cache_key = hashlib.sha256(url.encode()).hexdigest()
-        cached_response = cache.get(cache_key)
-        if cached_response:
-            print("hii")
-            return cached_response
-            
+
+        # Check if the response is already cached
+        if cache_key in cache:
+            cached_response = cache[cache_key]
+            if time.time() - cached_response["timestamp"] <= 300:  # Check if cached data is still valid (5 minutes)
+                return cached_response["data"]
+
         status,content=scrweb(url)
         if status==200:
             response=genetate_gemini_content(prompt6,content)
             # print(response)
-            cache[cache_key] = response
+            cache[cache_key] = {"data": response, "timestamp": time.time()}
             return response
         else:
             return status
@@ -241,7 +282,6 @@ async def test(url:str):
     except:
         raise HTTPException(status_code=500, detail="Internal Server Error")
 
-
     
 
 @app.get("/ut")