Trololindo commited on
Commit
1a97c8b
·
2 Parent(s): fb28d925870f43

Merge branch 'main' into hackaton-delenda-est

Browse files
backend/app/api/routes.py CHANGED
@@ -1,5 +1,7 @@
1
  import logging
2
  from fastapi import APIRouter, HTTPException, Request, status
 
 
3
 
4
  from app.models.schemas import (
5
  AnalysisRequest,
@@ -120,9 +122,12 @@ async def get_discord_guild_config(guild_id: str):
120
  tags=["Analysis"],
121
  summary="Analyze content for deepfake detection",
122
  )
123
- @limiter.limit("1/5seconds")
124
  async def analyze(request: Request, payload: AnalysisRequest) -> AnalysisResponse:
125
  guild_id = payload.guild_id
 
 
 
 
126
 
127
  if isinstance(payload, TextAnalysisRequest):
128
  content_type = "text"
 
1
  import logging
2
  from fastapi import APIRouter, HTTPException, Request, status
3
+ from slowapi.errors import RateLimitExceeded
4
+ from limits import parse
5
 
6
  from app.models.schemas import (
7
  AnalysisRequest,
 
122
  tags=["Analysis"],
123
  summary="Analyze content for deepfake detection",
124
  )
 
125
  async def analyze(request: Request, payload: AnalysisRequest) -> AnalysisResponse:
126
  guild_id = payload.guild_id
127
+ limit_item = parse("1/5seconds")
128
+
129
+ if not limiter.limiter.hit(limit_item, f"analyze:{guild_id}"):
130
+ raise HTTPException(status_code=429, detail="Rate limit exceeded for this guild")
131
 
132
  if isinstance(payload, TextAnalysisRequest):
133
  content_type = "text"
backend/app/core/config.py CHANGED
@@ -30,8 +30,10 @@ class Settings:
30
 
31
  AVAILABLE_MODELS = {
32
  "text": ["yaya36095/xlm-roberta-text-detector",
33
- "almanach/xlmr-chatgptdetect-noisy"],
34
- "image": ["capcheck/ai-image-detection"],
 
 
35
  }
36
 
37
  MAX_CONTENT_SIZES = {
 
30
 
31
  AVAILABLE_MODELS = {
32
  "text": ["yaya36095/xlm-roberta-text-detector",
33
+ "almanach/xlmr-chatgptdetect-noisy",
34
+ "bibbbu/multilingual-ai-human-detector_xlm-roberta-base"],
35
+ "image": ["capcheck/ai-image-detection",
36
+ "Hemg/Deepfake-image"],
37
  }
38
 
39
  MAX_CONTENT_SIZES = {