Update modules/app.py

modules/app.py

@@ -18,13 +18,6 @@ pixart_sigma_base_url = os.getenv("PIXART_SIGMA_BASE_URL")
 allowed_user_agent = os.getenv("ALLOWED_USER_AGENT")
 
 
-async def verify_user_agent(request: Request):
-    print(request.headers)
-    user_agent = request.headers.get('user-agent', None)
-    if user_agent != allowed_user_agent:
-        raise HTTPException(status_code=403, detail="Access denied")
-    return True
-
 def generate_hash(length=12):
     # Characters that can appear in the hash
     characters = string.ascii_lowercase + string.digits
@@ -36,7 +29,7 @@ def generate_hash(length=12):
 async def read_root():
     return {"message": "Saqib's API"}
 
-@app.post("/whisper", dependencies=[Depends(verify_user_agent)])
+@app.post("/whisper")
 async def whisper(request: Request):
     data = await request.json()  # Extracting JSON data from request
     if "audio_url" not in data:
@@ -80,7 +73,7 @@ async def whisper(request: Request):
                     return f"API request failed: {post_resp.status}"
                 return await post_resp.json()
 
-@app.post("/img2location", dependencies=[Depends(verify_user_agent)])
+@app.post("/img2location")
 async def img2location(request: Request):
     request_json = await request.json()
     image_url = request_json.get("image_url", None)
@@ -163,7 +156,7 @@ async def img2location(request: Request):
             
             raise ValueError(f"Unexpected response: {json_response}")
 
-@app.post("/pixart-sigma", dependencies=[Depends(verify_user_agent)])
+@app.post("/pixart-sigma")
 async def pixart_sigma(request: Request):
     request_json = await request.json()
     prompt = request_json.get("prompt", None)