feat: longer token secrets

backend/main.py

@@ -13,7 +13,7 @@ from slowapi.middleware import SlowAPIMiddleware
 from slowapi.errors import RateLimitExceeded
 
 
-limiter = Limiter(key_func=get_remote_address, default_limits=["5/minute"])
+limiter = Limiter(key_func=get_remote_address, default_limits=['5/minute'])
 app = FastAPI()
 app.state.limiter = limiter
 app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)

backend/microverses.py

@@ -22,10 +22,10 @@ def create_microverse(modality, query, auth_result, text_encoder, text_image_enc
 
         query_embedding = encode(
             modality, query, text_encoder, text_image_encoder)
-        token = secrets.token_urlsafe(8)
+        token = secrets.token_urlsafe(16)
 
         if modality == 'text':
-            filename = secrets.token_urlsafe(8) + '.md'
+            filename = secrets.token_urlsafe(16) + '.md'
             open(knowledge_base_path / filename, 'w').write(query)
 
             microverses = json.load(open(microverses_path))
@@ -38,7 +38,7 @@ def create_microverse(modality, query, auth_result, text_encoder, text_image_enc
             }]
             json.dump(microverses, open(microverses_path, 'w'))
         elif modality == 'image':
-            filename = secrets.token_urlsafe(8) + '.jpg'
+            filename = secrets.token_urlsafe(16) + '.jpg'
             query = Image.open(io.BytesIO(query)).convert('RGB')
             query.save(knowledge_base_path / filename, quality=50)
 

backend/requirements.txt

@@ -0,0 +1,9 @@
+--find-links https://download.pytorch.org/whl/torch_stable.html
+
+torch==1.9.0+cpu
+numpy==1.17.4
+sentence_transformers==2.0.0
+fastapi==0.73.0
+Pillow==9.0.0
+secrets==1.0.2
+slowapi==0.1.5

backend/util.py

@@ -77,14 +77,14 @@ def save(modality, query, auth_result, text_encoder, text_image_encoder, silent=
                           'text' and open(knowledge_base_path / e['filename']).read() == query]
 
             if len(duplicates) == 0:
-                filename = secrets.token_urlsafe(8) + '.md'
+                filename = secrets.token_urlsafe(16) + '.md'
                 open(knowledge_base_path / filename, 'w').write(query)
         elif modality == 'image':
             duplicates = [e for e in thoughts if e['modality'] ==
                           'image' and open(knowledge_base_path / e['filename'], 'rb').read() == query]
 
             if len(duplicates) == 0:
-                filename = secrets.token_urlsafe(8) + '.jpg'
+                filename = secrets.token_urlsafe(16) + '.jpg'
                 query = Image.open(io.BytesIO(query)).convert('RGB')
                 query.save(knowledge_base_path / filename, quality=50)