cache-fix

cache/local_cache.py

@@ -24,16 +24,16 @@ def ttl_cache(key_name, media_type=None, ttl_secs=20):
                     del _cache[key]
                     del _cache_time[key]
                 else:
-                    if media_type == 'image/png':
-                        return StreamingResponse(BytesIO(_cache[key]), media_type=media_type)
-                    else:
-                        return _cache[key]
+                    # if media_type == 'image/png':
+                    #     return StreamingResponse(BytesIO(_cache[key]), media_type=media_type)
+                    # else:
+                    return _cache[key]
 
             # Call the actual function if not in cache or expired
             response = await func(*args, **kwargs)
-
+            print(response)
             # Cache the result
-            _cache[key] = response
+            _cache[key] = response.body
             _cache_time[key] = datetime.now()
 
             return response